tx-spammer/pkg/auto/sender.go
2020-10-23 12:27:49 -05:00

59 lines
1.7 KiB
Go

// VulcanizeDB
// Copyright © 2020 Vulcanize
// This program 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.
// This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
package auto
import (
"github.com/ethereum/go-ethereum/rpc"
"github.com/sirupsen/logrus"
"github.com/vulcanize/tx_spammer/pkg/shared"
)
// EthSender sends eth value transfer txs
type EthSender struct {
client *rpc.Client
}
// NewEthSender returns a new EthSender
func NewEthSender(config *Config) *EthSender {
return &EthSender{
client: config.Client,
}
}
// Send awaits txs off the provided work queue and sends them
func (s *EthSender) Send(quitChan <-chan bool, txRlpChan <-chan []byte) (<-chan bool, <-chan error) {
// done channel to signal quit signal has been received (any ongoing jobs were finished)
doneChan := make(chan bool)
// err channel returned to calling context
errChan := make(chan error)
go func() {
for {
select {
case tx := <-txRlpChan:
if err := shared.SendRawTransaction(s.client, tx); err != nil {
errChan <- err
}
case <-quitChan:
logrus.Info("quitting the Send loop")
close(doneChan)
return
}
}
}()
return doneChan, errChan
}