From: HAOYUatHZ Date: Wed, 19 Jun 2019 08:26:24 +0000 (+0800) Subject: fmtKey X-Git-Url: http://git.osdn.net/view?p=bytom%2Fvapor.git;a=commitdiff_plain;h=f308f94f3d727902512af3a8f8b9e9b98d4a9f71 fmtKey --- diff --git a/federation/database/asset_store.go b/federation/database/asset_store.go index 6019ce58..e5918f17 100644 --- a/federation/database/asset_store.go +++ b/federation/database/asset_store.go @@ -1,6 +1,8 @@ package database import ( + "fmt" + "github.com/golang/groupcache/lru" "github.com/jinzhu/gorm" @@ -8,7 +10,20 @@ import ( "github.com/vapor/federation/database/orm" ) -const maxAssetCached = 1024 +const ( + maxAssetCached = 1024 + + ormIDPrefix = "ormID" + assetIDPrefix = "assetID" +) + +func fmtOrmIDKey(ormID uint64) string { + return fmt.Sprintf("%s:%d", ormIDPrefix, ormID) +} + +func fmtAssetIDKey(assetID string) string { + return fmt.Sprintf("%s:%s", assetIDPrefix, assetID) +} type AssetStore struct { cache *lru.Cache @@ -22,8 +37,8 @@ func NewAssetStore(db *gorm.DB) *AssetStore { } } -func (a *AssetStore) GetByOrmID(id uint64) (*orm.Asset, error) { - asset := &orm.Asset{ID: id} +func (a *AssetStore) GetByOrmID(ormID uint64) (*orm.Asset, error) { + asset := &orm.Asset{ID: ormID} if err := a.db.Where(asset).First(asset).Error; err != nil { return nil, errors.Wrap(err, "asset not found by orm id") }