-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1456 from mesg-foundation/feature/runners
Update Instance SDK and create Runner SDK
- Loading branch information
Showing
37 changed files
with
2,009 additions
and
714 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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,119 +1,93 @@ | ||
| package database | ||
|
|
||
| import ( | ||
| "encoding/json" | ||
| "errors" | ||
| "fmt" | ||
|
|
||
| "github.com/cosmos/cosmos-sdk/codec" | ||
| "github.com/mesg-foundation/engine/database/store" | ||
| "github.com/mesg-foundation/engine/hash" | ||
| "github.com/mesg-foundation/engine/instance" | ||
| "github.com/syndtr/goleveldb/leveldb" | ||
| ) | ||
|
|
||
| var ( | ||
| errSaveInstanceWithoutHash = errors.New("database: can't save instance without hash") | ||
| errCannotSaveInstanceWithoutHash = errors.New("database: can't save instance without hash") | ||
| ) | ||
|
|
||
| // InstanceDB describes the API of Instance database. | ||
| type InstanceDB interface { | ||
| // Exist check if instance with given hash exist. | ||
| Exist(hash hash.Hash) (bool, error) | ||
|
|
||
| // Get retrives instance by instance hash. | ||
| Get(hash hash.Hash) (*instance.Instance, error) | ||
|
|
||
| // GetAll retrieves all instances. | ||
| GetAll() ([]*instance.Instance, error) | ||
|
|
||
| // Save saves instance to database. | ||
| Save(i *instance.Instance) error | ||
|
|
||
| // Delete an instance by instance hash. | ||
| Delete(hash hash.Hash) error | ||
|
|
||
| // Close closes underlying database connection. | ||
| Close() error | ||
| } | ||
|
|
||
| // LevelDBInstanceDB is a database for storing services' instances. | ||
| type LevelDBInstanceDB struct { | ||
| db *leveldb.DB | ||
| // InstanceDB is a database for storing instance definition. | ||
| type InstanceDB struct { | ||
| s store.Store | ||
| cdc *codec.Codec | ||
| } | ||
|
|
||
| // NewInstanceDB returns the database which is located under given path. | ||
| func NewInstanceDB(path string) (*LevelDBInstanceDB, error) { | ||
| db, err := leveldb.OpenFile(path, nil) | ||
| if err != nil { | ||
| return nil, err | ||
| func NewInstanceDB(s store.Store, cdc *codec.Codec) *InstanceDB { | ||
| return &InstanceDB{ | ||
| s: s, | ||
| cdc: cdc, | ||
| } | ||
| return &LevelDBInstanceDB{db: db}, nil | ||
| } | ||
|
|
||
| // marshal returns the byte slice from service. | ||
| func (d *LevelDBInstanceDB) marshal(i *instance.Instance) ([]byte, error) { | ||
| return json.Marshal(i) | ||
| } | ||
|
|
||
| // unmarshal returns the service from byte slice. | ||
| func (d *LevelDBInstanceDB) unmarshal(hash hash.Hash, value []byte) (*instance.Instance, error) { | ||
| // unmarshal returns the instance from byte slice. | ||
| func (d *InstanceDB) unmarshalInstance(hash hash.Hash, value []byte) (*instance.Instance, error) { | ||
| var s instance.Instance | ||
| if err := json.Unmarshal(value, &s); err != nil { | ||
| return nil, fmt.Errorf("database: could not decode instance %q: %s", hash, err) | ||
| if err := d.cdc.UnmarshalBinaryBare(value, &s); err != nil { | ||
| return nil, fmt.Errorf("database: could not decode instance %q: %w", hash.String(), err) | ||
| } | ||
| return &s, nil | ||
| } | ||
|
|
||
| // Exist check if instance with given hash exist. | ||
| func (d *LevelDBInstanceDB) Exist(hash hash.Hash) (bool, error) { | ||
| return d.db.Has(hash, nil) | ||
| } | ||
|
|
||
| // Get retrives instance by instance hash. | ||
| func (d *LevelDBInstanceDB) Get(hash hash.Hash) (*instance.Instance, error) { | ||
| b, err := d.db.Get(hash, nil) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return d.unmarshal(hash, b) | ||
| } | ||
|
|
||
| // GetAll retrieves all instances. | ||
| func (d *LevelDBInstanceDB) GetAll() ([]*instance.Instance, error) { | ||
| instances := []*instance.Instance{} | ||
| iter := d.db.NewIterator(nil, nil) | ||
| // All returns every instance in database. | ||
| func (d *InstanceDB) All() ([]*instance.Instance, error) { | ||
| var ( | ||
| instances []*instance.Instance | ||
| iter = d.s.NewIterator() | ||
| ) | ||
| for iter.Next() { | ||
| i, err := d.unmarshal(iter.Key(), iter.Value()) | ||
| hash := hash.Hash(iter.Key()) | ||
| s, err := d.unmarshalInstance(hash, iter.Value()) | ||
| if err != nil { | ||
| iter.Release() | ||
| return nil, err | ||
| } | ||
| instances = append(instances, i) | ||
| instances = append(instances, s) | ||
| } | ||
| iter.Release() | ||
| return instances, iter.Error() | ||
| } | ||
|
|
||
| // Save saves instance to database. | ||
| func (d *LevelDBInstanceDB) Save(i *instance.Instance) error { | ||
| if i.Hash.IsZero() { | ||
| return errSaveInstanceWithoutHash | ||
| // Save stores instance in database. | ||
| // If there is an another instance that uses the same sid, it'll be deleted. | ||
| func (d *InstanceDB) Save(r *instance.Instance) error { | ||
| if r.Hash.IsZero() { | ||
| return errCannotSaveInstanceWithoutHash | ||
| } | ||
|
|
||
| // encode service | ||
| b, err := d.marshal(i) | ||
| b, err := d.cdc.MarshalBinaryBare(r) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return d.db.Put(i.Hash, b, nil) | ||
| return d.s.Put(r.Hash, b) | ||
| } | ||
|
|
||
| // Close closes database. | ||
| func (d *LevelDBInstanceDB) Close() error { | ||
| return d.db.Close() | ||
| func (d *InstanceDB) Close() error { | ||
| return d.s.Close() | ||
| } | ||
|
|
||
| // Exist check if instance with given hash exist. | ||
| func (d *InstanceDB) Exist(hash hash.Hash) (bool, error) { | ||
| return d.s.Has(hash) | ||
| } | ||
|
|
||
| // Get retrives instance from database. | ||
| func (d *InstanceDB) Get(hash hash.Hash) (*instance.Instance, error) { | ||
| b, err := d.s.Get(hash) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return d.unmarshalInstance(hash, b) | ||
| } | ||
|
|
||
| // Delete deletes an instance from database. | ||
| func (d *LevelDBInstanceDB) Delete(hash hash.Hash) error { | ||
| return d.db.Delete(hash, nil) | ||
| // Delete deletes instance from database. | ||
| func (d *InstanceDB) Delete(hash hash.Hash) error { | ||
| return d.s.Delete(hash) | ||
| } |
Oops, something went wrong.