-
Notifications
You must be signed in to change notification settings - Fork 0
/
store.go
90 lines (81 loc) · 2.55 KB
/
store.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
/*
* Copyright (C) 2018 The DNA Authors
* This file is part of The DNA library.
*
* The DNA is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* The DNA is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with The DNA. If not, see <http://www.gnu.org/licenses/>.
*/
package utils
import (
"bytes"
"github.com/dnaproject2/DNA/common/serialization"
cstates "github.com/dnaproject2/DNA/core/states"
"github.com/dnaproject2/DNA/errors"
"github.com/dnaproject2/DNA/smartcontract/service/native"
)
func GetStorageItem(native *native.NativeService, key []byte) (*cstates.StorageItem, error) {
store, err := native.CacheDB.Get(key)
if err != nil {
return nil, errors.NewDetailErr(err, errors.ErrNoCode, "[GetStorageItem] storage error!")
}
if store == nil {
return nil, nil
}
item := new(cstates.StorageItem)
err = item.Deserialize(bytes.NewBuffer(store))
if err != nil {
return nil, errors.NewDetailErr(err, errors.ErrNoCode, "[GetStorageItem] instance doesn't StorageItem!")
}
return item, nil
}
func GetStorageUInt64(native *native.NativeService, key []byte) (uint64, error) {
item, err := GetStorageItem(native, key)
if err != nil {
return 0, err
}
if item == nil {
return 0, nil
}
v, err := serialization.ReadUint64(bytes.NewBuffer(item.Value))
if err != nil {
return 0, err
}
return v, nil
}
func GetStorageUInt32(native *native.NativeService, key []byte) (uint32, error) {
item, err := GetStorageItem(native, key)
if err != nil {
return 0, err
}
if item == nil {
return 0, nil
}
v, err := serialization.ReadUint32(bytes.NewBuffer(item.Value))
if err != nil {
return 0, err
}
return v, nil
}
func GenUInt64StorageItem(value uint64) *cstates.StorageItem {
bf := new(bytes.Buffer)
serialization.WriteUint64(bf, value)
return &cstates.StorageItem{Value: bf.Bytes()}
}
func GenUInt32StorageItem(value uint32) *cstates.StorageItem {
bf := new(bytes.Buffer)
serialization.WriteUint32(bf, value)
return &cstates.StorageItem{Value: bf.Bytes()}
}
func PutBytes(native *native.NativeService, key []byte, value []byte) {
native.CacheDB.Put(key, cstates.GenRawStorageItem(value))
}