towk/makeshift
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

refactor: minor changes and fixes

Browse source
This commit is contained in:
David Allen 2025-09-20 15:44:08 -06:00
parent 72c52fbac6
commit c1c5ec1625
Signed by: towk
GPG key ID: 0430CDBE22619155
3 changed files with 39 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

8
pkg/storage/memory.go
View file

@ -20,11 +20,15 @@ func (ms *MemoryStorage) Cleanup() error {
}
func (ms *MemoryStorage) SetKWArgs(kw *kwargs.KWArgs) error {
return ms.Set(kwargs.RESERVED_KEY, kw)
ms.Data[kwargs.RESERVED_KEY] = kw
return nil
}
func (ms *MemoryStorage) GetKWArgs() (*kwargs.KWArgs, error) {
kw, err := ms.Get(kwargs.RESERVED_KEY)
if err != nil {
return nil, err
}
return kw.(*kwargs.KWArgs), err
}
@ -37,7 +41,7 @@ func (ms *MemoryStorage) Get(k string) (any, error) {
}
func (ms *MemoryStorage) Set(k string, v any) error {
if k == "kwargs" {
if k == kwargs.RESERVED_KEY {
return fmt.Errorf("cannot set reserved key '%s' (use SetKWArgs() instead)", k)
}
ms.Data[k] = v