添加发票提醒脚本

cmd/task/invoice_remind/main.go

+package main
+
+/**
+	发票提醒监听任务
+	1. 任务未结束前，此任务唯一，不重复生成
+	2. 不设置截止时间
+ */
+import (
+	"crm-server/cmd/task/common"
+	"crm-server/configs"
+	"crm-server/internal/dao"
+	"crm-server/internal/model"
+	"crm-server/internal/service"
+	"flag"
+	"fmt"
+	"github.com/ichunt2019/logger"
+	"strconv"
+	"time"
+)
+
+// 日志目录
+var LogDir string
+
+// 解析命令行参数
+func initArgs() {
+	flag.StringVar(&LogDir, "logDir", "", "日志目录")
+	flag.Parse()
+}
+
+func main () {
+	initArgs()
+
+	logConfig := make(map[string]string)
+	logConfig["log_path"] = LogDir+"/task/invoice_remind"
+	logConfig["log_chan_size"] = "100"
+	logger.InitLogger("file",logConfig)
+	logger.Init()
+
+	// 查看任务是否开启
+	task_setup := common.GetTaskSetup(8)
+
+	setTaskInfo(task_setup.Status, task_setup.Id, task_setup.TaskType, task_setup.TaskDisplayDegree)
+	fmt.Println("任务执行完成")
+}
+
+func setTaskInfo(status, task_id, task_type, task_display_degree int) {
+	if status == 0 || status == -1 { // 数据不存在或未开启则直接中断
+		return
+	}
+
+	db := dao.GetNewDb()
+	// 获取H5留言的用户
+	row, err := db.Queryx("select * from lie_feedback where type = ? and is_add_task = ? order by id desc", 12, 0)
+	defer row.Close()
+
+	if err != nil {
+		logger.Info("获取发票提醒失败: ", err.Error())
+		return
+	}
+
+	for row.Next() {
+		var feedback model.Feedback
+		row.StructScan(&feedback)
+
+		// 判断该用户是否存在待处理的发票提醒任务记录，若存在则跳过
+		var task_info model.TaskInfo
+		db.Get(&task_info, "select * from lie_task_info where user_id = ? and status = ? and task_type = ? limit 1", feedback.Outter_Uid, 1, 8)
+
+		if task_info.Id != 0 {
+			continue
+		}
+
+		// 查看用户是否分配
+		var salesman model.Salesman
+		db.Get(&salesman, "select * from lie_salesman where user_id = ?", feedback.Outter_Uid)
+
+		if salesman.Sale_Id == 0 { // 若未分配，则跳过
+			continue
+		}
+
+		current_time := time.Now().Unix() // 当前时间
+
+		db.Exec("insert into lie_task_info (user_id, task_id, task_type, task_display_degree, sale_id, create_time, update_time) value (?, ?, ?, ?, " +
+			"?, ?, ?)", feedback.Outter_Uid, task_id, task_type, task_display_degree, salesman.Sale_Id, current_time, current_time)
+		db.Exec("update lie_feedback set is_add_task = ? where id = ?", 1, feedback.Id)
+
+		service.DingAlert(configs.Ding_crm_task_token, "任务告警：发票提醒任务脚本添加任务记录成功，CRM用户ID："+strconv.Itoa(feedback.Outter_Uid), "")
+	}
+
+
+}
\ No newline at end of file