grassrootseconomics/sarafu-api
7
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

Make prefix settable in dev api

Browse Source
This commit is contained in:
lash 2025-01-15 07:22:14 +00:00
parent c128678f4a
commit bca7c5de96
Signed by: lash
GPG Key ID: 21D2E7BB88C2A746
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
dev/api.go
View File

@ -103,6 +103,7 @@ type DevAccountService struct {
autoVoucherValue map[string]int autoVoucherValue map[string]int
defaultAccount string defaultAccount string
emitterFunc event.EmitterFunc emitterFunc event.EmitterFunc
pfx []byte
// accountsSession map[string]string // accountsSession map[string]string
} }
@ -117,6 +118,7 @@ func NewDevAccountService(ctx context.Context, ss storage.StorageService) *DevAc
txsTrack: make(map[string]string), txsTrack: make(map[string]string),
autoVoucherValue: make(map[string]int), autoVoucherValue: make(map[string]int),
defaultAccount: zeroAddress, defaultAccount: zeroAddress,
pfx: []byte("__"),
} }
if ss != nil { if ss != nil {
var err error var err error
@ -143,6 +145,15 @@ func (das *DevAccountService) WithEmitter(fn event.EmitterFunc) *DevAccountServi
return das return das
} }
func (das *DevAccountService) WithPrefix(pfx []byte) *DevAccountService {
das.pfx = pfx
return das
}
func (das *DevAccountService) prefixKeyFor(k string, v string) []byte {
return append(das.pfx, []byte(k + "_" + v)...)
}
func (das *DevAccountService) loadAccount(ctx context.Context, pubKey string, v []byte) error { func (das *DevAccountService) loadAccount(ctx context.Context, pubKey string, v []byte) error {
var acc Account var acc Account
@ -201,7 +212,7 @@ func (das *DevAccountService) loadAll(ctx context.Context) error {
if k == nil { if k == nil {
break break
} }
if !bytes.HasPrefix(k, []byte("__")) { if !bytes.HasPrefix(k, das.pfx) {
continue continue
} }
@ -305,7 +316,7 @@ func (das *DevAccountService) saveAccount(ctx context.Context, acc Account) erro
if das.db == nil { if das.db == nil {
return nil return nil
} }
k := "__account_" + acc.Address k := das.prefixKeyFor("account", acc.Address)
v, err := json.Marshal(acc) v, err := json.Marshal(acc)
if err != nil { if err != nil {
return err return err
@ -450,7 +461,7 @@ func (das *DevAccountService) VoucherData(ctx context.Context, address string) (
} }
func (das *DevAccountService) saveTokenTransfer(ctx context.Context, mytx Tx) error { func (das *DevAccountService) saveTokenTransfer(ctx context.Context, mytx Tx) error {
k := "__tx_" + mytx.Hsh k := das.prefixKeyFor("tx", mytx.Hsh)
v, err := json.Marshal(mytx) v, err := json.Marshal(mytx)
if err != nil { if err != nil {
return err return err