package abcicli import ( "sync" types "github.com/tendermint/tendermint/abci/types" "github.com/tendermint/tendermint/libs/service" ) var _ Client = (*localClient)(nil) // NOTE: use defer to unlock mutex because Application might panic (e.g., in // case of malicious tx or query). It only makes sense for publicly exposed // methods like CheckTx (/broadcast_tx_* RPC endpoint) or Query (/abci_query // RPC endpoint), but defers are used everywhere for the sake of consistency. type localClient struct { service.BaseService mtx *sync.Mutex types.Application Callback } func NewLocalClient(mtx *sync.Mutex, app types.Application) Client { if mtx == nil { mtx = new(sync.Mutex) } cli := &localClient{ mtx: mtx, Application: app, } cli.BaseService = *service.NewBaseService(nil, "localClient", cli) return cli } func (app *localClient) SetResponseCallback(cb Callback) { app.mtx.Lock() app.Callback = cb app.mtx.Unlock() } // 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 { app.mtx.Lock() defer app.mtx.Unlock() return app.callback( types.ToRequestEcho(msg), types.ToResponseEcho(msg), ) } func (app *localClient) InfoAsync(req types.RequestInfo) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.Info(req) return app.callback( types.ToRequestInfo(req), types.ToResponseInfo(res), ) } func (app *localClient) SetOptionAsync(req types.RequestSetOption) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.SetOption(req) return app.callback( types.ToRequestSetOption(req), types.ToResponseSetOption(res), ) } func (app *localClient) DeliverTxAsync(params types.RequestDeliverTx) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.DeliverTx(params) return app.callback( types.ToRequestDeliverTx(params), types.ToResponseDeliverTx(res), ) } func (app *localClient) CheckTxAsync(req types.RequestCheckTx) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.CheckTx(req) return app.callback( types.ToRequestCheckTx(req), types.ToResponseCheckTx(res), ) } func (app *localClient) QueryAsync(req types.RequestQuery) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.Query(req) return app.callback( types.ToRequestQuery(req), types.ToResponseQuery(res), ) } func (app *localClient) CommitAsync() *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.Commit() return app.callback( types.ToRequestCommit(), types.ToResponseCommit(res), ) } func (app *localClient) InitChainAsync(req types.RequestInitChain) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.InitChain(req) return app.callback( types.ToRequestInitChain(req), types.ToResponseInitChain(res), ) } func (app *localClient) BeginBlockAsync(req types.RequestBeginBlock) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.BeginBlock(req) return app.callback( types.ToRequestBeginBlock(req), types.ToResponseBeginBlock(res), ) } func (app *localClient) EndBlockAsync(req types.RequestEndBlock) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.EndBlock(req) return app.callback( types.ToRequestEndBlock(req), types.ToResponseEndBlock(res), ) } func (app *localClient) ListSnapshotsAsync(req types.RequestListSnapshots) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.ListSnapshots(req) return app.callback( types.ToRequestListSnapshots(req), types.ToResponseListSnapshots(res), ) } func (app *localClient) OfferSnapshotAsync(req types.RequestOfferSnapshot) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.OfferSnapshot(req) return app.callback( types.ToRequestOfferSnapshot(req), types.ToResponseOfferSnapshot(res), ) } func (app *localClient) LoadSnapshotChunkAsync(req types.RequestLoadSnapshotChunk) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.LoadSnapshotChunk(req) return app.callback( types.ToRequestLoadSnapshotChunk(req), types.ToResponseLoadSnapshotChunk(res), ) } func (app *localClient) ApplySnapshotChunkAsync(req types.RequestApplySnapshotChunk) *ReqRes { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.ApplySnapshotChunk(req) return app.callback( types.ToRequestApplySnapshotChunk(req), types.ToResponseApplySnapshotChunk(res), ) } //------------------------------------------------------- func (app *localClient) FlushSync() error { return nil } func (app *localClient) EchoSync(msg string) (*types.ResponseEcho, error) { return &types.ResponseEcho{Message: msg}, nil } func (app *localClient) InfoSync(req types.RequestInfo) (*types.ResponseInfo, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.Info(req) return &res, nil } func (app *localClient) SetOptionSync(req types.RequestSetOption) (*types.ResponseSetOption, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.SetOption(req) return &res, nil } func (app *localClient) DeliverTxSync(req types.RequestDeliverTx) (*types.ResponseDeliverTx, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.DeliverTx(req) return &res, nil } func (app *localClient) CheckTxSync(req types.RequestCheckTx) (*types.ResponseCheckTx, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.CheckTx(req) return &res, nil } func (app *localClient) QuerySync(req types.RequestQuery) (*types.ResponseQuery, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.Query(req) return &res, nil } func (app *localClient) CommitSync() (*types.ResponseCommit, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.Commit() return &res, nil } func (app *localClient) InitChainSync(req types.RequestInitChain) (*types.ResponseInitChain, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.InitChain(req) return &res, nil } func (app *localClient) BeginBlockSync(req types.RequestBeginBlock) (*types.ResponseBeginBlock, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.BeginBlock(req) return &res, nil } func (app *localClient) EndBlockSync(req types.RequestEndBlock) (*types.ResponseEndBlock, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.EndBlock(req) return &res, nil } func (app *localClient) ListSnapshotsSync(req types.RequestListSnapshots) (*types.ResponseListSnapshots, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.ListSnapshots(req) return &res, nil } func (app *localClient) OfferSnapshotSync(req types.RequestOfferSnapshot) (*types.ResponseOfferSnapshot, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.OfferSnapshot(req) return &res, nil } func (app *localClient) LoadSnapshotChunkSync( req types.RequestLoadSnapshotChunk) (*types.ResponseLoadSnapshotChunk, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.LoadSnapshotChunk(req) return &res, nil } func (app *localClient) ApplySnapshotChunkSync( req types.RequestApplySnapshotChunk) (*types.ResponseApplySnapshotChunk, error) { app.mtx.Lock() defer app.mtx.Unlock() res := app.Application.ApplySnapshotChunk(req) return &res, 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 }