|
|
@@ -0,0 +1,309 @@
|
|
|
+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
|
|
|
+}
|
lk