-
Notifications
You must be signed in to change notification settings - Fork 3.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
refactor(stf): remove RunWithCtx (#21739)
- Loading branch information
1 parent
98eb0b7
commit d6364b8
Showing
16 changed files
with
218 additions
and
85 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,107 @@ | ||
| package services | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
|
|
||
| "cosmossdk.io/core/header" | ||
| "cosmossdk.io/core/store" | ||
| ) | ||
|
|
||
| var ( | ||
| _ store.KVStoreService = (*GenesisKVStoreService)(nil) | ||
| _ header.Service = (*GenesisHeaderService)(nil) | ||
| ) | ||
|
|
||
| type genesisContextKeyType struct{} | ||
|
|
||
| var genesisContextKey = genesisContextKeyType{} | ||
|
|
||
| // genesisContext is a context that is used during genesis initialization. | ||
| // it backs the store.KVStoreService and header.Service interface implementations | ||
| // defined in this file. | ||
| type genesisContext struct { | ||
| state store.WriterMap | ||
| } | ||
|
|
||
| // NewGenesisContext creates a new genesis context. | ||
| func NewGenesisContext(state store.WriterMap) genesisContext { | ||
| return genesisContext{ | ||
| state: state, | ||
| } | ||
| } | ||
|
|
||
| // Run runs the provided function within the genesis context and returns an | ||
| // updated store.WriterMap containing the state modifications made during InitGenesis. | ||
| func (g *genesisContext) Run( | ||
| ctx context.Context, | ||
| fn func(ctx context.Context) error, | ||
| ) (store.WriterMap, error) { | ||
| ctx = context.WithValue(ctx, genesisContextKey, g) | ||
| err := fn(ctx) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return g.state, nil | ||
| } | ||
|
|
||
| // GenesisKVStoreService is a store.KVStoreService implementation that is used during | ||
| // genesis initialization. It wraps an inner execution context store.KVStoreService. | ||
| type GenesisKVStoreService struct { | ||
| actor []byte | ||
| executionService store.KVStoreService | ||
| } | ||
|
|
||
| // NewGenesisKVService creates a new GenesisKVStoreService. | ||
| // - actor is the module store key. | ||
| // - executionService is the store.KVStoreService to use when the genesis context is not active. | ||
| func NewGenesisKVService( | ||
| actor []byte, | ||
| executionService store.KVStoreService, | ||
| ) *GenesisKVStoreService { | ||
| return &GenesisKVStoreService{ | ||
| actor: actor, | ||
| executionService: executionService, | ||
| } | ||
| } | ||
|
|
||
| // OpenKVStore implements store.KVStoreService. | ||
| func (g *GenesisKVStoreService) OpenKVStore(ctx context.Context) store.KVStore { | ||
| v := ctx.Value(genesisContextKey) | ||
| if v == nil { | ||
| return g.executionService.OpenKVStore(ctx) | ||
| } | ||
| genCtx, ok := v.(*genesisContext) | ||
| if !ok { | ||
| panic(fmt.Errorf("unexpected genesis context type: %T", v)) | ||
| } | ||
| state, err := genCtx.state.GetWriter(g.actor) | ||
| if err != nil { | ||
| panic(err) | ||
| } | ||
| return state | ||
| } | ||
|
|
||
| // GenesisHeaderService is a header.Service implementation that is used during | ||
| // genesis initialization. It wraps an inner execution context header.Service. | ||
| type GenesisHeaderService struct { | ||
| executionService header.Service | ||
| } | ||
|
|
||
| // HeaderInfo implements header.Service. | ||
| // During genesis initialization, it returns an empty header.Info. | ||
| func (g *GenesisHeaderService) HeaderInfo(ctx context.Context) header.Info { | ||
| v := ctx.Value(genesisContextKey) | ||
| if v == nil { | ||
| return g.executionService.HeaderInfo(ctx) | ||
| } | ||
| return header.Info{} | ||
| } | ||
|
|
||
| // NewGenesisHeaderService creates a new GenesisHeaderService. | ||
| // - executionService is the header.Service to use when the genesis context is not active. | ||
| func NewGenesisHeaderService(executionService header.Service) *GenesisHeaderService { | ||
| return &GenesisHeaderService{ | ||
| executionService: executionService, | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.