rename dir (#283)

[bytom/vapor.git] / toolbar / federation / database / asset_store.go

diff --git a/toolbar/federation/database/asset_store.go b/toolbar/federation/database/asset_store.go
new file mode 100644 (file)
index 0000000..54b09c5
--- /dev/null
+++ b/toolbar/federation/database/asset_store.go
@@ -0,0 +1,68 @@
+package database
+
+import (
+       "fmt"
+
+       "github.com/golang/groupcache/lru"
+       "github.com/jinzhu/gorm"
+
+       "github.com/vapor/errors"
+       "github.com/vapor/toolbar/federation/database/orm"
+)
+
+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
+       db    *gorm.DB
+}
+
+func NewAssetStore(db *gorm.DB) *AssetStore {
+       return &AssetStore{
+               cache: lru.New(maxAssetCached),
+               db:    db,
+       }
+}
+
+func (a *AssetStore) GetByOrmID(ormID uint64) (*orm.Asset, error) {
+       if v, ok := a.cache.Get(fmtOrmIDKey(ormID)); ok {
+               return v.(*orm.Asset), nil
+       }
+
+       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")
+       }
+
+       a.cache.Add(fmtOrmIDKey(asset.ID), asset)
+       a.cache.Add(fmtAssetIDKey(asset.AssetID), asset)
+       return asset, nil
+}
+
+func (a *AssetStore) GetByAssetID(assetID string) (*orm.Asset, error) {
+       if v, ok := a.cache.Get(fmtAssetIDKey(assetID)); ok {
+               return v.(*orm.Asset), nil
+       }
+
+       asset := &orm.Asset{AssetID: assetID}
+       if err := a.db.Where(asset).First(asset).Error; err != nil {
+               return nil, errors.Wrap(err, "asset not found in memory and mysql")
+       }
+
+       a.cache.Add(fmtOrmIDKey(asset.ID), asset)
+       a.cache.Add(fmtAssetIDKey(asset.AssetID), asset)
+       return asset, nil
+}