Merge pull request #48 from calvinmclean/fix/SetStorage

Re-add SetStorage method
calvinmclean · Feb 27, 2024 · d785a01 · d785a01
2 parents 71d1ae0 + 55ade53
commit d785a01
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/babyapi.go b/babyapi.go
@@ -377,6 +377,13 @@ func (a *API[T]) ChildAPIs() map[string]RelatedAPI {
  return children
 }
 
+func (a *API[T]) SetStorage(s Storage[T]) *API[T] {
+ a.panicIfReadOnly()
+
+ a.Storage = s
+ return a
+}
+
 func (a *API[T]) panicIfReadOnly() {
  if !a.readOnly.TryLock() {
  panic(errors.New("API cannot be modified after starting"))

diff --git a/examples/storage/main.go b/examples/storage/main.go
@@ -25,7 +25,7 @@ func main() {
  panic(err)
  }
 
- api.Storage = storage.NewClient[*User](db, "User")
+ api.SetStorage(storage.NewClient[*User](db, "User"))
 
  api.RunCLI()
 }
diff --git a/extensions/kv_storage.go b/extensions/kv_storage.go
@@ -57,7 +57,7 @@ func (h KeyValueStorage[T]) Apply(api *babyapi.API[T]) error {
  storageKeyPrefix = api.Name()
  }
 
- api.Storage = storage.NewClient[T](db, storageKeyPrefix)
+ api.SetStorage(storage.NewClient[T](db, storageKeyPrefix))
 
  return nil
 }