Feature/api batchesfeature/sql-semaphore1
@ -0,0 +1,116 @@ |
|||||
|
package api |
||||
|
|
||||
|
import ( |
||||
|
"errors" |
||||
|
"net/http" |
||||
|
|
||||
|
"github.com/gin-gonic/gin" |
||||
|
"github.com/hermeznetwork/hermez-node/common" |
||||
|
"github.com/hermeznetwork/hermez-node/db" |
||||
|
"github.com/hermeznetwork/hermez-node/db/historydb" |
||||
|
) |
||||
|
|
||||
|
func getBatches(c *gin.Context) { |
||||
|
// Get query parameters
|
||||
|
// minBatchNum
|
||||
|
minBatchNum, err := parseQueryUint("minBatchNum", nil, 0, maxUint32, c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// maxBatchNum
|
||||
|
maxBatchNum, err := parseQueryUint("maxBatchNum", nil, 0, maxUint32, c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// slotNum
|
||||
|
slotNum, err := parseQueryUint("slotNum", nil, 0, maxUint32, c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// forgerAddr
|
||||
|
forgerAddr, err := parseQueryEthAddr("forgerAddr", c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// pagination
|
||||
|
fromItem, order, limit, err := parsePagination(c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// Fetch batches from historyDB
|
||||
|
batches, pagination, err := h.GetBatchesAPI( |
||||
|
minBatchNum, maxBatchNum, slotNum, forgerAddr, fromItem, limit, order, |
||||
|
) |
||||
|
if err != nil { |
||||
|
retSQLErr(err, c) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// Build succesfull response
|
||||
|
type batchesResponse struct { |
||||
|
Batches []historydb.BatchAPI `json:"batches"` |
||||
|
Pagination *db.Pagination `json:"pagination"` |
||||
|
} |
||||
|
c.JSON(http.StatusOK, &batchesResponse{ |
||||
|
Batches: batches, |
||||
|
Pagination: pagination, |
||||
|
}) |
||||
|
} |
||||
|
|
||||
|
func getBatch(c *gin.Context) { |
||||
|
// Get batchNum
|
||||
|
batchNum, err := parseParamUint("batchNum", nil, 0, maxUint32, c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
if batchNum == nil { // batchNum is required
|
||||
|
retBadReq(errors.New("Invalid batchNum"), c) |
||||
|
return |
||||
|
} |
||||
|
// Fetch batch from historyDB
|
||||
|
batch, err := h.GetBatchAPI(common.BatchNum(*batchNum)) |
||||
|
if err != nil { |
||||
|
retSQLErr(err, c) |
||||
|
return |
||||
|
} |
||||
|
// JSON response
|
||||
|
c.JSON(http.StatusOK, batch) |
||||
|
} |
||||
|
|
||||
|
type fullBatch struct { |
||||
|
Batch *historydb.BatchAPI |
||||
|
Txs []historyTxAPI |
||||
|
} |
||||
|
|
||||
|
func getFullBatch(c *gin.Context) { |
||||
|
// Get batchNum
|
||||
|
batchNum, err := parseParamUint("batchNum", nil, 0, maxUint32, c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
if batchNum == nil { |
||||
|
retBadReq(errors.New("Invalid batchNum"), c) |
||||
|
return |
||||
|
} |
||||
|
// Fetch batch from historyDB
|
||||
|
batch, err := h.GetBatchAPI(common.BatchNum(*batchNum)) |
||||
|
if err != nil { |
||||
|
retSQLErr(err, c) |
||||
|
return |
||||
|
} |
||||
|
// Fetch txs from historyDB
|
||||
|
// TODO
|
||||
|
txs := []historyTxAPI{} |
||||
|
// JSON response
|
||||
|
c.JSON(http.StatusOK, fullBatch{ |
||||
|
Batch: batch, |
||||
|
Txs: txs, |
||||
|
}) |
||||
|
} |
@ -0,0 +1,243 @@ |
|||||
|
package api |
||||
|
|
||||
|
import ( |
||||
|
"fmt" |
||||
|
"strconv" |
||||
|
"testing" |
||||
|
"time" |
||||
|
|
||||
|
ethCommon "github.com/ethereum/go-ethereum/common" |
||||
|
"github.com/hermeznetwork/hermez-node/common" |
||||
|
"github.com/hermeznetwork/hermez-node/db" |
||||
|
"github.com/hermeznetwork/hermez-node/db/historydb" |
||||
|
"github.com/mitchellh/copystructure" |
||||
|
"github.com/stretchr/testify/assert" |
||||
|
) |
||||
|
|
||||
|
type testBatch struct { |
||||
|
ItemID int `json:"itemId"` |
||||
|
BatchNum common.BatchNum `json:"batchNum"` |
||||
|
EthBlockNum int64 `json:"ethereumBlockNum"` |
||||
|
EthBlockHash ethCommon.Hash `json:"ethereumBlockHash"` |
||||
|
Timestamp time.Time `json:"timestamp"` |
||||
|
ForgerAddr ethCommon.Address `json:"forgerAddr"` |
||||
|
CollectedFees map[common.TokenID]string `json:"collectedFees"` |
||||
|
TotalFeesUSD *float64 `json:"historicTotalCollectedFeesUSD"` |
||||
|
StateRoot string `json:"stateRoot"` |
||||
|
NumAccounts int `json:"numAccounts"` |
||||
|
ExitRoot string `json:"exitRoot"` |
||||
|
ForgeL1TxsNum *int64 `json:"forgeL1TransactionsNum"` |
||||
|
SlotNum int64 `json:"slotNum"` |
||||
|
} |
||||
|
type testBatchesResponse struct { |
||||
|
Batches []testBatch `json:"batches"` |
||||
|
Pagination *db.Pagination `json:"pagination"` |
||||
|
} |
||||
|
|
||||
|
func (t testBatchesResponse) GetPagination() *db.Pagination { |
||||
|
if t.Batches[0].ItemID < t.Batches[len(t.Batches)-1].ItemID { |
||||
|
t.Pagination.FirstReturnedItem = t.Batches[0].ItemID |
||||
|
t.Pagination.LastReturnedItem = t.Batches[len(t.Batches)-1].ItemID |
||||
|
} else { |
||||
|
t.Pagination.LastReturnedItem = t.Batches[0].ItemID |
||||
|
t.Pagination.FirstReturnedItem = t.Batches[len(t.Batches)-1].ItemID |
||||
|
} |
||||
|
return t.Pagination |
||||
|
} |
||||
|
|
||||
|
func (t testBatchesResponse) Len() int { |
||||
|
return len(t.Batches) |
||||
|
} |
||||
|
|
||||
|
func genTestBatches(blocks []common.Block, cBatches []common.Batch) []testBatch { |
||||
|
tBatches := []testBatch{} |
||||
|
for _, cBatch := range cBatches { |
||||
|
block := common.Block{} |
||||
|
found := false |
||||
|
for _, b := range blocks { |
||||
|
if b.EthBlockNum == cBatch.EthBlockNum { |
||||
|
block = b |
||||
|
found = true |
||||
|
break |
||||
|
} |
||||
|
} |
||||
|
if !found { |
||||
|
panic("block not found") |
||||
|
} |
||||
|
collectedFees := make(map[common.TokenID]string) |
||||
|
for k, v := range cBatch.CollectedFees { |
||||
|
collectedFees[k] = v.String() |
||||
|
} |
||||
|
tBatch := testBatch{ |
||||
|
BatchNum: cBatch.BatchNum, |
||||
|
EthBlockNum: cBatch.EthBlockNum, |
||||
|
EthBlockHash: block.Hash, |
||||
|
Timestamp: block.Timestamp, |
||||
|
ForgerAddr: cBatch.ForgerAddr, |
||||
|
CollectedFees: collectedFees, |
||||
|
TotalFeesUSD: cBatch.TotalFeesUSD, |
||||
|
StateRoot: cBatch.StateRoot.String(), |
||||
|
NumAccounts: cBatch.NumAccounts, |
||||
|
ExitRoot: cBatch.ExitRoot.String(), |
||||
|
ForgeL1TxsNum: cBatch.ForgeL1TxsNum, |
||||
|
SlotNum: cBatch.SlotNum, |
||||
|
} |
||||
|
tBatches = append(tBatches, tBatch) |
||||
|
} |
||||
|
return tBatches |
||||
|
} |
||||
|
|
||||
|
func TestGetBatches(t *testing.T) { |
||||
|
endpoint := apiURL + "batches" |
||||
|
fetchedBatches := []testBatch{} |
||||
|
appendIter := func(intr interface{}) { |
||||
|
for i := 0; i < len(intr.(*testBatchesResponse).Batches); i++ { |
||||
|
tmp, err := copystructure.Copy(intr.(*testBatchesResponse).Batches[i]) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
fetchedBatches = append(fetchedBatches, tmp.(testBatch)) |
||||
|
} |
||||
|
} |
||||
|
// Get all (no filters)
|
||||
|
limit := 3 |
||||
|
path := fmt.Sprintf("%s?limit=%d&fromItem=", endpoint, limit) |
||||
|
err := doGoodReqPaginated(path, historydb.OrderAsc, &testBatchesResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
assertBatches(t, tc.batches, fetchedBatches) |
||||
|
|
||||
|
// minBatchNum
|
||||
|
fetchedBatches = []testBatch{} |
||||
|
limit = 2 |
||||
|
minBatchNum := tc.batches[len(tc.batches)/2].BatchNum |
||||
|
path = fmt.Sprintf("%s?minBatchNum=%d&limit=%d&fromItem=", endpoint, minBatchNum, limit) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testBatchesResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
minBatchNumBatches := []testBatch{} |
||||
|
for i := 0; i < len(tc.batches); i++ { |
||||
|
if tc.batches[i].BatchNum > minBatchNum { |
||||
|
minBatchNumBatches = append(minBatchNumBatches, tc.batches[i]) |
||||
|
} |
||||
|
} |
||||
|
assertBatches(t, minBatchNumBatches, fetchedBatches) |
||||
|
|
||||
|
// maxBatchNum
|
||||
|
fetchedBatches = []testBatch{} |
||||
|
limit = 1 |
||||
|
maxBatchNum := tc.batches[len(tc.batches)/2].BatchNum |
||||
|
path = fmt.Sprintf("%s?maxBatchNum=%d&limit=%d&fromItem=", endpoint, maxBatchNum, limit) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testBatchesResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
maxBatchNumBatches := []testBatch{} |
||||
|
for i := 0; i < len(tc.batches); i++ { |
||||
|
if tc.batches[i].BatchNum < maxBatchNum { |
||||
|
maxBatchNumBatches = append(maxBatchNumBatches, tc.batches[i]) |
||||
|
} |
||||
|
} |
||||
|
assertBatches(t, maxBatchNumBatches, fetchedBatches) |
||||
|
|
||||
|
// slotNum
|
||||
|
fetchedBatches = []testBatch{} |
||||
|
limit = 5 |
||||
|
slotNum := tc.batches[len(tc.batches)/2].SlotNum |
||||
|
path = fmt.Sprintf("%s?slotNum=%d&limit=%d&fromItem=", endpoint, slotNum, limit) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testBatchesResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
slotNumBatches := []testBatch{} |
||||
|
for i := 0; i < len(tc.batches); i++ { |
||||
|
if tc.batches[i].SlotNum == slotNum { |
||||
|
slotNumBatches = append(slotNumBatches, tc.batches[i]) |
||||
|
} |
||||
|
} |
||||
|
assertBatches(t, slotNumBatches, fetchedBatches) |
||||
|
|
||||
|
// forgerAddr
|
||||
|
fetchedBatches = []testBatch{} |
||||
|
limit = 10 |
||||
|
forgerAddr := tc.batches[len(tc.batches)/2].ForgerAddr |
||||
|
path = fmt.Sprintf("%s?forgerAddr=%s&limit=%d&fromItem=", endpoint, forgerAddr.String(), limit) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testBatchesResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
forgerAddrBatches := []testBatch{} |
||||
|
for i := 0; i < len(tc.batches); i++ { |
||||
|
if tc.batches[i].ForgerAddr == forgerAddr { |
||||
|
forgerAddrBatches = append(forgerAddrBatches, tc.batches[i]) |
||||
|
} |
||||
|
} |
||||
|
assertBatches(t, forgerAddrBatches, fetchedBatches) |
||||
|
|
||||
|
// All, in reverse order
|
||||
|
fetchedBatches = []testBatch{} |
||||
|
limit = 6 |
||||
|
path = fmt.Sprintf("%s?limit=%d&fromItem=", endpoint, limit) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderDesc, &testBatchesResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
flippedBatches := []testBatch{} |
||||
|
for i := len(tc.batches) - 1; i >= 0; i-- { |
||||
|
flippedBatches = append(flippedBatches, tc.batches[i]) |
||||
|
} |
||||
|
assertBatches(t, flippedBatches, fetchedBatches) |
||||
|
|
||||
|
// Mixed filters
|
||||
|
fetchedBatches = []testBatch{} |
||||
|
limit = 1 |
||||
|
maxBatchNum = tc.batches[len(tc.batches)-len(tc.batches)/4].BatchNum |
||||
|
minBatchNum = tc.batches[len(tc.batches)/4].BatchNum |
||||
|
path = fmt.Sprintf("%s?minBatchNum=%d&maxBatchNum=%d&limit=%d&fromItem=", endpoint, minBatchNum, maxBatchNum, limit) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testBatchesResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
minMaxBatchNumBatches := []testBatch{} |
||||
|
for i := 0; i < len(tc.batches); i++ { |
||||
|
if tc.batches[i].BatchNum < maxBatchNum && tc.batches[i].BatchNum > minBatchNum { |
||||
|
minMaxBatchNumBatches = append(minMaxBatchNumBatches, tc.batches[i]) |
||||
|
} |
||||
|
} |
||||
|
assertBatches(t, minMaxBatchNumBatches, fetchedBatches) |
||||
|
|
||||
|
// 400
|
||||
|
// Invalid minBatchNum
|
||||
|
path = fmt.Sprintf("%s?minBatchNum=%d", endpoint, -2) |
||||
|
err = doBadReq("GET", path, nil, 400) |
||||
|
assert.NoError(t, err) |
||||
|
// Invalid forgerAddr
|
||||
|
path = fmt.Sprintf("%s?forgerAddr=%s", endpoint, "0xG0000001") |
||||
|
err = doBadReq("GET", path, nil, 400) |
||||
|
assert.NoError(t, err) |
||||
|
// 404
|
||||
|
path = fmt.Sprintf("%s?slotNum=%d&minBatchNum=%d", endpoint, 1, 25) |
||||
|
err = doBadReq("GET", path, nil, 404) |
||||
|
assert.NoError(t, err) |
||||
|
} |
||||
|
|
||||
|
func TestGetBatch(t *testing.T) { |
||||
|
endpoint := apiURL + "batches/" |
||||
|
for _, batch := range tc.batches { |
||||
|
fetchedBatch := testBatch{} |
||||
|
assert.NoError( |
||||
|
t, doGoodReq( |
||||
|
"GET", |
||||
|
endpoint+strconv.Itoa(int(batch.BatchNum)), |
||||
|
nil, &fetchedBatch, |
||||
|
), |
||||
|
) |
||||
|
assertBatch(t, batch, fetchedBatch) |
||||
|
} |
||||
|
// 400
|
||||
|
assert.NoError(t, doBadReq("GET", endpoint+"foo", nil, 400)) |
||||
|
// 404
|
||||
|
assert.NoError(t, doBadReq("GET", endpoint+"99999", nil, 404)) |
||||
|
} |
||||
|
|
||||
|
func assertBatches(t *testing.T, expected, actual []testBatch) { |
||||
|
assert.Equal(t, len(expected), len(actual)) |
||||
|
for i := 0; i < len(expected); i++ { |
||||
|
assertBatch(t, expected[i], actual[i]) |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
func assertBatch(t *testing.T, expected, actual testBatch) { |
||||
|
assert.Equal(t, expected.Timestamp.Unix(), actual.Timestamp.Unix()) |
||||
|
expected.Timestamp = actual.Timestamp |
||||
|
actual.ItemID = expected.ItemID |
||||
|
assert.Equal(t, expected, actual) |
||||
|
} |
@ -0,0 +1,50 @@ |
|||||
|
package api |
||||
|
|
||||
|
import ( |
||||
|
"net/http" |
||||
|
|
||||
|
"github.com/gin-gonic/gin" |
||||
|
) |
||||
|
|
||||
|
func getHistoryTxs(c *gin.Context) { |
||||
|
// Get query parameters
|
||||
|
tokenID, addr, bjj, idx, err := parseAccountFilters(c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// BatchNum
|
||||
|
batchNum, err := parseQueryUint("batchNum", nil, 0, maxUint32, c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// TxType
|
||||
|
txType, err := parseQueryTxType(c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// Pagination
|
||||
|
fromItem, order, limit, err := parsePagination(c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// Fetch txs from historyDB
|
||||
|
txs, pagination, err := h.GetHistoryTxs( |
||||
|
addr, bjj, tokenID, idx, batchNum, txType, fromItem, limit, order, |
||||
|
) |
||||
|
if err != nil { |
||||
|
retSQLErr(err, c) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// Build succesfull response
|
||||
|
apiTxs := historyTxsToAPI(txs) |
||||
|
c.JSON(http.StatusOK, &historyTxsAPI{ |
||||
|
Txs: apiTxs, |
||||
|
Pagination: pagination, |
||||
|
}) |
||||
|
} |
@ -0,0 +1,195 @@ |
|||||
|
package apitypes |
||||
|
|
||||
|
import ( |
||||
|
"database/sql/driver" |
||||
|
"encoding/base64" |
||||
|
"encoding/json" |
||||
|
"errors" |
||||
|
"fmt" |
||||
|
"math/big" |
||||
|
"strings" |
||||
|
|
||||
|
ethCommon "github.com/ethereum/go-ethereum/common" |
||||
|
"github.com/hermeznetwork/hermez-node/common" |
||||
|
"github.com/iden3/go-iden3-crypto/babyjub" |
||||
|
) |
||||
|
|
||||
|
// BigIntStr is used to scan/value *big.Int directly into strings from/to sql DBs.
|
||||
|
// It assumes that *big.Int are inserted/fetched to/from the DB using the BigIntMeddler meddler
|
||||
|
// defined at github.com/hermeznetwork/hermez-node/db
|
||||
|
type BigIntStr string |
||||
|
|
||||
|
// NewBigIntStr creates a *BigIntStr from a *big.Int.
|
||||
|
// If the provided bigInt is nil the returned *BigIntStr will also be nil
|
||||
|
func NewBigIntStr(bigInt *big.Int) *BigIntStr { |
||||
|
if bigInt == nil { |
||||
|
return nil |
||||
|
} |
||||
|
bigIntStr := BigIntStr(bigInt.String()) |
||||
|
return &bigIntStr |
||||
|
} |
||||
|
|
||||
|
// Scan implements Scanner for database/sql
|
||||
|
func (b *BigIntStr) Scan(src interface{}) error { |
||||
|
// decode base64 src
|
||||
|
var decoded []byte |
||||
|
var err error |
||||
|
if srcStr, ok := src.(string); ok { |
||||
|
// src is a string
|
||||
|
decoded, err = base64.StdEncoding.DecodeString(srcStr) |
||||
|
} else if srcBytes, ok := src.([]byte); ok { |
||||
|
// src is []byte
|
||||
|
decoded, err = base64.StdEncoding.DecodeString(string(srcBytes)) |
||||
|
} else { |
||||
|
// unexpected src
|
||||
|
return fmt.Errorf("can't scan %T into apitypes.BigIntStr", src) |
||||
|
} |
||||
|
if err != nil { |
||||
|
return err |
||||
|
} |
||||
|
// decoded bytes to *big.Int
|
||||
|
bigInt := &big.Int{} |
||||
|
bigInt = bigInt.SetBytes(decoded) |
||||
|
// *big.Int to BigIntStr
|
||||
|
bigIntStr := NewBigIntStr(bigInt) |
||||
|
if bigIntStr == nil { |
||||
|
return nil |
||||
|
} |
||||
|
*b = *bigIntStr |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
// Value implements valuer for database/sql
|
||||
|
func (b BigIntStr) Value() (driver.Value, error) { |
||||
|
// string to *big.Int
|
||||
|
bigInt := &big.Int{} |
||||
|
bigInt, ok := bigInt.SetString(string(b), 10) |
||||
|
if !ok || bigInt == nil { |
||||
|
return nil, errors.New("invalid representation of a *big.Int") |
||||
|
} |
||||
|
// *big.Int to base64
|
||||
|
return base64.StdEncoding.EncodeToString(bigInt.Bytes()), nil |
||||
|
} |
||||
|
|
||||
|
// CollectedFees is used to retrieve common.batch.CollectedFee from the DB
|
||||
|
type CollectedFees map[common.TokenID]BigIntStr |
||||
|
|
||||
|
// UnmarshalJSON unmarshals a json representation of map[common.TokenID]*big.Int
|
||||
|
func (c *CollectedFees) UnmarshalJSON(text []byte) error { |
||||
|
bigIntMap := make(map[common.TokenID]*big.Int) |
||||
|
if err := json.Unmarshal(text, &bigIntMap); err != nil { |
||||
|
return err |
||||
|
} |
||||
|
bStrMap := make(map[common.TokenID]BigIntStr) |
||||
|
for k, v := range bigIntMap { |
||||
|
bStr := NewBigIntStr(v) |
||||
|
bStrMap[k] = *bStr |
||||
|
} |
||||
|
*c = CollectedFees(bStrMap) |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
// HezEthAddr is used to scan/value Ethereum Address directly into strings that follow the Ethereum address hez fotmat (^hez:0x[a-fA-F0-9]{40}$) from/to sql DBs.
|
||||
|
// It assumes that Ethereum Address are inserted/fetched to/from the DB using the default Scan/Value interface
|
||||
|
type HezEthAddr string |
||||
|
|
||||
|
// NewHezEthAddr creates a HezEthAddr from an Ethereum addr
|
||||
|
func NewHezEthAddr(addr ethCommon.Address) HezEthAddr { |
||||
|
return HezEthAddr("hez:" + addr.String()) |
||||
|
} |
||||
|
|
||||
|
// ToEthAddr returns an Ethereum Address created from HezEthAddr
|
||||
|
func (a HezEthAddr) ToEthAddr() (ethCommon.Address, error) { |
||||
|
addrStr := strings.TrimPrefix(string(a), "hez:") |
||||
|
var addr ethCommon.Address |
||||
|
return addr, addr.UnmarshalText([]byte(addrStr)) |
||||
|
} |
||||
|
|
||||
|
// Scan implements Scanner for database/sql
|
||||
|
func (a *HezEthAddr) Scan(src interface{}) error { |
||||
|
ethAddr := ðCommon.Address{} |
||||
|
if err := ethAddr.Scan(src); err != nil { |
||||
|
return err |
||||
|
} |
||||
|
if ethAddr == nil { |
||||
|
return nil |
||||
|
} |
||||
|
*a = NewHezEthAddr(*ethAddr) |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
// Value implements valuer for database/sql
|
||||
|
func (a HezEthAddr) Value() (driver.Value, error) { |
||||
|
ethAddr, err := a.ToEthAddr() |
||||
|
if err != nil { |
||||
|
return nil, err |
||||
|
} |
||||
|
return ethAddr.Value() |
||||
|
} |
||||
|
|
||||
|
// HezBJJ is used to scan/value *babyjub.PublicKey directly into strings that follow the BJJ public key hez fotmat (^hez:[A-Za-z0-9_-]{44}$) from/to sql DBs.
|
||||
|
// It assumes that *babyjub.PublicKey are inserted/fetched to/from the DB using the default Scan/Value interface
|
||||
|
type HezBJJ string |
||||
|
|
||||
|
// NewHezBJJ creates a HezBJJ from a *babyjub.PublicKey.
|
||||
|
// Calling this method with a nil bjj causes panic
|
||||
|
func NewHezBJJ(bjj *babyjub.PublicKey) HezBJJ { |
||||
|
pkComp := [32]byte(bjj.Compress()) |
||||
|
sum := pkComp[0] |
||||
|
for i := 1; i < len(pkComp); i++ { |
||||
|
sum += pkComp[i] |
||||
|
} |
||||
|
bjjSum := append(pkComp[:], sum) |
||||
|
return HezBJJ("hez:" + base64.RawURLEncoding.EncodeToString(bjjSum)) |
||||
|
} |
||||
|
|
||||
|
// ToBJJ returns a *babyjub.PublicKey created from HezBJJ
|
||||
|
func (b HezBJJ) ToBJJ() (*babyjub.PublicKey, error) { |
||||
|
const decodedLen = 33 |
||||
|
const encodedLen = 44 |
||||
|
formatErr := errors.New("invalid BJJ format. Must follow this regex: ^hez:[A-Za-z0-9_-]{44}$") |
||||
|
encoded := strings.TrimPrefix(string(b), "hez:") |
||||
|
if len(encoded) != encodedLen { |
||||
|
return nil, formatErr |
||||
|
} |
||||
|
decoded, err := base64.RawURLEncoding.DecodeString(encoded) |
||||
|
if err != nil { |
||||
|
return nil, formatErr |
||||
|
} |
||||
|
if len(decoded) != decodedLen { |
||||
|
return nil, formatErr |
||||
|
} |
||||
|
bjjBytes := [decodedLen - 1]byte{} |
||||
|
copy(bjjBytes[:decodedLen-1], decoded[:decodedLen-1]) |
||||
|
sum := bjjBytes[0] |
||||
|
for i := 1; i < len(bjjBytes); i++ { |
||||
|
sum += bjjBytes[i] |
||||
|
} |
||||
|
if decoded[decodedLen-1] != sum { |
||||
|
return nil, errors.New("checksum verification failed") |
||||
|
} |
||||
|
bjjComp := babyjub.PublicKeyComp(bjjBytes) |
||||
|
return bjjComp.Decompress() |
||||
|
} |
||||
|
|
||||
|
// Scan implements Scanner for database/sql
|
||||
|
func (b *HezBJJ) Scan(src interface{}) error { |
||||
|
bjj := &babyjub.PublicKey{} |
||||
|
if err := bjj.Scan(src); err != nil { |
||||
|
return err |
||||
|
} |
||||
|
if bjj == nil { |
||||
|
return nil |
||||
|
} |
||||
|
*b = NewHezBJJ(bjj) |
||||
|
return nil |
||||
|
} |
||||
|
|
||||
|
// Value implements valuer for database/sql
|
||||
|
func (b HezBJJ) Value() (driver.Value, error) { |
||||
|
bjj, err := b.ToBJJ() |
||||
|
if err != nil { |
||||
|
return nil, err |
||||
|
} |
||||
|
return bjj.Value() |
||||
|
} |
@ -0,0 +1,285 @@ |
|||||
|
package apitypes |
||||
|
|
||||
|
import ( |
||||
|
"database/sql" |
||||
|
"io/ioutil" |
||||
|
"math/big" |
||||
|
"os" |
||||
|
"testing" |
||||
|
|
||||
|
ethCommon "github.com/ethereum/go-ethereum/common" |
||||
|
dbUtils "github.com/hermeznetwork/hermez-node/db" |
||||
|
"github.com/iden3/go-iden3-crypto/babyjub" |
||||
|
_ "github.com/mattn/go-sqlite3" //nolint sqlite driver
|
||||
|
"github.com/russross/meddler" |
||||
|
"github.com/stretchr/testify/assert" |
||||
|
) |
||||
|
|
||||
|
var db *sql.DB |
||||
|
|
||||
|
func TestMain(m *testing.M) { |
||||
|
// Register meddler
|
||||
|
meddler.Default = meddler.SQLite |
||||
|
meddler.Register("bigint", dbUtils.BigIntMeddler{}) |
||||
|
meddler.Register("bigintnull", dbUtils.BigIntNullMeddler{}) |
||||
|
// Create temporary sqlite DB
|
||||
|
dir, err := ioutil.TempDir("", "db") |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
db, err = sql.Open("sqlite3", dir+"sqlite.db") |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
defer os.RemoveAll(dir) //nolint
|
||||
|
schema := `CREATE TABLE test (i BLOB);` |
||||
|
if _, err := db.Exec(schema); err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
// Run tests
|
||||
|
result := m.Run() |
||||
|
os.Exit(result) |
||||
|
} |
||||
|
|
||||
|
func TestBigIntStrScannerValuer(t *testing.T) { |
||||
|
// Clean DB
|
||||
|
_, err := db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Example structs
|
||||
|
type bigInMeddlerStruct struct { |
||||
|
I *big.Int `meddler:"i,bigint"` // note the bigint that instructs meddler to use BigIntMeddler
|
||||
|
} |
||||
|
type bigIntStrStruct struct { |
||||
|
I BigIntStr `meddler:"i"` // note that no meddler is specified, and Scan/Value will be used
|
||||
|
} |
||||
|
type bigInMeddlerStructNil struct { |
||||
|
I *big.Int `meddler:"i,bigintnull"` // note the bigint that instructs meddler to use BigIntNullMeddler
|
||||
|
} |
||||
|
type bigIntStrStructNil struct { |
||||
|
I *BigIntStr `meddler:"i"` // note that no meddler is specified, and Scan/Value will be used
|
||||
|
} |
||||
|
|
||||
|
// Not nil case
|
||||
|
// Insert into DB using meddler
|
||||
|
const x = int64(12345) |
||||
|
fromMeddler := bigInMeddlerStruct{ |
||||
|
I: big.NewInt(x), |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromMeddler) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using BigIntStr
|
||||
|
toBigIntStr := bigIntStrStruct{} |
||||
|
err = meddler.QueryRow(db, &toBigIntStr, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Equal(t, fromMeddler.I.String(), string(toBigIntStr.I)) |
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using BigIntStr
|
||||
|
fromBigIntStr := bigIntStrStruct{ |
||||
|
I: "54321", |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromBigIntStr) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using meddler
|
||||
|
toMeddler := bigInMeddlerStruct{} |
||||
|
err = meddler.QueryRow(db, &toMeddler, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Equal(t, string(fromBigIntStr.I), toMeddler.I.String()) |
||||
|
|
||||
|
// Nil case
|
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using meddler
|
||||
|
fromMeddlerNil := bigInMeddlerStructNil{ |
||||
|
I: nil, |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromMeddlerNil) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using BigIntStr
|
||||
|
foo := BigIntStr("foo") |
||||
|
toBigIntStrNil := bigIntStrStructNil{ |
||||
|
I: &foo, // check that this will be set to nil, not because of not being initialized
|
||||
|
} |
||||
|
err = meddler.QueryRow(db, &toBigIntStrNil, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Nil(t, toBigIntStrNil.I) |
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using BigIntStr
|
||||
|
fromBigIntStrNil := bigIntStrStructNil{ |
||||
|
I: nil, |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromBigIntStrNil) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using meddler
|
||||
|
toMeddlerNil := bigInMeddlerStructNil{ |
||||
|
I: big.NewInt(x), // check that this will be set to nil, not because of not being initialized
|
||||
|
} |
||||
|
err = meddler.QueryRow(db, &toMeddlerNil, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Nil(t, toMeddlerNil.I) |
||||
|
} |
||||
|
|
||||
|
func TestHezEthAddr(t *testing.T) { |
||||
|
// Clean DB
|
||||
|
_, err := db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Example structs
|
||||
|
type ethAddrStruct struct { |
||||
|
I ethCommon.Address `meddler:"i"` |
||||
|
} |
||||
|
type hezEthAddrStruct struct { |
||||
|
I HezEthAddr `meddler:"i"` |
||||
|
} |
||||
|
type ethAddrStructNil struct { |
||||
|
I *ethCommon.Address `meddler:"i"` |
||||
|
} |
||||
|
type hezEthAddrStructNil struct { |
||||
|
I *HezEthAddr `meddler:"i"` |
||||
|
} |
||||
|
|
||||
|
// Not nil case
|
||||
|
// Insert into DB using ethCommon.Address Scan/Value
|
||||
|
fromEth := ethAddrStruct{ |
||||
|
I: ethCommon.BigToAddress(big.NewInt(73737373)), |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromEth) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using HezEthAddr Scan/Value
|
||||
|
toHezEth := hezEthAddrStruct{} |
||||
|
err = meddler.QueryRow(db, &toHezEth, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Equal(t, NewHezEthAddr(fromEth.I), toHezEth.I) |
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using HezEthAddr Scan/Value
|
||||
|
fromHezEth := hezEthAddrStruct{ |
||||
|
I: NewHezEthAddr(ethCommon.BigToAddress(big.NewInt(3786872586))), |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromHezEth) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using ethCommon.Address Scan/Value
|
||||
|
toEth := ethAddrStruct{} |
||||
|
err = meddler.QueryRow(db, &toEth, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Equal(t, fromHezEth.I, NewHezEthAddr(toEth.I)) |
||||
|
|
||||
|
// Nil case
|
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using ethCommon.Address Scan/Value
|
||||
|
fromEthNil := ethAddrStructNil{ |
||||
|
I: nil, |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromEthNil) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using HezEthAddr Scan/Value
|
||||
|
foo := HezEthAddr("foo") |
||||
|
toHezEthNil := hezEthAddrStructNil{ |
||||
|
I: &foo, // check that this will be set to nil, not because of not being initialized
|
||||
|
} |
||||
|
err = meddler.QueryRow(db, &toHezEthNil, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Nil(t, toHezEthNil.I) |
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using HezEthAddr Scan/Value
|
||||
|
fromHezEthNil := hezEthAddrStructNil{ |
||||
|
I: nil, |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromHezEthNil) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using ethCommon.Address Scan/Value
|
||||
|
fooAddr := ethCommon.BigToAddress(big.NewInt(1)) |
||||
|
toEthNil := ethAddrStructNil{ |
||||
|
I: &fooAddr, // check that this will be set to nil, not because of not being initialized
|
||||
|
} |
||||
|
err = meddler.QueryRow(db, &toEthNil, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Nil(t, toEthNil.I) |
||||
|
} |
||||
|
|
||||
|
func TestHezBJJ(t *testing.T) { |
||||
|
// Clean DB
|
||||
|
_, err := db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Example structs
|
||||
|
type bjjStruct struct { |
||||
|
I *babyjub.PublicKey `meddler:"i"` |
||||
|
} |
||||
|
type hezBJJStruct struct { |
||||
|
I HezBJJ `meddler:"i"` |
||||
|
} |
||||
|
type hezBJJStructNil struct { |
||||
|
I *HezBJJ `meddler:"i"` |
||||
|
} |
||||
|
|
||||
|
// Not nil case
|
||||
|
// Insert into DB using *babyjub.PublicKey Scan/Value
|
||||
|
priv := babyjub.NewRandPrivKey() |
||||
|
fromBJJ := bjjStruct{ |
||||
|
I: priv.Public(), |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromBJJ) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using HezBJJ Scan/Value
|
||||
|
toHezBJJ := hezBJJStruct{} |
||||
|
err = meddler.QueryRow(db, &toHezBJJ, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Equal(t, NewHezBJJ(fromBJJ.I), toHezBJJ.I) |
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using HezBJJ Scan/Value
|
||||
|
fromHezBJJ := hezBJJStruct{ |
||||
|
I: NewHezBJJ(priv.Public()), |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromHezBJJ) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using *babyjub.PublicKey Scan/Value
|
||||
|
toBJJ := bjjStruct{} |
||||
|
err = meddler.QueryRow(db, &toBJJ, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Equal(t, fromHezBJJ.I, NewHezBJJ(toBJJ.I)) |
||||
|
|
||||
|
// Nil case
|
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using *babyjub.PublicKey Scan/Value
|
||||
|
fromBJJNil := bjjStruct{ |
||||
|
I: nil, |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromBJJNil) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using HezBJJ Scan/Value
|
||||
|
foo := HezBJJ("foo") |
||||
|
toHezBJJNil := hezBJJStructNil{ |
||||
|
I: &foo, // check that this will be set to nil, not because of not being initialized
|
||||
|
} |
||||
|
err = meddler.QueryRow(db, &toHezBJJNil, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Nil(t, toHezBJJNil.I) |
||||
|
// Clean DB
|
||||
|
_, err = db.Exec("delete from test") |
||||
|
assert.NoError(t, err) |
||||
|
// Insert into DB using HezBJJ Scan/Value
|
||||
|
fromHezBJJNil := hezBJJStructNil{ |
||||
|
I: nil, |
||||
|
} |
||||
|
err = meddler.Insert(db, "test", &fromHezBJJNil) |
||||
|
assert.NoError(t, err) |
||||
|
// Read from DB using *babyjub.PublicKey Scan/Value
|
||||
|
toBJJNil := bjjStruct{ |
||||
|
I: priv.Public(), // check that this will be set to nil, not because of not being initialized
|
||||
|
} |
||||
|
err = meddler.QueryRow(db, &toBJJNil, "select * from test") |
||||
|
assert.NoError(t, err) |
||||
|
assert.Nil(t, toBJJNil.I) |
||||
|
} |