opms_backend/opms_parent/app/service/contract/ctr_contract_cron.go

package service

import (
	"dashoo.cn/micro/app/service"
	"dashoo.cn/opms_libary/plugin/dingtalk"
	"dashoo.cn/opms_libary/plugin/dingtalk/message/corpconversation"
	"database/sql"
	"fmt"
	"github.com/360EntSecGroup-Skylar/excelize"
	"github.com/gogf/gf/frame/g"
	"github.com/gogf/gf/os/glog"
	"github.com/gogf/gf/os/gtime"
	"github.com/gogf/gf/util/gconv"
	"github.com/robfig/cron"
	"os"
)

// 初始化，创建每10分钟执行一次的定时任务
func init() {
	// 定时任务
	c := cron.New()
	spec1 := "0 0 20 * * 0" // 每周日晚八点
	spec2 := "0 0 20 1 * *" // 每月1号晚八点

	if err := c.AddJob(spec1, CollectionCron{}); err != nil {
		glog.Error(err)
	}
	if err := c.AddJob(spec2, ContractCron{}); err != nil {
		glog.Error(err)
	}

	c.Start()
}

// CollectionCron 回款定义
type CollectionCron struct {
}

// ContractCron 合同任务定义
type ContractCron struct {
}

// Run 计划回款到期提醒
func (c CollectionCron) Run() {
	tenant := g.Config().GetString("micro_srv.tenant")
	if tenant == "" {
		glog.Error("交接任务单定时任务租户码未设置，请前往配置")
		return
	}
	// 当前时间
	now := gtime.Now()
	collections, err := g.DB(tenant).Model("ctr_contract_collection_plan p").InnerJoin("ctr_contract c", "c.id=p.contract_id").Where(fmt.Sprintf("(p.contract_status='10' OR p.contract_status='20') AND p.plan_datetime>='%v'", now.Format("Y-m-d 00:00:00"))).Fields("p.*,c.incharge_id,c.incharge_name").FindAll()
	if err != nil && err != sql.ErrNoRows {
		glog.Error(err)
		return
	}

	data := make(map[int][]map[string]string)

	for _, collection := range collections {
		// 校验当前时间
		remindTime := collection["cashed_datetime"].GTime()
		if isMatchDate(remindTime, 5) || isMatchDate(remindTime, 15) || isMatchDate(remindTime, 30) || isMatchDate(remindTime, 60) || isMatchDate(remindTime, 90) {
			id := gconv.Int(collection["incharge_id"])
			if _, ok := data[id]; !ok {
				data[id] = make([]map[string]string, 0)
			}
			d := map[string]string{
				"contract_code":   gconv.String(collection["contract_code"]),
				"cashed_datetime": collection["cashed_datetime"].GTime().Format("Y-m-d"),
				"cust_name":       gconv.String(collection["cust_name"]),
				"plan_amount":     fmt.Sprintf("%.2f", collection["plan_amount"].Float64()),
			}
			data[id] = append(data[id], d)
		}
	}
	if len(data) > 0 {
		for id, collections := range data {
			f := excelize.NewFile()
			// Create a new sheet.
			sheet := f.NewSheet("Sheet1")

			// 表头
			f.SetCellValue("Sheet1", Div(1)+"1", "合同编号")
			f.SetCellValue("Sheet1", Div(2)+"1", "计划回款时间")
			f.SetCellValue("Sheet1", Div(3)+"1", "客户名称")
			f.SetCellValue("Sheet1", Div(4)+"1", "回款金额")

			for index, collection := range collections {
				f.SetCellValue("Sheet1", Div(1)+gconv.String(index+2), collection["contract_code"])
				f.SetCellValue("Sheet1", Div(2)+gconv.String(index+2), collection["cashed_datetime"])
				f.SetCellValue("Sheet1", Div(3)+gconv.String(index+2), collection["cust_name"])
				f.SetCellValue("Sheet1", Div(4)+gconv.String(index+2), collection["plan_amount"])
			}

			// Set active sheet of the workbook.
			f.SetActiveSheet(sheet)
			// Save xlsx file by the given path.
			file := "./temp/计划回款到期合同.xlsx"
			_, err = createPath("./temp")
			if err != nil {
				glog.Error(err)
				return
			}
			if err = f.SaveAs(file); err != nil {
				glog.Error(err)
				return
			}

			fileId, err := uploadFile(file)
			if err != nil {
				glog.Error(err)
				return
			}

			notifyMessage("计划回款到期提醒", gconv.String(id), fileId)
			// 删除已存在的文件
			err = os.Remove(file)
			if err != nil {
				glog.Error(err)
				return
			}
		}
	}
}

