bot/runner.go

package main

import (
	"github.com/sirupsen/logrus"
	"math/rand"
	"sync/atomic"
	"time"
)

const (
	MAX_RUN    = 3
	MIN_RUN    = 1
	TIME_CHECK = 10 * 60 * time.Second // 十分钟检查一次
)

var BotQueue *queue
var Working int64

func init() {
	BotQueue = newQueue()
}

func Run(bots []Account) {
	AccountShuffle(bots)

	for _, bot := range bots {
		BotQueue.rPush(bot)
	}
	logrus.Info("bots queue init")

	ticker := time.NewTicker(TIME_CHECK)
	defer ticker.Stop()

	botCh := make(chan Account)
	master(botCh)

	for {
		select {
		case <-ticker.C:
			master(botCh)

		case exitBot := <-botCh:
			logrus.Warnf("bot: %s exit!", exitBot.Player)
			atomic.AddInt64(&Working, -1)
			BotQueue.rPush(exitBot) // 退出机器人重新加到队尾

		}
	}
}

func master(botCh chan<- Account) {
	logrus.Info("check running worker")

	running := atomic.LoadInt64(&Working)
	logrus.Infof("running %d bots, queue len: %d", running, BotQueue.getLen())

	if running <= MIN_RUN {
		need := MAX_RUN - running

		var i int64
		for ; i < need; i++ {
			bot := BotQueue.lPop()
			if bot == nil {
				logrus.Error("bot queue empty!!!")
				break
			}
			go worker(bot.(Account), botCh)
		}
	}
}

func worker(bot Account, botCh chan<- Account) {
	logrus.Infof("bot: %s running~", bot.Player)
	atomic.AddInt64(&Working, 1)
	for {
		rand.Seed(time.Now().UnixNano())
		c := rand.Intn(51) + 30

		err := Transfer(bot.Player, bot.StarId)
		if err != nil { // 下注失败退出
			logrus.WithField("error", err).Warnf("bot: %s transfer error", bot.Player)
			break
		}

		time.Sleep(time.Duration(c) * time.Second)
	}

	botCh <- bot
}