chirpnest/job.go

109 lines
2.5 KiB
Go
Raw Normal View History

2019-05-11 06:54:12 +02:00
package main
import (
2019-05-11 06:55:05 +02:00
"encoding/json"
2019-05-11 06:54:12 +02:00
"fmt"
2019-05-11 06:55:05 +02:00
"time"
2019-05-11 06:54:12 +02:00
)
2019-05-11 06:54:49 +02:00
func jobRescan(j Job) {
2019-05-11 07:06:40 +02:00
var r JobPayloadRescanMsg
2019-05-11 06:54:12 +02:00
err := setJobStart(j.ID64)
logOnError(err, "jobRescan : setJobStart")
2019-05-11 07:06:40 +02:00
2019-05-11 06:56:48 +02:00
err = json.Unmarshal(j.Payload, &r)
2019-05-11 07:06:40 +02:00
logOnError(err, "jobRescan : Unmarshal payload")
2019-05-11 06:54:12 +02:00
ids := getSQLListID64(r.Query)
2019-05-11 07:06:40 +02:00
2019-05-11 06:54:12 +02:00
if len(ids) > 1 {
for _, id := range ids {
2019-05-11 07:20:44 +02:00
p := JobPayloadRescanMsg{
2019-05-16 05:22:20 +02:00
Query: fmt.Sprintf("SELECT %d id FROM DUAL;", id),
2019-05-16 05:06:38 +02:00
MsgID64: 0,
ChatID64: 0,
2019-05-11 07:20:44 +02:00
}
2019-05-11 06:54:12 +02:00
b, _ := json.Marshal(p)
2019-05-16 04:49:34 +02:00
err := createJob(objSubTypeJobRescanMsg, objJobPriorityRescanChildMsg, j.UserID64, time.Now(), b)
2019-05-11 06:54:12 +02:00
logOnError(err, "jobRescan : createJob(objSubTypeJobRescanMsg)")
}
p := JobPayloadSetDone{
2019-05-16 04:49:34 +02:00
JobID64: j.ID64,
2019-05-16 04:52:30 +02:00
MsgID64: r.MsgID64,
ChatID64: r.ChatID64,
2019-05-16 04:49:34 +02:00
Text: fmt.Sprintf("%d messages processed.", len(ids)),
2019-05-11 06:54:12 +02:00
}
b, _ := json.Marshal(p)
2019-05-16 04:49:34 +02:00
err := createJob(objSubTypeJobSetJobDone, objJobPriority, j.UserID64, time.Now(), b)
2019-05-11 06:54:12 +02:00
logOnError(err, "jobRescan : createJob(objSubTypeJobSetJobDone)")
} else if len(ids) == 1 {
SQLMsgIdentifyQueue <- ids[0]
err = setJobDone(j.ID64)
2019-05-11 07:06:40 +02:00
logOnError(err, "jobRescan : setJobDone(1)")
2019-05-16 05:06:38 +02:00
if r.MsgID64 != 0 || r.ChatID64 != 0 {
m := TGCommand{
Type: commandReplyMsg,
Text: "One message processed.",
FromMsgID64: r.MsgID64,
FromChatID64: r.ChatID64,
}
TGCmdQueue <- m
2019-05-16 04:49:34 +02:00
}
2019-05-11 07:06:40 +02:00
} else {
err = setJobDone(j.ID64)
logOnError(err, "jobRescan : setJobDone(0)")
2019-05-16 05:06:38 +02:00
if r.MsgID64 != 0 || r.ChatID64 != 0 {
m := TGCommand{
Type: commandReplyMsg,
2019-05-16 05:11:56 +02:00
Text: "No message processed.",
2019-05-16 05:06:38 +02:00
FromMsgID64: r.MsgID64,
FromChatID64: r.ChatID64,
}
TGCmdQueue <- m
2019-05-16 04:49:34 +02:00
}
2019-05-11 06:54:12 +02:00
}
return
}
2019-05-11 06:54:49 +02:00
func jobSetDone(j Job) {
2019-05-11 07:06:40 +02:00
var r JobPayloadSetDone
2019-05-11 06:54:12 +02:00
err := setJobStart(j.ID64)
logOnError(err, "jobSetDone : setJobStart")
2019-05-11 07:06:40 +02:00
2019-05-11 06:56:48 +02:00
err = json.Unmarshal(j.Payload, &r)
2019-05-11 07:06:40 +02:00
logOnError(err, "jobSetDone : Unmarshal payload")
2019-05-11 06:54:12 +02:00
err = setJobDone(r.JobID64)
logOnError(err, "jobSetDone : setJobDone(child)")
2019-05-11 07:06:40 +02:00
2019-05-11 06:54:12 +02:00
err = setJobDone(j.ID64)
logOnError(err, "jobSetDone : setJobDone")
2019-05-11 07:06:40 +02:00
2019-05-16 04:52:30 +02:00
m := TGCommand{
Type: commandReplyMsg,
Text: r.Text,
FromMsgID64: r.MsgID64,
FromChatID64: r.ChatID64,
}
TGCmdQueue <- m
2019-05-11 06:54:12 +02:00
return
}
2019-05-16 14:39:12 +02:00
func jobPillage(j Job) {
var r JobPayloadPillage
err := setJobStart(j.ID64)
logOnError(err, "jobPillage : setJobStart")
err = json.Unmarshal(j.Payload, &r)
logOnError(err, "jobPillage : Unmarshal payload")
err = setJobDone(j.ID64)
logOnError(err, "jobPillage : setJobDone")
return
}