129 lines
2.9 KiB
Go
129 lines
2.9 KiB
Go
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/ethereum/go-ethereum/eth"
|
||
|
|
"github.com/ethereum/go-ethereum/rpc/codec"
|
||
|
|
"github.com/ethereum/go-ethereum/rpc/shared"
|
||
|
|
"github.com/ethereum/go-ethereum/xeth"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
DbApiversion = "1.0"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
// mapping between methods and handlers
|
||
|
|
DbMapping = map[string]dbhandler{
|
||
|
|
"db_getString": (*dbApi).GetString,
|
||
|
|
"db_putString": (*dbApi).PutString,
|
||
|
|
"db_getHex": (*dbApi).GetHex,
|
||
|
|
"db_putHex": (*dbApi).PutHex,
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
// db callback handler
|
||
|
|
type dbhandler func(*dbApi, *shared.Request) (interface{}, error)
|
||
|
|
|
||
|
|
// db api provider
|
||
|
|
type dbApi struct {
|
||
|
|
xeth *xeth.XEth
|
||
|
|
ethereum *eth.Ethereum
|
||
|
|
methods map[string]dbhandler
|
||
|
|
codec codec.ApiCoder
|
||
|
|
}
|
||
|
|
|
||
|
|
// create a new db api instance
|
||
|
|
func NewDbApi(xeth *xeth.XEth, ethereum *eth.Ethereum, coder codec.Codec) *dbApi {
|
||
|
|
return &dbApi{
|
||
|
|
xeth: xeth,
|
||
|
|
ethereum: ethereum,
|
||
|
|
methods: DbMapping,
|
||
|
|
codec: coder.New(nil),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// collection with supported methods
|
||
|
|
func (self *dbApi) Methods() []string {
|
||
|
|
methods := make([]string, len(self.methods))
|
||
|
|
i := 0
|
||
|
|
for k := range self.methods {
|
||
|
|
methods[i] = k
|
||
|
|
i++
|
||
|
|
}
|
||
|
|
return methods
|
||
|
|
}
|
||
|
|
|
||
|
|
// Execute given request
|
||
|
|
func (self *dbApi) Execute(req *shared.Request) (interface{}, error) {
|
||
|
|
if callback, ok := self.methods[req.Method]; ok {
|
||
|
|
return callback(self, req)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil, &shared.NotImplementedError{req.Method}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (self *dbApi) Name() string {
|
||
|
|
return DbApiName
|
||
|
|
}
|
||
|
|
|
||
|
|
func (self *dbApi) ApiVersion() string {
|
||
|
|
return DbApiversion
|
||
|
|
}
|
||
|
|
|
||
|
|
func (self *dbApi) GetString(req *shared.Request) (interface{}, error) {
|
||
|
|
args := new(DbArgs)
|
||
|
|
if err := self.codec.Decode(req.Params, &args); err != nil {
|
||
|
|
return nil, shared.NewDecodeParamError(err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := args.requirements(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
ret, err := self.xeth.DbGet([]byte(args.Database + args.Key))
|
||
|
|
return string(ret), err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (self *dbApi) PutString(req *shared.Request) (interface{}, error) {
|
||
|
|
args := new(DbArgs)
|
||
|
|
if err := self.codec.Decode(req.Params, &args); err != nil {
|
||
|
|
return nil, shared.NewDecodeParamError(err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := args.requirements(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return self.xeth.DbPut([]byte(args.Database+args.Key), args.Value), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (self *dbApi) GetHex(req *shared.Request) (interface{}, error) {
|
||
|
|
args := new(DbHexArgs)
|
||
|
|
if err := self.codec.Decode(req.Params, &args); err != nil {
|
||
|
|
return nil, shared.NewDecodeParamError(err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := args.requirements(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
if res, err := self.xeth.DbGet([]byte(args.Database + args.Key)); err == nil {
|
||
|
|
return newHexData(res), nil
|
||
|
|
} else {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (self *dbApi) PutHex(req *shared.Request) (interface{}, error) {
|
||
|
|
args := new(DbHexArgs)
|
||
|
|
if err := self.codec.Decode(req.Params, &args); err != nil {
|
||
|
|
return nil, shared.NewDecodeParamError(err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := args.requirements(); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return self.xeth.DbPut([]byte(args.Database+args.Key), args.Value), nil
|
||
|
|
}
|