/
wallet.go
107 lines (85 loc) · 2.24 KB
/
wallet.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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package daos
import (
"github.com/ethereum/go-ethereum/common"
"github.com/globalsign/mgo/bson"
"github.com/tomochain/tomodex/app"
"github.com/tomochain/tomodex/types"
)
// TokenDao contains:
// collectionName: MongoDB collection name
// dbName: name of mongodb to interact with
type WalletDao struct {
collectionName string
dbName string
}
func NewWalletDao() *WalletDao {
return &WalletDao{"wallets", app.Config.DBName}
}
func (dao *WalletDao) Create(wallet *types.Wallet) error {
err := wallet.Validate()
if err != nil {
logger.Error(err)
return err
}
wallet.ID = bson.NewObjectId()
err = db.Create(dao.dbName, dao.collectionName, wallet)
if err != nil {
logger.Error(err)
return err
}
return nil
}
func (dao *WalletDao) GetAll() ([]types.Wallet, error) {
var response []types.Wallet
err := db.Get(dao.dbName, dao.collectionName, bson.M{}, 0, 0, &response)
if err != nil {
logger.Error(err)
return nil, err
}
return response, nil
}
// GetByID function fetches details of a token based on its mongo id
func (dao *WalletDao) GetByID(id bson.ObjectId) (*types.Wallet, error) {
var response *types.Wallet
err := db.GetByID(dao.dbName, dao.collectionName, id, &response)
if err != nil {
logger.Error(err)
return nil, err
}
return response, nil
}
// GetByAddress function fetches details of a token based on its contract address
func (dao *WalletDao) GetByAddress(a common.Address) (*types.Wallet, error) {
q := bson.M{"address": a.Hex()}
var resp []types.Wallet
err := db.Get(dao.dbName, dao.collectionName, q, 0, 1, &resp)
if err != nil {
logger.Error(err)
return nil, err
}
if len(resp) == 0 {
logger.Info("No wallets found")
return nil, nil
}
return &resp[0], nil
}
func (dao *WalletDao) GetDefaultAdminWallet() (*types.Wallet, error) {
q := bson.M{"admin": true}
var resp types.Wallet
err := db.GetOne(dao.dbName, dao.collectionName, q, &resp)
if err != nil {
logger.Error(err)
return nil, err
}
return &resp, nil
}
func (dao *WalletDao) GetOperatorWallets() ([]*types.Wallet, error) {
q := bson.M{"operator": true}
res := []*types.Wallet{}
err := db.Get(dao.dbName, dao.collectionName, q, 0, 0, &res)
if err != nil || len(res) == 0 {
logger.Error(err)
return nil, err
}
return res, nil
}