feat: use cache

formancehq · Aug 16, 2023 · a29d448 · a29d448
1 parent 34f16e8
commit a29d448
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/pkg/core/account.go b/pkg/core/account.go
@@ -15,6 +15,18 @@ type Account struct {
 	Metadata Metadata       `json:"metadata" swaggertype:"object"`
 }
 
+func (v Account) Copy() *Account {
+	data, err := json.Marshal(v)
+	if err != nil {
+		panic(err)
+	}
+	ret := &Account{}
+	if err := json.Unmarshal(data, ret); err != nil {
+		panic(err)
+	}
+	return ret
+}
+
 type AccountWithVolumes struct {
 	Account
 	Volumes  AssetsVolumes  `json:"volumes"`

diff --git a/pkg/storage/sqlstorage/accounts.go b/pkg/storage/sqlstorage/accounts.go
@@ -200,6 +200,12 @@ func (s *Store) GetAccounts(ctx context.Context, q ledger.AccountsQuery) (api.Cu
 }
 
 func (s *Store) GetAccount(ctx context.Context, addr string) (*core.Account, error) {
+
+	entry, ok := s.cache.Get(addr)
+	if ok {
+		return entry.(*core.AccountWithVolumes).Account.Copy(), nil
+	}
+
 	sb := sqlbuilder.NewSelectBuilder()
 	sb.Select("address", "metadata").
 		From(s.schema.Table("accounts")).