forked from TrueCloudLab/frostfs-contract
[#65] reputation: Update reputation contract
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
parent
23f5f1e0e7
commit
88a82820b8
2 changed files with 80 additions and 57 deletions
|
@ -1,2 +1,11 @@
|
|||
name: "NeoFS Reputation"
|
||||
safemethods: ["list"]
|
||||
safemethods: ["get, getByID, listByEpoch"]
|
||||
events:
|
||||
- name: reputationPut
|
||||
parameters:
|
||||
- name: epoch
|
||||
type: Integer
|
||||
- name: peerID
|
||||
type: ByteArray
|
||||
- name: value
|
||||
type: ByteArray
|
||||
|
|
|
@ -2,35 +2,21 @@ package reputationcontract
|
|||
|
||||
import (
|
||||
"github.com/nspcc-dev/neo-go/pkg/interop"
|
||||
"github.com/nspcc-dev/neo-go/pkg/interop/iterator"
|
||||
"github.com/nspcc-dev/neo-go/pkg/interop/native/management"
|
||||
"github.com/nspcc-dev/neo-go/pkg/interop/native/std"
|
||||
"github.com/nspcc-dev/neo-go/pkg/interop/runtime"
|
||||
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
|
||||
"github.com/nspcc-dev/neofs-contract/common"
|
||||
)
|
||||
|
||||
const version = 1
|
||||
|
||||
const (
|
||||
peerIDSize = 46 // NeoFS PeerIDSize
|
||||
trustValSize = 8 // float64
|
||||
|
||||
trustStructSize = 0 +
|
||||
peerIDSize + // manager ID
|
||||
peerIDSize + // trusted ID
|
||||
trustValSize // trust value
|
||||
version = 1
|
||||
)
|
||||
|
||||
var (
|
||||
trustJournalPrefix = []byte("trustJournal")
|
||||
|
||||
ctx storage.Context
|
||||
)
|
||||
|
||||
func init() {
|
||||
ctx = storage.GetContext()
|
||||
}
|
||||
|
||||
func Init(owner interop.Hash160) {
|
||||
ctx := storage.GetContext()
|
||||
|
||||
if !common.HasUpdateAccess(ctx) {
|
||||
panic("only owner can reinitialize contract")
|
||||
}
|
||||
|
@ -41,6 +27,8 @@ func Init(owner interop.Hash160) {
|
|||
}
|
||||
|
||||
func Migrate(script []byte, manifest []byte) bool {
|
||||
ctx := storage.GetReadOnlyContext()
|
||||
|
||||
if !common.HasUpdateAccess(ctx) {
|
||||
runtime.Log("only owner can update contract")
|
||||
return false
|
||||
|
@ -52,47 +40,73 @@ func Migrate(script []byte, manifest []byte) bool {
|
|||
return true
|
||||
}
|
||||
|
||||
func Put(manager, epoch, typ []byte, newTrustList [][]byte) bool {
|
||||
if !runtime.CheckWitness(manager) {
|
||||
panic("put: incorrect manager key")
|
||||
func Put(epoch int, peerID []byte, value []byte) {
|
||||
ctx := storage.GetContext()
|
||||
|
||||
multiaddr := common.AlphabetAddress()
|
||||
if !runtime.CheckWitness(multiaddr) {
|
||||
runtime.Notify("reputationPut", epoch, peerID, value)
|
||||
return
|
||||
}
|
||||
|
||||
for i := 0; i < len(newTrustList); i++ {
|
||||
trustData := newTrustList[i]
|
||||
id := storageID(epoch, peerID)
|
||||
|
||||
if len(trustData) != trustStructSize {
|
||||
panic("list: invalid trust value")
|
||||
reputationValues := GetByID(id)
|
||||
reputationValues = append(reputationValues, value)
|
||||
|
||||
rawValues := std.Serialize(reputationValues)
|
||||
storage.Put(ctx, id, rawValues)
|
||||
}
|
||||
|
||||
func Get(epoch int, peerID []byte) [][]byte {
|
||||
id := storageID(epoch, peerID)
|
||||
return GetByID(id)
|
||||
}
|
||||
|
||||
func GetByID(id []byte) [][]byte {
|
||||
ctx := storage.GetReadOnlyContext()
|
||||
|
||||
data := storage.Get(ctx, id)
|
||||
if data == nil {
|
||||
return [][]byte{}
|
||||
}
|
||||
|
||||
return std.Deserialize(data.([]byte)).([][]byte)
|
||||
}
|
||||
|
||||
// ListByEpoch returns list of IDs that may be used to get reputation data
|
||||
// via GetByID method.
|
||||
func ListByEpoch(epoch int) [][]byte {
|
||||
ctx := storage.GetReadOnlyContext()
|
||||
it := storage.Find(ctx, epoch, storage.KeysOnly)
|
||||
|
||||
var result [][]byte
|
||||
|
||||
ignore := [][]byte{
|
||||
[]byte(common.OwnerKey),
|
||||
}
|
||||
|
||||
loop:
|
||||
for iterator.Next(it) {
|
||||
key := iterator.Value(it).([]byte) // iterator MUST BE `storage.KeysOnly`
|
||||
for _, ignoreKey := range ignore {
|
||||
if common.BytesEqual(key, ignoreKey) {
|
||||
continue loop
|
||||
}
|
||||
}
|
||||
|
||||
// todo: consider using notification for inner ring node
|
||||
|
||||
// todo: limit size of the trust journal:
|
||||
// history will be stored in chain (args or notifies)
|
||||
// contract storage will be used as a cache if needed
|
||||
key := append(trustJournalPrefix, append(epoch, typ...)...)
|
||||
|
||||
trustList := common.GetList(ctx, key)
|
||||
|
||||
// fixme: with neo3.0 it is kinda unnecessary
|
||||
if len(trustList) == 0 {
|
||||
// if journal slice is not initialized, then `append` will throw
|
||||
trustList = newTrustList
|
||||
} else {
|
||||
for i := 0; i < len(newTrustList); i++ {
|
||||
trustList = append(trustList, newTrustList[i])
|
||||
}
|
||||
result = append(result, key)
|
||||
}
|
||||
|
||||
common.SetSerialized(ctx, key, trustList)
|
||||
|
||||
runtime.Log("trust list was successfully updated")
|
||||
|
||||
return true
|
||||
return result
|
||||
}
|
||||
|
||||
func List(epoch, typ []byte) [][]byte {
|
||||
key := append(trustJournalPrefix, append(epoch, typ...)...)
|
||||
|
||||
return common.GetList(ctx, key)
|
||||
func Version() int {
|
||||
return version
|
||||
}
|
||||
|
||||
func storageID(epoch int, peerID []byte) []byte {
|
||||
var buf interface{} = epoch
|
||||
|
||||
return append(buf.([]byte), peerID...)
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue