forked from DNAProject/DNA
-
Notifications
You must be signed in to change notification settings - Fork 1
/
txnpool.go
68 lines (60 loc) · 1.99 KB
/
txnpool.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
/*
* 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 req
import (
"time"
"github.com/dnaproject2/DNA/common"
"github.com/dnaproject2/DNA/common/log"
"github.com/dnaproject2/DNA/core/types"
"github.com/dnaproject2/DNA/errors"
p2pcommon "github.com/dnaproject2/DNA/p2pserver/common"
tc "github.com/dnaproject2/DNA/txnpool/common"
"github.com/ontio/ontology-eventbus/actor"
)
const txnPoolReqTimeout = p2pcommon.ACTOR_TIMEOUT * time.Second
var txnPoolPid *actor.PID
func SetTxnPoolPid(txnPid *actor.PID) {
txnPoolPid = txnPid
}
//add txn to txnpool
func AddTransaction(transaction *types.Transaction) {
if txnPoolPid == nil {
log.Error("[p2p]net_server AddTransaction(): txnpool pid is nil")
return
}
txReq := &tc.TxReq{
Tx: transaction,
Sender: tc.NetSender,
TxResultCh: nil,
}
txnPoolPid.Tell(txReq)
}
//get txn according to hash
func GetTransaction(hash common.Uint256) (*types.Transaction, error) {
if txnPoolPid == nil {
log.Warn("[p2p]net_server tx pool pid is nil")
return nil, errors.NewErr("[p2p]net_server tx pool pid is nil")
}
future := txnPoolPid.RequestFuture(&tc.GetTxnReq{Hash: hash}, txnPoolReqTimeout)
result, err := future.Result()
if err != nil {
log.Warnf("[p2p]net_server GetTransaction error: %v\n", err)
return nil, err
}
return result.(tc.GetTxnRsp).Txn, nil
}