// Run 售后运维到期提醒
func (c ContractCron) Run() {
	tenant := g.Config().GetString("micro_srv.tenant")
	if tenant == "" {
		glog.Error("交接任务单定时任务租户码未设置，请前往配置")
		return
	}
	// 当前时间
	date := gtime.Now().AddDate(1, 0, 0)
	// 获取数据
	contracts, err := g.DB(tenant).Model("ctr_contract").Where(fmt.Sprintf("appro_status='30' AND contract_end_time LIKE '%v%%'", date.Format("Y-m"))).Order("contract_end_time ASC").FindAll()
	if err != nil && err != sql.ErrNoRows {
		glog.Error(err)
		return
	}

	data := make(map[int][]map[string]string)

	for _, contract := range contracts {
		id := gconv.Int(contract["incharge_id"])
		if _, ok := data[id]; !ok {
			data[id] = make([]map[string]string, 0)
		}
		d := map[string]string{
			"contract_code":     gconv.String(contract["contract_code"]),
			"contract_end_time": contract["contract_end_time"].GTime().Format("Y-m-d"),
			"cust_name":         gconv.String(contract["cust_name"]),
			"contract_amount":   fmt.Sprintf("%.2f", contract["contract_amount"].Float64()),
		}
		data[id] = append(data[id], d)
	}
	if len(data) > 0 {
		for id, collections := range data {
			f := excelize.NewFile()
			// Create a new sheet.
			sheet := f.NewSheet("Sheet1")

			// 表头
			f.SetCellValue("Sheet1", Div(1)+"1", "合同编号")
			f.SetCellValue("Sheet1", Div(2)+"1", "合同结束时间")
			f.SetCellValue("Sheet1", Div(3)+"1", "客户名称")
			f.SetCellValue("Sheet1", Div(4)+"1", "合同总金额")

			for index, collection := range collections {
				f.SetCellValue("Sheet1", Div(1)+gconv.String(index+2), collection["contract_code"])
				f.SetCellValue("Sheet1", Div(2)+gconv.String(index+2), collection["contract_end_time"])
				f.SetCellValue("Sheet1", Div(3)+gconv.String(index+2), collection["cust_name"])
				f.SetCellValue("Sheet1", Div(4)+gconv.String(index+2), collection["contract_amount"])
			}

			// Set active sheet of the workbook.
			f.SetActiveSheet(sheet)
			// Save xlsx file by the given path.
			file := "./temp/售后运维到期提醒.xlsx"
			_, err = createPath("./temp")
			if err != nil {
				glog.Error(err)
				return
			}
			if err = f.SaveAs(file); err != nil {
				glog.Error(err)
				return
			}

			fileId, err := uploadFile(file)
			if err != nil {
				glog.Error(err)
				return
			}

			notifyMessage("售后运维到期提醒", gconv.String(id), fileId)
			// 删除已存在的文件
			err = os.Remove(file)
			if err != nil {
				glog.Error(err)
				return
			}
		}
	}
}

func uploadFile(file string) (string, error) {
	var request corpconversation.UploadConversationFileRequest
	request.Type = "file"
	request.FileData = "media"
	request.FilePath = file

	c := dingtalk.Client.GetCorpConversation()
	resp, err := c.UploadConversationFile(&request)
	if err != nil {
		fmt.Println(err)
		return "", err
	}

	return resp.MediaId, nil
}

// notifyMessage 发送消息通知
func notifyMessage(title, ids, message string) {
	msg := g.MapStrStr{
		"msgTitle":    title,
		"msgContent":  message,
		"msgType":     "40", // 文件处理
		"recvUserIds": ids,
		"msgStatus":   "10",
		"sendType":    "30",
	}
	if err := service.CreateSystemMessage(msg); err != nil {
		glog.Error("消息提醒异常：", err)
	}
}

func isMatchDate(date *gtime.Time, days int) bool {
	begin := gtime.Now().AddDate(0, 0, days)
	end := gtime.Now().AddDate(0, 0, days+7)
	return begin.Format("Y-m-d 00:00:00") <= date.Format("Y-m-d H:i:s") && date.Format("Y-m-d H:i:s") <= end.Format("Y-m-d 23:59:59")
}

// 判断文件夹是否存在
func pathExists(path string) (bool, error) {
	_, err := os.Stat(path)
	if err == nil {
		return true, nil
	}
	if os.IsNotExist(err) {
		return false, nil
	}
	return false, err
}

func createPath(path string) (bool, error) {
	exist, err := pathExists(path)
	if err != nil {
		return false, err
	}

	if exist {
		return true, nil
	} else {
		// 创建文件夹
		err := os.MkdirAll(path, os.ModePerm)
		if err != nil {
			return false, err
		} else {
			return true, err
		}
	}
}

// Div 数字转字母
func Div(Num int) string {
	var (
		Str  string = ""
		k    int
		temp []int //保存转化后每一位数据的值，然后通过索引的方式匹配A-Z
	)
	//用来匹配的字符A-Z
	Slice := []string{"", "A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O",
		"P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"}

	if Num > 26 { //数据大于26需要进行拆分
		for {
			k = Num % 26 //从个位开始拆分，如果求余为0，说明末尾为26，也就是Z，如果是转化为26进制数，则末尾是可以为0的，这里必须为A-Z中的一个
			if k == 0 {
				temp = append(temp, 26)
				k = 26
			} else {
				temp = append(temp, k)
			}
			Num = (Num - k) / 26 //减去Num最后一位数的值，因为已经记录在temp中
			if Num <= 26 {       //小于等于26直接进行匹配，不需要进行数据拆分
				temp = append(temp, Num)
				break
			}
		}
	} else {
		return Slice[Num]
	}
	for _, value := range temp {
		Str = Slice[value] + Str //因为数据切分后存储顺序是反的，所以Str要放在后面
	}
	return Str
}