-
Notifications
You must be signed in to change notification settings - Fork 6
/
scheduled.go
71 lines (61 loc) · 2.23 KB
/
scheduled.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
//go:generate protoc -I=. -I=$GOPATH/src -I=$GOPATH/src/github.com/ElrondNetwork/protobuf/protobuf --gogoslick_out=. scheduled.proto
package scheduled
import (
"github.com/ElrondNetwork/elrond-go-core/data"
"github.com/ElrondNetwork/elrond-go-core/data/block"
"github.com/ElrondNetwork/elrond-go-core/data/smartContractResult"
"github.com/ElrondNetwork/elrond-go-core/data/transaction"
)
// GetTransactionHandlersMap returns the smart contract results as a map of transaction handlers
func (sscr *ScheduledSCRs) GetTransactionHandlersMap() map[block.Type][]data.TransactionHandler {
if sscr == nil {
return nil
}
if len(sscr.Scrs) == 0 && len(sscr.InvalidTransactions) == 0 {
return nil
}
result := make(map[block.Type][]data.TransactionHandler)
var smartContractResults []data.TransactionHandler
for i := range sscr.Scrs {
smartContractResults = append(smartContractResults, sscr.Scrs[i])
}
result[block.SmartContractResultBlock] = smartContractResults
var invalidTxs []data.TransactionHandler
for i := range sscr.InvalidTransactions {
invalidTxs = append(invalidTxs, sscr.InvalidTransactions[i])
}
result[block.InvalidBlock] = invalidTxs
return result
}
// SetTransactionHandlersMap fills the smart contract results map from the given transaction handlers map
func (sscr *ScheduledSCRs) SetTransactionHandlersMap(txHandlersMap map[block.Type][]data.TransactionHandler) error {
if sscr == nil {
return data.ErrNilPointerReceiver
}
if txHandlersMap == nil {
sscr.Scrs = nil
sscr.InvalidTransactions = nil
return nil
}
var smartContractResults []*smartContractResult.SmartContractResult
txHandlers := txHandlersMap[block.SmartContractResultBlock]
for j := range txHandlers {
scr, ok := txHandlers[j].(*smartContractResult.SmartContractResult)
if !ok {
return data.ErrInvalidTypeAssertion
}
smartContractResults = append(smartContractResults, scr)
}
var invalidTxs []*transaction.Transaction
txHandlers = txHandlersMap[block.InvalidBlock]
for j := range txHandlers {
invalidTx, ok := txHandlers[j].(*transaction.Transaction)
if !ok {
return data.ErrInvalidTypeAssertion
}
invalidTxs = append(invalidTxs, invalidTx)
}
sscr.Scrs = smartContractResults
sscr.InvalidTransactions = invalidTxs
return nil
}