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.
85 lines
2.1 KiB
85 lines
2.1 KiB
package counter
|
|
|
|
import (
|
|
"encoding/binary"
|
|
|
|
. "github.com/tendermint/go-common"
|
|
"github.com/tendermint/tmsp/types"
|
|
)
|
|
|
|
type CounterApplication struct {
|
|
hashCount int
|
|
txCount int
|
|
serial bool
|
|
}
|
|
|
|
func NewCounterApplication(serial bool) *CounterApplication {
|
|
return &CounterApplication{serial: serial}
|
|
}
|
|
|
|
func (app *CounterApplication) Info() types.ResponseInfo {
|
|
return types.ResponseInfo{Data: Fmt("{\"hashes\":%v,\"txs\":%v}", app.hashCount, app.txCount)}
|
|
}
|
|
|
|
func (app *CounterApplication) SetOption(key string, value string) (log string) {
|
|
if key == "serial" && value == "on" {
|
|
app.serial = true
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (app *CounterApplication) AppendTx(tx []byte) types.Result {
|
|
if app.serial {
|
|
if len(tx) > 8 {
|
|
return types.ErrEncodingError.SetLog(Fmt("Max tx size is 8 bytes, got %d", len(tx)))
|
|
}
|
|
tx8 := make([]byte, 8)
|
|
copy(tx8[len(tx8)-len(tx):], tx)
|
|
txValue := binary.BigEndian.Uint64(tx8)
|
|
if txValue != uint64(app.txCount) {
|
|
return types.ErrBadNonce.SetLog(Fmt("Invalid nonce. Expected %v, got %v", app.txCount, txValue))
|
|
}
|
|
}
|
|
app.txCount += 1
|
|
return types.OK
|
|
}
|
|
|
|
func (app *CounterApplication) CheckTx(tx []byte) types.Result {
|
|
if app.serial {
|
|
if len(tx) > 8 {
|
|
return types.ErrEncodingError.SetLog(Fmt("Max tx size is 8 bytes, got %d", len(tx)))
|
|
}
|
|
tx8 := make([]byte, 8)
|
|
copy(tx8[len(tx8)-len(tx):], tx)
|
|
txValue := binary.BigEndian.Uint64(tx8)
|
|
if txValue < uint64(app.txCount) {
|
|
return types.ErrBadNonce.SetLog(Fmt("Invalid nonce. Expected >= %v, got %v", app.txCount, txValue))
|
|
}
|
|
}
|
|
return types.OK
|
|
}
|
|
|
|
func (app *CounterApplication) Commit() types.Result {
|
|
app.hashCount += 1
|
|
|
|
if app.txCount == 0 {
|
|
return types.OK
|
|
} else {
|
|
hash := make([]byte, 8)
|
|
binary.BigEndian.PutUint64(hash, uint64(app.txCount))
|
|
return types.NewResultOK(hash, "")
|
|
}
|
|
}
|
|
|
|
func (app *CounterApplication) Query(query []byte) types.Result {
|
|
queryStr := string(query)
|
|
|
|
switch queryStr {
|
|
case "hash":
|
|
return types.NewResultOK(nil, Fmt("%v", app.hashCount))
|
|
case "tx":
|
|
return types.NewResultOK(nil, Fmt("%v", app.txCount))
|
|
}
|
|
|
|
return types.ErrUnknownRequest.SetLog(Fmt("Invalid nonce. Expected hash or tx, got %v", queryStr))
|
|
}
|