Cache scripts

cmd/api/handlers/code.go

@@ -71,7 +71,7 @@ func getContractCodeJSON(c context.Context, ctx *config.Context, address string,
  if err != nil {
  return res, err
  }
- script, err := ctx.Contracts.Script(c, address, symLink)
+ script, err := ctx.Cache.Script(c, address, symLink)
  if err != nil {
  return res, err
  }

cmd/api/handlers/fork.go

@@ -45,7 +45,7 @@ func buildStorageDataFromForkRequest(c context.Context, ctxs config.Contexts, re
  if err != nil {
  return nil, err
  }
- scriptData, err = getScriptBytes(c, ctx.Contracts, req.Address, symLink)
+ scriptData, err = getScriptBytes(c, ctx.Cache, req.Address, symLink)
  if err != nil {
  return nil, err
  }

cmd/api/handlers/prepare.go

@@ -326,7 +326,7 @@ func getErrorLocation(c context.Context, ctx *config.Context, operation operatio
  if err != nil {
  return GetErrorLocationResponse{}, err
  }
- code, err := getScriptBytes(c, ctx.Contracts, operation.Destination.Address, proto.SymLink)
+ code, err := getScriptBytes(c, ctx.Cache, operation.Destination.Address, proto.SymLink)
  if err != nil {
  return GetErrorLocationResponse{}, err
  }

cmd/api/handlers/run_code.go

@@ -148,7 +148,7 @@ func RunCode() gin.HandlerFunc {
  return
  }
 
- scriptBytes, err := getScriptBytes(c.Request.Context(), ctx.Contracts, req.Address, state.Protocol.SymLink)
+ scriptBytes, err := getScriptBytes(c.Request.Context(), ctx.Cache, req.Address, state.Protocol.SymLink)
  if handleError(c, ctx.Storage, err, 0) {
  return
  }
@@ -246,7 +246,7 @@ func parseAppliedRunCode(c context.Context, ctx *config.Context, response noderp
  s = script
  } else {
  var err error
- s, err = getScript(c, ctx.Contracts, op.Destination, proto.SymLink)
+ s, err = getScript(c, ctx.Cache, op.Destination, proto.SymLink)
  if err != nil {
  return nil, err
  }

cmd/api/handlers/script.go

@@ -5,20 +5,21 @@ import (
 
  "github.com/baking-bad/bcdhub/internal/bcd/ast"
  "github.com/baking-bad/bcdhub/internal/bcd/consts"
+ "github.com/baking-bad/bcdhub/internal/cache"
  "github.com/baking-bad/bcdhub/internal/models/block"
  "github.com/baking-bad/bcdhub/internal/models/contract"
 )
 
-func getScript(ctx context.Context, contracts contract.Repository, address, symLink string) (*ast.Script, error) {
- data, err := getScriptBytes(ctx, contracts, address, symLink)
+func getScript(ctx context.Context, cache *cache.Cache, address, symLink string) (*ast.Script, error) {
+ data, err := getScriptBytes(ctx, cache, address, symLink)
  if err != nil {
  return nil, err
  }
  return ast.NewScriptWithoutCode(data)
 }
 
-func getScriptBytes(ctx context.Context, contracts contract.Repository, address, symLink string) ([]byte, error) {
- script, err := contracts.Script(ctx, address, symLink)
+func getScriptBytes(ctx context.Context, cache *cache.Cache, address, symLink string) ([]byte, error) {
+ script, err := cache.Script(ctx, address, symLink)
  if err != nil {
  return nil, err
  }

internal/cache/cache.go

@@ -56,6 +56,7 @@ func (cache *Cache) ContractTags(ctx context.Context, address string) (types.Tag
  return c.Tags, nil
  })
  if err != nil {
+ cache.Delete(key)
  return 0, err
  }
  return item.Value().(types.Tags), nil
@@ -68,6 +69,7 @@ func (cache *Cache) TezosBalance(ctx context.Context, address string, level int6
  return cache.rpc.GetContractBalance(ctx, address, level)
  })
  if err != nil {
+ cache.Delete(key)
  return 0, err
  }
  return item.Value().(int64), nil
@@ -84,6 +86,7 @@ func (cache *Cache) StorageTypeBytes(ctx context.Context, address, symLink strin
  return cache.contracts.ScriptPart(ctx, address, symLink, consts.STORAGE)
  })
  if err != nil {
+ cache.Delete(key)
  return nil, err
  }
  return item.Value().([]byte), nil
@@ -96,7 +99,20 @@ func (cache *Cache) ProtocolByID(ctx context.Context, id int64) (protocol.Protoc
  return cache.protocols.GetByID(ctx, id)
  })
  if err != nil {
+ cache.Delete(key)
  return protocol.Protocol{}, err
  }
  return item.Value().(protocol.Protocol), nil
 }
+
+func (cache *Cache) Script(ctx context.Context, address, symLink string) (contract.Script, error) {
+ key := fmt.Sprintf("script:%s:%s", address, symLink)
+ item, err := cache.Fetch(key, time.Hour, func() (interface{}, error) {
+ return cache.contracts.Script(ctx, address, symLink)
+ })
+ if err != nil {
+ cache.Delete(key)
+ return contract.Script{}, err
+ }
+ return item.Value().(contract.Script), nil
+}

internal/parsers/operations/transaction.go

@@ -112,7 +112,7 @@ func (p Transaction) parseContractParams(ctx context.Context, data noderpc.Opera
  }
  }
 
- scriptEntity, err := p.ctx.Contracts.Script(ctx, tx.Destination.Address, p.protocol.SymLink)
+ scriptEntity, err := p.ctx.Cache.Script(ctx, tx.Destination.Address, p.protocol.SymLink)
  if err != nil {
  if !tx.Internal {
  return nil