// 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 . package manual import ( "sync" "github.com/sirupsen/logrus" ) type Service interface { Loop(wg *sync.WaitGroup, quitChan <-chan bool) } type Spammer struct { Sender *TxSender } func NewTxSpammer(params []TxParams) Service { return &Spammer{ Sender: NewTxSender(params), } } func (s *Spammer) Loop(wg *sync.WaitGroup, quitChan <-chan bool) { forwardQuit := make(chan bool) doneChan, errChan := s.Sender.Send(forwardQuit) wg.Add(1) go func() { defer wg.Done() for { select { case err := <-errChan: logrus.Error(err) case forwardQuit <- <-quitChan: return case <-doneChan: return } } }() }