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
108
109
110
111
112
113
114
115
116
117
118
119
|
// Copyright (C) 2019-2021 Algorand, Inc.
// This file is part of go-algorand
//
// go-algorand is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// go-algorand 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 Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with go-algorand. If not, see <https://www.gnu.org/licenses/>.
package compactcert
import (
"context"
"database/sql"
"sync"
"github.com/algorand/go-algorand/crypto/compactcert"
"github.com/algorand/go-algorand/data/basics"
"github.com/algorand/go-algorand/data/bookkeeping"
"github.com/algorand/go-algorand/ledger/ledgercore"
"github.com/algorand/go-algorand/logging"
"github.com/algorand/go-algorand/network"
"github.com/algorand/go-algorand/protocol"
"github.com/algorand/go-algorand/util/db"
"github.com/algorand/go-deadlock"
)
type builder struct {
*compactcert.Builder
voters *ledgercore.VotersForRound
votersHdr bookkeeping.BlockHeader
}
// Worker builds compact certificates, by broadcasting
// signatures using this node's participation keys, by collecting
// signatures sent by others, and by sending out the resulting
// compact certs in a transaction.
type Worker struct {
// The mutex serializes concurrent message handler invocations
// from the network stack.
mu deadlock.Mutex
db db.Accessor
log logging.Logger
accts Accounts
ledger Ledger
net Network
txnSender TransactionSender
// builders is indexed by the round of the block being signed.
builders map[basics.Round]builder
ctx context.Context
shutdown context.CancelFunc
wg sync.WaitGroup
signed basics.Round
signedCh chan struct{}
}
// NewWorker constructs a new Worker, as used by the node.
func NewWorker(db db.Accessor, log logging.Logger, accts Accounts, ledger Ledger, net Network, txnSender TransactionSender) *Worker {
ctx, cancel := context.WithCancel(context.Background())
return &Worker{
db: db,
log: log,
accts: accts,
ledger: ledger,
net: net,
txnSender: txnSender,
builders: make(map[basics.Round]builder),
ctx: ctx,
shutdown: cancel,
signedCh: make(chan struct{}, 1),
}
}
// Start starts the goroutines for the worker.
func (ccw *Worker) Start() {
err := ccw.db.Atomic(func(ctx context.Context, tx *sql.Tx) error {
return initDB(tx)
})
if err != nil {
ccw.log.Warnf("ccw.Start(): initDB: %v", err)
return
}
ccw.initBuilders()
handlers := []network.TaggedMessageHandler{
{Tag: protocol.CompactCertSigTag, MessageHandler: network.HandlerFunc(ccw.handleSigMessage)},
}
ccw.net.RegisterHandlers(handlers)
latest := ccw.ledger.Latest()
ccw.wg.Add(1)
go ccw.signer(latest)
ccw.wg.Add(1)
go ccw.builder(latest)
}
// Shutdown stops any goroutines associated with this worker.
func (ccw *Worker) Shutdown() {
ccw.shutdown()
ccw.wg.Wait()
ccw.db.Close()
}
|