chirpnest/job.go

package main

import (
	"encoding/json"
	"fmt"
	"time"
)

func jobRescan(j Job) {
	var r JobPayloadRescanMsg

	err := setJobStart(j.ID64)
	logOnError(err, "jobRescan : setJobStart")

	err = json.Unmarshal(j.Payload, &r)
	logOnError(err, "jobRescan : Unmarshal payload")

	ids := getSQLListID64(r.Query)

	if len(ids) > 1 {
		for _, id := range ids {
			p := JobPayloadRescanMsg{
				Query: fmt.Sprintf("SELECT o.id FROM obj o WHERE o.id = %d AND o.obj_type_id = %d AND o.obj_sub_type_id = %d;", id, objTypeMessage, objSubTypeMessageUnknown),
			}

			b, _ := json.Marshal(p)
			err := createJob(objSubTypeJobRescanMsg, objJobPriorityRescanChildMsg, j.UserID64, time.Now(), b)
			logOnError(err, "jobRescan : createJob(objSubTypeJobRescanMsg)")
		}
		p := JobPayloadSetDone{
			JobID64:  j.ID64,
			MsgID64:  r.MsgID64,
			ChatID64: r.ChatID64,
			Text:     fmt.Sprintf("%d messages processed.", len(ids)),
		}
		b, _ := json.Marshal(p)
		err := createJob(objSubTypeJobSetJobDone, objJobPriority, j.UserID64, time.Now(), b)
		logOnError(err, "jobRescan : createJob(objSubTypeJobSetJobDone)")
	} else if len(ids) == 1 {
		SQLMsgIdentifyQueue <- ids[0]
		err = setJobDone(j.ID64)
		logOnError(err, "jobRescan : setJobDone(1)")
		m := TGCommand{
			Type:         commandReplyMsg,
			Text:         "One message processed.",
			FromMsgID64:  r.MsgID64,
			FromChatID64: r.ChatID64,
		}
		TGCmdQueue <- m
	} else {
		err = setJobDone(j.ID64)
		logOnError(err, "jobRescan : setJobDone(0)")
		m := TGCommand{
			Type:         commandReplyMsg,
			Text:         "No message processed.",
			FromMsgID64:  r.MsgID64,
			FromChatID64: r.ChatID64,
		}
		TGCmdQueue <- m
	}
	return
}

func jobSetDone(j Job) {
	var r JobPayloadSetDone

	err := setJobStart(j.ID64)
	logOnError(err, "jobSetDone : setJobStart")

	err = json.Unmarshal(j.Payload, &r)
	logOnError(err, "jobSetDone : Unmarshal payload")

	err = setJobDone(r.JobID64)
	logOnError(err, "jobSetDone : setJobDone(child)")

	err = setJobDone(j.ID64)
	logOnError(err, "jobSetDone : setJobDone")

	m := TGCommand{
		Type:         commandReplyMsg,
		Text:         r.Text,
		FromMsgID64:  r.MsgID64,
		FromChatID64: r.ChatID64,
	}
	TGCmdQueue <- m

	return
}