/
mixbiztable.go
89 lines (78 loc) · 1.91 KB
/
mixbiztable.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package wallet
import (
"fmt"
"github.com/33cn/chain33/common/db"
"github.com/33cn/chain33/common/db/table"
"github.com/33cn/chain33/types"
mix "github.com/33cn/plugin/plugin/dapp/mix/types"
)
/*
table struct
data: self consens stage
index: status
*/
var boardOpt = &table.Option{
Prefix: "LODB-mix",
Name: "wallet",
Primary: "heightindex",
Index: []string{
"noteHash",
"nullifier",
"authHash",
"authSpendHash",
"account",
"status",
"owner_status"},
}
//NewStageTable 新建表
func NewMixTable(kvdb db.KV) *table.Table {
rowmeta := NewMixRow()
table, err := table.NewTable(rowmeta, kvdb, boardOpt)
if err != nil {
panic(err)
}
return table
}
//MixRow table meta 结构
type MixRow struct {
*mix.WalletDbMixInfo
}
//NewStageRow 新建一个meta 结构
func NewMixRow() *MixRow {
return &MixRow{WalletDbMixInfo: &mix.WalletDbMixInfo{}}
}
//CreateRow 新建数据行(注意index 数据一定也要保存到数据中,不能就保存heightindex)
func (r *MixRow) CreateRow() *table.Row {
return &table.Row{Data: &mix.WalletDbMixInfo{}}
}
//SetPayload 设置数据
func (r *MixRow) SetPayload(data types.Message) error {
if d, ok := data.(*mix.WalletDbMixInfo); ok {
r.WalletDbMixInfo = d
return nil
}
return types.ErrTypeAsset
}
//Get 按照indexName 查询 indexValue
func (r *MixRow) Get(key string) ([]byte, error) {
switch key {
case "heightindex":
return []byte(r.TxIndex), nil
case "noteHash":
return []byte(r.Info.NoteHash), nil
case "nullifier":
return []byte(r.Info.Nullifier), nil
case "authHash":
return []byte(r.Info.AuthorizeHash), nil
case "authSpendHash":
return []byte(r.Info.AuthorizeSpendHash), nil
case "account":
return []byte(r.Info.Account), nil
case "status":
return []byte(fmt.Sprintf("%2d", r.Info.Status)), nil
case "owner_status":
return []byte(fmt.Sprintf("%s_%2d", r.Info.Account, r.Info.Status)), nil
default:
return nil, types.ErrNotFound
}
}