You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
241 lines
5.4 KiB
241 lines
5.4 KiB
package tmspcli
|
|
|
|
import (
|
|
"sync"
|
|
|
|
. "github.com/tendermint/go-common"
|
|
types "github.com/tendermint/tmsp/types"
|
|
)
|
|
|
|
type localClient struct {
|
|
BaseService
|
|
mtx *sync.Mutex
|
|
types.Application
|
|
Callback
|
|
}
|
|
|
|
func NewLocalClient(mtx *sync.Mutex, app types.Application) *localClient {
|
|
if mtx == nil {
|
|
mtx = new(sync.Mutex)
|
|
}
|
|
cli := &localClient{
|
|
mtx: mtx,
|
|
Application: app,
|
|
}
|
|
cli.BaseService = *NewBaseService(log, "localClient", cli)
|
|
return cli
|
|
}
|
|
|
|
func (app *localClient) SetResponseCallback(cb Callback) {
|
|
app.mtx.Lock()
|
|
defer app.mtx.Unlock()
|
|
app.Callback = cb
|
|
}
|
|
|
|
// TODO: change types.Application to include Error()?
|
|
func (app *localClient) Error() error {
|
|
return nil
|
|
}
|
|
|
|
func (app *localClient) FlushAsync() *ReqRes {
|
|
// Do nothing
|
|
return newLocalReqRes(types.ToRequestFlush(), nil)
|
|
}
|
|
|
|
func (app *localClient) EchoAsync(msg string) *ReqRes {
|
|
return app.callback(
|
|
types.ToRequestEcho(msg),
|
|
types.ToResponseEcho(msg),
|
|
)
|
|
}
|
|
|
|
func (app *localClient) InfoAsync() *ReqRes {
|
|
app.mtx.Lock()
|
|
info := app.Application.Info()
|
|
app.mtx.Unlock()
|
|
return app.callback(
|
|
types.ToRequestInfo(),
|
|
types.ToResponseInfo(info),
|
|
)
|
|
}
|
|
|
|
func (app *localClient) SetOptionAsync(key string, value string) *ReqRes {
|
|
app.mtx.Lock()
|
|
log := app.Application.SetOption(key, value)
|
|
app.mtx.Unlock()
|
|
return app.callback(
|
|
types.ToRequestSetOption(key, value),
|
|
types.ToResponseSetOption(log),
|
|
)
|
|
}
|
|
|
|
func (app *localClient) AppendTxAsync(tx []byte) *ReqRes {
|
|
app.mtx.Lock()
|
|
res := app.Application.AppendTx(tx)
|
|
app.mtx.Unlock()
|
|
return app.callback(
|
|
types.ToRequestAppendTx(tx),
|
|
types.ToResponseAppendTx(res.Code, res.Data, res.Log),
|
|
)
|
|
}
|
|
|
|
func (app *localClient) CheckTxAsync(tx []byte) *ReqRes {
|
|
app.mtx.Lock()
|
|
res := app.Application.CheckTx(tx)
|
|
app.mtx.Unlock()
|
|
return app.callback(
|
|
types.ToRequestCheckTx(tx),
|
|
types.ToResponseCheckTx(res.Code, res.Data, res.Log),
|
|
)
|
|
}
|
|
|
|
func (app *localClient) QueryAsync(tx []byte) *ReqRes {
|
|
app.mtx.Lock()
|
|
res := app.Application.Query(tx)
|
|
app.mtx.Unlock()
|
|
return app.callback(
|
|
types.ToRequestQuery(tx),
|
|
types.ToResponseQuery(res.Code, res.Data, res.Log),
|
|
)
|
|
}
|
|
|
|
func (app *localClient) CommitAsync() *ReqRes {
|
|
app.mtx.Lock()
|
|
res := app.Application.Commit()
|
|
app.mtx.Unlock()
|
|
return app.callback(
|
|
types.ToRequestCommit(),
|
|
types.ToResponseCommit(res.Code, res.Data, res.Log),
|
|
)
|
|
}
|
|
|
|
func (app *localClient) InitChainAsync(validators []*types.Validator) *ReqRes {
|
|
app.mtx.Lock()
|
|
if bcApp, ok := app.Application.(types.BlockchainAware); ok {
|
|
bcApp.InitChain(validators)
|
|
}
|
|
reqRes := app.callback(
|
|
types.ToRequestInitChain(validators),
|
|
types.ToResponseInitChain(),
|
|
)
|
|
app.mtx.Unlock()
|
|
return reqRes
|
|
}
|
|
|
|
func (app *localClient) BeginBlockAsync(height uint64) *ReqRes {
|
|
app.mtx.Lock()
|
|
if bcApp, ok := app.Application.(types.BlockchainAware); ok {
|
|
bcApp.BeginBlock(height)
|
|
}
|
|
app.mtx.Unlock()
|
|
return app.callback(
|
|
types.ToRequestBeginBlock(height),
|
|
types.ToResponseBeginBlock(),
|
|
)
|
|
}
|
|
|
|
func (app *localClient) EndBlockAsync(height uint64) *ReqRes {
|
|
app.mtx.Lock()
|
|
var validators []*types.Validator
|
|
if bcApp, ok := app.Application.(types.BlockchainAware); ok {
|
|
validators = bcApp.EndBlock(height)
|
|
}
|
|
app.mtx.Unlock()
|
|
return app.callback(
|
|
types.ToRequestEndBlock(height),
|
|
types.ToResponseEndBlock(validators),
|
|
)
|
|
}
|
|
|
|
//-------------------------------------------------------
|
|
|
|
func (app *localClient) FlushSync() error {
|
|
return nil
|
|
}
|
|
|
|
func (app *localClient) EchoSync(msg string) (res types.Result) {
|
|
return types.OK.SetData([]byte(msg))
|
|
}
|
|
|
|
func (app *localClient) InfoSync() (res types.Result) {
|
|
app.mtx.Lock()
|
|
info := app.Application.Info()
|
|
app.mtx.Unlock()
|
|
return types.OK.SetData([]byte(info))
|
|
}
|
|
|
|
func (app *localClient) SetOptionSync(key string, value string) (res types.Result) {
|
|
app.mtx.Lock()
|
|
log := app.Application.SetOption(key, value)
|
|
app.mtx.Unlock()
|
|
return types.OK.SetLog(log)
|
|
}
|
|
|
|
func (app *localClient) AppendTxSync(tx []byte) (res types.Result) {
|
|
app.mtx.Lock()
|
|
res = app.Application.AppendTx(tx)
|
|
app.mtx.Unlock()
|
|
return res
|
|
}
|
|
|
|
func (app *localClient) CheckTxSync(tx []byte) (res types.Result) {
|
|
app.mtx.Lock()
|
|
res = app.Application.CheckTx(tx)
|
|
app.mtx.Unlock()
|
|
return res
|
|
}
|
|
|
|
func (app *localClient) QuerySync(query []byte) (res types.Result) {
|
|
app.mtx.Lock()
|
|
res = app.Application.Query(query)
|
|
app.mtx.Unlock()
|
|
return res
|
|
}
|
|
|
|
func (app *localClient) CommitSync() (res types.Result) {
|
|
app.mtx.Lock()
|
|
res = app.Application.Commit()
|
|
app.mtx.Unlock()
|
|
return res
|
|
}
|
|
|
|
func (app *localClient) InitChainSync(validators []*types.Validator) (err error) {
|
|
app.mtx.Lock()
|
|
if bcApp, ok := app.Application.(types.BlockchainAware); ok {
|
|
bcApp.InitChain(validators)
|
|
}
|
|
app.mtx.Unlock()
|
|
return nil
|
|
}
|
|
|
|
func (app *localClient) BeginBlockSync(height uint64) (err error) {
|
|
app.mtx.Lock()
|
|
if bcApp, ok := app.Application.(types.BlockchainAware); ok {
|
|
bcApp.BeginBlock(height)
|
|
}
|
|
app.mtx.Unlock()
|
|
return nil
|
|
}
|
|
|
|
func (app *localClient) EndBlockSync(height uint64) (changedValidators []*types.Validator, err error) {
|
|
app.mtx.Lock()
|
|
if bcApp, ok := app.Application.(types.BlockchainAware); ok {
|
|
changedValidators = bcApp.EndBlock(height)
|
|
}
|
|
app.mtx.Unlock()
|
|
return changedValidators, nil
|
|
}
|
|
|
|
//-------------------------------------------------------
|
|
|
|
func (app *localClient) callback(req *types.Request, res *types.Response) *ReqRes {
|
|
app.Callback(req, res)
|
|
return newLocalReqRes(req, res)
|
|
}
|
|
|
|
func newLocalReqRes(req *types.Request, res *types.Response) *ReqRes {
|
|
reqRes := NewReqRes(req)
|
|
reqRes.Response = res
|
|
reqRes.SetDone()
|
|
return reqRes
|
|
}
|