chirpnest/workers.go

456 lines
18 KiB
Go
Raw Normal View History

2019-05-05 13:29:28 +02:00
package main
import (
2019-05-06 04:54:24 +02:00
"encoding/json"
2019-05-16 14:41:21 +02:00
"fmt"
2019-05-05 13:29:28 +02:00
"log"
2019-05-16 14:41:21 +02:00
"math/rand"
2019-05-05 13:29:28 +02:00
"strconv"
2019-05-08 12:17:07 +02:00
"time"
2019-05-06 04:54:24 +02:00
"github.com/streadway/amqp"
2019-05-16 04:21:36 +02:00
tb "gopkg.in/tucnak/telebot.v2"
2019-05-05 13:29:28 +02:00
)
2019-05-06 04:54:24 +02:00
func MQGetMsgWorker(id int, msgs chan<- ChatWarsMessage) {
2019-05-17 09:02:35 +02:00
log.Printf("MQGetMsgWorker[" + strconv.Itoa(id) + "] : Starting.")
2019-05-06 04:54:24 +02:00
var x ChatWarsMessage
conn, err := amqp.Dial("amqp://" + cfg.Rabbit.User + ":" + cfg.Rabbit.Password + "@" + cfg.Rabbit.Host + "/" + cfg.Rabbit.Queue)
failOnError(err, "MQGetMsgWorker["+strconv.Itoa(id)+"] : Failed to connect to RabbitMQ")
defer conn.Close()
ch, err := conn.Channel()
failOnError(err, "MQGetMsgWorker["+strconv.Itoa(id)+"] : Failed to open a channel")
defer ch.Close()
q, err := ch.QueueDeclare(
"msg", // name
false, // durable
false, // delete when unused
false, // exclusive
false, // no-wait
nil, // arguments
)
failOnError(err, "MQGetMsgWorker["+strconv.Itoa(id)+"] : Failed to declare a queue")
m, err := ch.Consume(
q.Name, // queue
"", // consumer
true, // auto-ack
false, // exclusive
false, // no-local
false, // no-wait
nil, // args
)
failOnError(err, "MQGetMsgWorker["+strconv.Itoa(id)+"] : Failed to register a consumer")
for d := range m {
2019-05-11 07:46:14 +02:00
// log.Printf("MQGetMsgWorker["+strconv.Itoa(id)+"] : Received a message: %s", string(d.Body))
2019-05-06 04:54:24 +02:00
err = json.Unmarshal(d.Body, &x)
logOnError(err, "MQGetMsgWorker["+strconv.Itoa(id)+"] : Can't unmarshal.\n"+string(d.Body))
if err == nil {
msgs <- x
}
}
log.Printf("MQGetMsgWorker[" + strconv.Itoa(id) + "] : Closing.")
}
func SQLCWMsgWorker(id int, msgs <-chan ChatWarsMessage, objIds chan<- int64) {
2019-05-05 13:29:28 +02:00
log.Printf("SQLCWMsgWorker[" + strconv.Itoa(id) + "] : Starting.")
for m := range msgs {
2019-05-25 09:25:11 +02:00
objId, err := addObjMsg(m.ID64, m.ChatID64, m.UserID64, m.SenderUserID64, m.Date, m.Text)
2019-05-06 04:54:24 +02:00
logOnError(err, "SQLCWMsgWorker["+strconv.Itoa(id)+"] : Inserting message.")
2019-05-11 09:01:24 +02:00
if err == nil && objId != 0 {
2019-05-11 07:46:14 +02:00
// log.Printf("SQLCWMsgWorker["+strconv.Itoa(id)+"] : Message inserted (%d).\n", objId)
2019-05-06 04:54:24 +02:00
objIds <- objId
2019-05-05 13:29:28 +02:00
}
}
log.Printf("SQLCWMsgWorker[" + strconv.Itoa(id) + "] : Closing.")
}
func SQLIdentifyMsgWorker(id int, objIds <-chan int64) {
2019-05-08 12:32:48 +02:00
var i int
2019-05-05 13:29:28 +02:00
log.Printf("SQLIdentifyMsgWorker[" + strconv.Itoa(id) + "] : Starting.")
for objId := range objIds {
2019-05-06 04:54:24 +02:00
m, err := getMsg(objId)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Retrieving message.")
if err == nil {
2019-05-11 07:47:32 +02:00
//log.Printf("SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Message retrieved (%d)\n%s\n", objId, m.Text)
2019-05-08 15:28:10 +02:00
for i = 0; i < len(msgParsingRules) && msgParsingRules[i].re.FindStringSubmatch(m.Text) == nil; i++ {
2019-05-08 12:31:52 +02:00
}
if i == len(msgParsingRules) {
2019-05-13 04:27:03 +02:00
log.Printf("SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Can't find message type in rules(%d)\n%s\n", objId, m.Text)
2019-05-06 09:28:34 +02:00
} else {
2019-05-08 12:34:51 +02:00
r := msgParsingRules[i].re
2019-05-08 12:31:52 +02:00
switch msgParsingRules[i].MsgTypeID {
2019-05-19 06:19:24 +02:00
case objSubTypeMessageReportAck:
2019-05-19 06:20:20 +02:00
cwm, err := parseSubTypeMessageReportAck(m, r)
2019-05-19 06:19:24 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageReportAck.")
2019-05-18 09:44:25 +02:00
cwm.ObjID64 = objId
2019-05-19 06:19:24 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageReportAck)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(ReportAck)")
2019-05-11 12:40:58 +02:00
case objSubTypeMessageMiniWar:
2019-05-13 08:12:31 +02:00
cwm, err := parseSubTypeMessageMiniWar(m, r)
2019-05-13 08:18:26 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageMiniWar.")
2019-05-14 04:28:26 +02:00
cwm.ObjID64 = objId
2019-05-14 04:29:47 +02:00
err = insertMsgMiniWar(cwm)
2019-05-14 04:28:26 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : insertMsgMiniWar")
err = setObjSubTypeId(objId, objSubTypeMessageMiniWar)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(MiniWar)")
2019-05-15 10:58:25 +02:00
case objSubTypeMessagePillageInc:
2019-05-16 14:39:12 +02:00
cwm, err := parseSubTypeMessagePillageInc(m, r)
2019-05-18 16:15:24 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessagePillageInc.")
2019-05-16 14:39:12 +02:00
cwm.ObjID64 = objId
err = insertMsgPillageInc(cwm)
2019-05-18 10:33:58 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : insertMsgPillageInc")
2019-05-15 10:58:25 +02:00
err = setObjSubTypeId(objId, objSubTypeMessagePillageInc)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(PillageInc)")
2019-05-18 12:33:53 +02:00
p := JobPayloadPillage{
ObjID64: objId,
}
b, _ := json.Marshal(&p)
err = createJob(objSubTypeJobPillage, objJobPriority, m.UserID64, m.Date.Add(time.Duration(25+rand.Intn(35))*time.Second), b)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : createJob(JobPillage)")
2019-05-18 12:59:43 +02:00
// only advertise live pillages
if m.Date.Add(3 * time.Minute).After(time.Now()) {
2019-05-18 12:38:24 +02:00
s := TGCommand{
Type: commandSendMsg,
Text: fmt.Sprintf("Catching pillage (%s)", m.Date.Format(time.RFC3339)),
ToUserID64: m.UserID64,
}
TGCmdQueue <- s
2019-05-18 12:59:43 +02:00
}
2019-05-16 14:39:12 +02:00
case objSubTypeMessageGo:
2019-05-18 09:44:25 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageGo)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(Go)")
2019-05-19 06:05:04 +02:00
case objSubTypeMessageReportReq:
err = setObjSubTypeId(objId, objSubTypeMessageReportReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(ReportReq)")
case objSubTypeMessageGReportReq:
err = setObjSubTypeId(objId, objSubTypeMessageGReportReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(GReportReq)")
case objSubTypeMessageHeroReq:
err = setObjSubTypeId(objId, objSubTypeMessageHeroReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(HeroReq)")
case objSubTypeMessageMeReq:
err = setObjSubTypeId(objId, objSubTypeMessageMeReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(MeReq)")
case objSubTypeMessageInventoryReq:
err = setObjSubTypeId(objId, objSubTypeMessageInventoryReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(InventoryReq)")
case objSubTypeMessageTimeReq:
err = setObjSubTypeId(objId, objSubTypeMessageTimeReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(TimeReq)")
case objSubTypeMessagePledge:
err = setObjSubTypeId(objId, objSubTypeMessagePledge)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(Pledge)")
2019-05-18 12:33:53 +02:00
case objSubTypeMessagePillageGo:
err = setObjSubTypeId(objId, objSubTypeMessagePillageGo)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(PillageGo)")
2019-05-18 12:59:43 +02:00
case objSubTypeMessagePillageTimeout:
err = setObjSubTypeId(objId, objSubTypeMessagePillageTimeout)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(PillageTimeout)")
2019-05-18 12:33:53 +02:00
case objSubTypeMessagePillageWin:
err = setObjSubTypeId(objId, objSubTypeMessagePillageWin)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(PillageWin)")
case objSubTypeMessagePillageLoss:
err = setObjSubTypeId(objId, objSubTypeMessagePillageLoss)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(PillageLoss)")
2019-05-18 13:15:50 +02:00
case objSubTypeMessageDuelFight:
2019-05-18 16:17:22 +02:00
cwm, err := parseSubTypeMessageDuelFight(m, r)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Parsing objSubTypeMessageDuelFight.")
cwm.ObjID64 = objId
2019-05-19 14:02:06 +02:00
err = insertMsgDuelFight(cwm)
2019-05-19 14:00:18 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : insertMsgAuctionAnnounce")
2019-05-18 13:15:50 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageDuelFight)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(DuelFight)")
2019-05-18 13:21:23 +02:00
case objSubTypeMessageGoQuest:
2019-05-18 16:15:24 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageGoQuest)
2019-05-18 13:21:23 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(GoQuest)")
2019-05-19 06:19:24 +02:00
case objSubTypeMessageGoFastFight:
err = setObjSubTypeId(objId, objSubTypeMessageGoFastFight)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(GoFastFight)")
2019-05-18 13:21:23 +02:00
case objSubTypeMessageGoArena:
2019-05-18 16:15:24 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageGoArena)
2019-05-18 13:21:23 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(GoArena)")
case objSubTypeMessageTop:
2019-05-18 16:15:24 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageTop)
2019-05-18 13:21:23 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(Top)")
2019-05-18 13:22:16 +02:00
case objSubTypeMessageMenu:
2019-05-18 16:15:24 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageMenu)
2019-05-18 13:22:16 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(Menu)")
2019-05-19 08:35:26 +02:00
case objSubTypeMessageBuyReq:
err = setObjSubTypeId(objId, objSubTypeMessageBuyReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(BuyReq)")
case objSubTypeMessageSellReq:
err = setObjSubTypeId(objId, objSubTypeMessageSellReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(SellReq)")
case objSubTypeMessageOrderbookReq:
err = setObjSubTypeId(objId, objSubTypeMessageOrderbookReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(OrderbookReq)")
2019-05-20 14:29:11 +02:00
case objSubTypeMessageOrderbookAck:
err = setObjSubTypeId(objId, objSubTypeMessageSellReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(SellReq)")
case objSubTypeMessageWithdrawReq:
err = setObjSubTypeId(objId, objSubTypeMessageWithdrawReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(WithdrawReq)")
case objSubTypeMessageWithdrawCode:
err = setObjSubTypeId(objId, objSubTypeMessageWithdrawCode)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(WithdrawCode)")
case objSubTypeMessageWithdrawRcv:
err = setObjSubTypeId(objId, objSubTypeMessageWithdrawRcv)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(WithdrawRcv)")
case objSubTypeMessageDepositReq:
err = setObjSubTypeId(objId, objSubTypeMessageDepositReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(DepositReq)")
case objSubTypeMessageDepositAck:
err = setObjSubTypeId(objId, objSubTypeMessageDepositAck)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(DepositAck)")
2019-05-20 16:16:49 +02:00
case objSubTypeMessageStockReq:
err = setObjSubTypeId(objId, objSubTypeMessageStockReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(StockReq)")
case objSubTypeMessageMiscReq:
err = setObjSubTypeId(objId, objSubTypeMessageMiscReq)
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(MiscReq)")
2019-05-08 16:34:47 +02:00
case objSubTypeMessageAuctionAnnounce:
2019-05-09 11:21:12 +02:00
cwm := ChatWarsMessageAuctionAnnounce{
2019-05-11 07:41:05 +02:00
ObjID64: objId,
2019-05-09 11:19:02 +02:00
}
2019-05-09 11:28:24 +02:00
l, _ := strconv.ParseInt(r.ReplaceAllString(m.Text, "${Lot}"), 10, 32)
cwm.LotID = int32(l)
2019-05-09 11:27:02 +02:00
cwm.Item = r.ReplaceAllString(m.Text, "${Item}")
2019-05-09 11:21:12 +02:00
cwm.Cond = r.ReplaceAllString(m.Text, "${Cond}")
cwm.Quality = r.ReplaceAllString(m.Text, "${Quality}")
cwm.Seller = r.ReplaceAllString(m.Text, "${Seller}")
cwm.Buyer = r.ReplaceAllString(m.Text, "${Buyer}")
2019-05-09 11:28:24 +02:00
p, _ := strconv.ParseInt(r.ReplaceAllString(m.Text, "${Price}"), 10, 32)
cwm.Price = int32(p)
2019-05-09 11:21:12 +02:00
cwm.Status = r.ReplaceAllString(m.Text, "${Status}")
2019-05-09 12:37:12 +02:00
cwm.End, _ = fromChatWarsDate(r.ReplaceAllString(m.Text, "${End}"))
2019-05-09 11:32:34 +02:00
err = insertMsgAuctionAnnounce(&cwm)
2019-05-14 04:28:26 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : insertMsgAuctionAnnounce")
2019-05-11 08:27:32 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageAuctionAnnounce)
2019-05-11 08:24:38 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(AuctionAnnounce)")
case objSubTypeMessageTimeAck:
2019-05-11 08:27:59 +02:00
_, err := fromChatWarsDate(r.ReplaceAllString(m.Text, "${Day}") + " " + r.ReplaceAllString(m.Text, "${Month}") + " " + r.ReplaceAllString(m.Text, "${Year}") + " " + r.ReplaceAllString(m.Text, "${Hour}") + ":" + r.ReplaceAllString(m.Text, "${Minute}"))
2019-05-10 11:23:47 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : fromChatWarsDate")
2019-05-11 08:27:32 +02:00
err = setObjSubTypeId(objId, objSubTypeMessageTimeAck)
2019-05-11 08:24:38 +02:00
logOnError(err, "SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : setObjSubTypeId(objSubTypeMessageTimeAck)")
2019-05-08 12:31:52 +02:00
default:
2019-05-11 07:46:14 +02:00
log.Printf("SQLIdentifyMsgWorker["+strconv.Itoa(id)+"] : Unknwon message type in rule %d : %d (%d)\n%s\n", msgParsingRules[i].ID, msgParsingRules[i].MsgTypeID, objId, m.Text)
2019-05-08 12:31:52 +02:00
}
2019-05-06 09:25:57 +02:00
}
2019-05-06 04:54:24 +02:00
}
2019-05-05 13:29:28 +02:00
}
log.Printf("SQLIdentifyMsgWorker[" + strconv.Itoa(id) + "] : Closing.")
}
2019-05-07 13:15:25 +02:00
func SQLJobWorker(id int) {
log.Printf("SQLJobWorker[" + strconv.Itoa(id) + "] : Starting.")
for true {
2019-05-10 11:23:47 +02:00
jobs, err := loadCurrentJobs()
logOnError(err, "SQLJobWorker["+strconv.Itoa(id)+"] : loadCurrentJobs")
2019-05-10 12:21:02 +02:00
if len(jobs) > 0 {
log.Printf("SQLJobWorker["+strconv.Itoa(id)+"] : %d jobs.\n", len(jobs))
2019-05-11 05:15:42 +02:00
}
2019-05-11 06:54:12 +02:00
for _, j := range jobs {
switch j.JobTypeID {
case objSubTypeJobRescanMsg:
jobRescan(j)
case objSubTypeJobSetJobDone:
jobSetDone(j)
2019-05-18 09:44:25 +02:00
case objSubTypeJobPillage:
jobPillage(j)
2019-05-11 06:54:12 +02:00
default:
log.Printf("SQLJobWorker["+strconv.Itoa(id)+"] : No handler for job type #%d.\n", j.JobTypeID)
2019-05-11 05:24:11 +02:00
}
2019-05-11 06:54:12 +02:00
}
2019-05-10 11:27:55 +02:00
if len(jobs) < SQLJobSliceSize {
time.Sleep(100 * time.Millisecond)
}
2019-05-07 13:15:25 +02:00
}
log.Printf("SQLJobWorker[" + strconv.Itoa(id) + "] : Closing.")
}
2019-05-11 05:15:42 +02:00
2019-05-16 04:16:35 +02:00
func TGCmdWorker(id int, b *tb.Bot, cmds <-chan TGCommand) {
log.Printf("TGCmdWorker[" + strconv.Itoa(id) + "] : Starting.")
for c := range cmds {
2019-05-16 13:34:26 +02:00
j, err := json.Marshal(c)
2019-05-16 04:54:52 +02:00
logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : Marshal(c)")
2019-05-16 13:34:26 +02:00
log.Printf("TGCmdWorker["+strconv.Itoa(id)+"] : new command.\n%s\n", string(j))
switch c.Type {
case commandReplyMsg:
2019-05-16 13:35:05 +02:00
ch := tb.Chat{
2019-05-16 13:34:26 +02:00
ID: c.FromChatID64,
}
2019-05-16 13:35:05 +02:00
m := tb.Message{
2019-05-16 13:34:26 +02:00
ID: int(c.FromMsgID64),
Chat: &ch,
}
2019-05-16 13:35:24 +02:00
_, err := b.Reply(&m, c.Text)
2019-05-16 13:34:26 +02:00
logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : Reply")
2019-05-17 09:20:50 +02:00
case commandSendMsg:
if c.ToChatID64 != 0 {
ch := tb.Chat{
ID: c.ToChatID64,
}
_, err := b.Send(&ch, c.Text)
logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : Send")
} else if c.ToUserID64 != 0 {
ch := tb.Chat{
ID: c.ToUserID64,
}
_, err := b.Send(&ch, c.Text)
logOnError(err, "TGCmdWorker["+strconv.Itoa(id)+"] : Send")
}
2019-05-16 13:34:26 +02:00
default:
}
2019-05-16 04:16:35 +02:00
}
log.Printf("TGCmdWorker[" + strconv.Itoa(id) + "] : Closing.")
}
2019-05-16 14:39:12 +02:00
2019-05-16 14:41:21 +02:00
func MQTGCmdWorker(id int, cmds <-chan TGCommand) {
2019-05-16 14:39:12 +02:00
log.Printf("MQTGCmdWorker[" + strconv.Itoa(id) + "] : Starting.")
for c := range cmds {
2019-05-17 10:30:01 +02:00
if _, ok := clientsKeepAlive[c.FromUserID64]; ok {
j, err := json.Marshal(c)
logOnError(err, "MQTGCmdWorker["+strconv.Itoa(id)+"] : Marshal(c)")
log.Printf("MQTGCmdWorker["+strconv.Itoa(id)+"] : new command.\n%s\n", string(j))
err = clientsQueue[c.FromUserID64].Channel.Publish(
2019-05-18 09:44:25 +02:00
"", // exchange
2019-05-17 10:30:01 +02:00
clientsQueue[c.FromUserID64].Queue.Name, // routing key
2019-05-18 09:44:25 +02:00
false, // mandatory
false, // immediate
2019-05-17 10:30:01 +02:00
amqp.Publishing{
ContentType: "application/json",
2019-05-17 10:31:13 +02:00
Body: []byte(j),
2019-05-17 10:30:01 +02:00
})
logOnError(err, "MQTGCmdWorker["+strconv.Itoa(id)+"] : Publishing message.")
}
2019-05-16 14:39:12 +02:00
}
log.Printf("MQTGCmdWorker[" + strconv.Itoa(id) + "] : Closing.")
}
2019-05-17 09:02:35 +02:00
func MQKeepAliveWorker() {
log.Printf("MQKeepAliveWorker : Starting.")
conn, err := amqp.Dial("amqp://" + cfg.Rabbit.User + ":" + cfg.Rabbit.Password + "@" + cfg.Rabbit.Host + "/" + cfg.Rabbit.Queue)
failOnError(err, "MQKeepAliveWorker : Failed to connect to RabbitMQ")
defer conn.Close()
ch, err := conn.Channel()
failOnError(err, "MQKeepAliveWorker : Failed to open a channel")
defer ch.Close()
q, err := ch.QueueDeclare(
"keepalive", // name
false, // durable
false, // delete when unused
false, // exclusive
false, // no-wait
nil, // arguments
)
failOnError(err, "MQKeepAliveWorker : Failed to declare a queue")
m, err := ch.Consume(
q.Name, // queue
"", // consumer
true, // auto-ack
false, // exclusive
false, // no-local
false, // no-wait
nil, // args
)
failOnError(err, "MQKeepAliveWorker : Failed to register a consumer")
for d := range m {
// log.Printf("MQKeepAliveWorker : Received a message: %s", string(d.Body))
x := MQKeepAlive{}
err = json.Unmarshal(d.Body, &x)
logOnError(err, "MQKeepAliveWorker : Can't unmarshal.\n"+string(d.Body))
if err == nil {
if x.Date.Add(time.Minute).Before(time.Now()) {
// outdate keep-alive
2019-05-17 10:09:09 +02:00
} else if _, ok := clientsKeepAlive[x.UserID64]; ok {
clientsKeepAlive[x.UserID64].Date = x.Date
2019-05-17 09:02:35 +02:00
} else {
2019-05-17 10:09:09 +02:00
clt := MQClient{}
clt.Connection, err = amqp.Dial("amqp://" + cfg.Rabbit.User + ":" + cfg.Rabbit.Password + "@" + cfg.Rabbit.Host + "/" + x.Queue)
logOnError(err, "MQKeepAliveWorker : Failed to connect to RabbitMQ")
2019-05-17 10:11:34 +02:00
clt.Channel, err = clt.Connection.Channel()
2019-05-17 10:09:09 +02:00
logOnError(err, "MQKeepAliveWorker : Failed to open a channel")
2019-05-17 10:11:34 +02:00
clt.Queue, err = clt.Channel.QueueDeclare(
2019-05-17 10:09:09 +02:00
"msg", // name
false, // durable
false, // delete when unused
false, // exclusive
false, // no-wait
nil, // arguments
)
logOnError(err, "MQKeepAliveWorker : Failed to declare a queue")
clientsKeepAlive[x.UserID64] = &x
2019-05-17 10:33:33 +02:00
clientsQueue[x.UserID64] = &clt
2019-05-17 10:09:09 +02:00
2019-05-17 09:02:35 +02:00
c := TGCommand{
Type: commandSendMsg,
2019-05-17 09:20:50 +02:00
ToUserID64: x.UserID64,
2019-05-17 09:02:35 +02:00
Text: "You are connected.",
}
TGCmdQueue <- c
2019-05-17 10:09:09 +02:00
c = TGCommand{
Type: commandSendMsg,
ToUserID64: cfg.Bot.Admin,
Text: fmt.Sprintf("%s is connected.", x.Nickname),
}
TGCmdQueue <- c
2019-05-17 09:02:35 +02:00
}
}
}
log.Printf("MQKeepAliveWorker : Closing.")
}
func MQTidyKeepAliveWorker() {
log.Printf("MQTidyKeepAliveWorker : Starting.")
for true {
t := time.Now()
2019-05-17 10:09:09 +02:00
for _, v := range clientsKeepAlive {
2019-05-17 09:02:35 +02:00
if v.Date.Add(90 * time.Second).Before(time.Now()) {
2019-05-17 10:12:51 +02:00
msgs, err := clientsQueue[v.UserID64].Channel.QueuePurge(clientsQueue[v.UserID64].Queue.Name, false)
2019-05-17 10:14:23 +02:00
logOnError(err, "MQTidyKeepAliveWorker : Channel.QueuePurge()")
err = clientsQueue[v.UserID64].Channel.Close()
logOnError(err, "MQTidyKeepAliveWorker : Channel.Close()")
err = clientsQueue[v.UserID64].Connection.Close()
logOnError(err, "MQTidyKeepAliveWorker : Connection.Close()")
2019-05-17 09:02:35 +02:00
c := TGCommand{
Type: commandSendMsg,
2019-05-17 09:20:50 +02:00
ToUserID64: v.UserID64,
2019-05-17 10:09:09 +02:00
Text: "Timeout, purging and closing command queue.",
2019-05-17 09:02:35 +02:00
}
TGCmdQueue <- c
2019-05-17 10:13:14 +02:00
c = TGCommand{
2019-05-17 10:09:09 +02:00
Type: commandSendMsg,
ToUserID64: cfg.Bot.Admin,
Text: fmt.Sprintf("Client %s timed out (%d messages purged).", v.Nickname, msgs),
}
TGCmdQueue <- c
delete(clientsKeepAlive, v.UserID64)
2019-05-17 09:02:35 +02:00
}
}
2019-05-17 09:07:47 +02:00
time.Sleep(time.Until(t.Add(time.Second)))
2019-05-17 09:02:35 +02:00
}
log.Printf("MQTidyKeepAliveWorker : Closing.")
}