Merge branch 'yb_admin/4.6' into debug

models/company_product.go

@@ -265,3 +265,14 @@ func GetAllCompanyProduct() (items []*CompanyProduct, err error) {
 	_, err = o.Raw(sql).QueryRows(&items)
 	return
 }
+
+// AddCompanyProductTryOutDayTotal 将客户产品的试用天数+1
+func AddCompanyProductTryOutDayTotal(companyId, productId int) (err error) {
+	o := orm.NewOrm()
+	sql := `UPDATE  company_product
+			SET
+			  try_out_day_total = try_out_day_total+1
+			WHERE company_id = ? AND product_id = ?`
+	_, err = o.Raw(sql, companyId, productId).Exec()
+	return
+}

models/company_product_try_out_update_log.go

@@ -0,0 +1,187 @@
+package models
+
+import (
+	"github.com/beego/beego/v2/client/orm"
+	"time"
+)
+
+// CompanyProductTryOutUpdateLog 客户产品试用变更日志表
+type CompanyProductTryOutUpdateLog struct {
+	Id          int       `orm:"column(id);pk"`
+	CompanyId   int       `description:"客户id"`
+	ProductId   int       `description:"产品id"`
+	SellerId    int       `description:"销售id"`
+	SellerName  string    `description:"销售名称"`
+	Source      string    `description:"来源"`
+	StartDate   time.Time `description:"开始日期"`
+	EndDate     time.Time `description:"结束日期"`
+	RealEndDate time.Time `description:"实际结束日期"`
+	IsStop      int       `description:"是否已经终止了，0：进行中；1：已终止"`
+	CreateTime  time.Time `description:"创建时间"`
+}
+
+// Update 更新客户产品试用变更日志
+func (item *CompanyProductTryOutUpdateLog) Update(cols []string) (err error) {
+	o := orm.NewOrm()
+	_, err = o.Update(item, cols...)
+	return
+}
+
+// CompanyProductTryOutUpdatePermissionLog 客户产品变更品种详情表
+type CompanyProductTryOutUpdatePermissionLog struct {
+	Id                        int       `orm:"column(id);pk"`
+	CompanyProductUpdateLogId int       `description:"产品变更id"`
+	ChartPermissionId         int       `description:"品种权限id"`
+	StartDate                 time.Time `description:"开始日期"`
+	EndDate                   time.Time `description:"结束日期"`
+	RealEndDate               time.Time `description:"实际结束日期"`
+	IsStop                    int       `description:"是否已经终止了，0：进行中；1：已终止"`
+	CreateTime                time.Time `description:"创建时间"`
+}
+
+// Update 更新客户产品变更品种详情
+func (item *CompanyProductTryOutUpdatePermissionLog) Update(cols []string) (err error) {
+	o := orm.NewOrm()
+	_, err = o.Update(item, cols...)
+	return
+}
+
+// AddCompanyProductTryOutUpdateLog 新增客户试用产品变更日志
+func AddCompanyProductTryOutUpdateLog(item *CompanyProductTryOutUpdateLog, companyReportPermissionList []*CompanyReportPermission) (err error) {
+	o := orm.NewOrm()
+	lastId, err := o.Insert(item)
+	if err != nil {
+		return
+	}
+	item.Id = int(lastId)
+
+	companyProductUpdatePermissionLogList := make([]*CompanyProductTryOutUpdatePermissionLog, 0)
+	for _, v := range companyReportPermissionList {
+		tmpCompanyProductUpdatePermissionLog := &CompanyProductTryOutUpdatePermissionLog{
+			//Id:                        0,
+			CompanyProductUpdateLogId: item.Id,
+			ChartPermissionId:         v.ChartPermissionId,
+			StartDate:                 item.StartDate,
+			EndDate:                   item.EndDate,
+			RealEndDate:               item.StartDate,
+			IsStop:                    item.IsStop,
+			CreateTime:                item.CreateTime,
+		}
+		companyProductUpdatePermissionLogList = append(companyProductUpdatePermissionLogList, tmpCompanyProductUpdatePermissionLog)
+	}
+
+	if len(companyProductUpdatePermissionLogList) > 0 {
+		_, err = o.InsertMulti(len(companyProductUpdatePermissionLogList), companyProductUpdatePermissionLogList)
+	}
+
+	{
+		// 获取当前正在进行中的客户变更试用的数据
+		findProgress, _ := GetCompanyProductTryOutUpdateLogItem(item.CompanyId, item.ProductId)
+		// 如果没有的话，那么就将试用天数+1
+		if findProgress != nil {
+			sql := `UPDATE  company_product
+			SET
+			  try_out_day_total = try_out_day_total+1
+			WHERE company_id = ? AND product_id = ?`
+			_, err = o.Raw(sql, item.CompanyId, item.ProductId).Exec()
+		}
+	}
+	return
+}
+
+// GetCompanyProductTryOutUpdateLogItem 获取当前正在进行中的客户变更试用的数据
+func GetCompanyProductTryOutUpdateLogItem(companyId, productId int) (item *CompanyProductTryOutUpdateLog, err error) {
+	o := orm.NewOrm()
+	sql := ` select * from company_product_try_out_update_log where company_id=? and product_id=? and is_stop = 0`
+	err = o.Raw(sql, companyId, productId).QueryRow(&item)
+	return
+}
+
+// GetCompanyProductTryOutUpdateGroup 获取当前正在进行中的客户变更试用的数据
+func GetCompanyProductTryOutUpdateGroup() (items []*CompanyProductTryOutUpdateLog, err error) {
+	o := orm.NewOrm()
+	sql := ` select * from company_product_try_out_update_log group by company_id,product_id;`
+	_, err = o.Raw(sql).QueryRows(&items)
+	return
+}
+
+// GetCompanyProductTryOutUpdateNoStopGroup 获取当前正在进行中的客户变更试用的数据（未停止的）
+func GetCompanyProductTryOutUpdateNoStopGroup() (items []*CompanyProductTryOutUpdateLog, err error) {
+	o := orm.NewOrm()
+	sql := ` select * from company_product_try_out_update_log where is_stop=0 group by company_id,product_id;`
+	_, err = o.Raw(sql).QueryRows(&items)
+	return
+}
+
+// GetCompanyProductTryOutUpdateList 获取当前正在进行中的客户变更试用的数据
+func GetCompanyProductTryOutUpdateList(companyId, productId int) (items []*CompanyProductTryOutUpdateLog, err error) {
+	o := orm.NewOrm()
+	sql := ` select * from company_product_try_out_update_log Where company_id=? and product_id=? order by create_time asc;`
+	_, err = o.Raw(sql, companyId, productId).QueryRows(&items)
+	return
+}
+
+// GetCompanyProductTryOutUpdateNoStopList 获取当前正在进行中的客户变更试用的数据（未停止的）
+func GetCompanyProductTryOutUpdateNoStopList(companyId, productId int) (items []*CompanyProductTryOutUpdateLog, err error) {
+	o := orm.NewOrm()
+	sql := ` select * from company_product_try_out_update_log Where company_id=? and product_id=? AND is_stop=0 order by create_time asc;`
+	_, err = o.Raw(sql, companyId, productId).QueryRows(&items)
+	return
+}
+
+// GetCompanyProductTryOutUpdateNoStopListByEndDate 获取当前正在进行中的客户变更试用的数据（未停止的）
+func GetCompanyProductTryOutUpdateNoStopListByEndDate(companyId, productId int) (items []*CompanyProductTryOutUpdateLog, err error) {
+	o := orm.NewOrm()
+	sql := ` select * from company_product_try_out_update_log Where company_id=? and product_id=? AND is_stop=0 order by end_date desc;`
+	_, err = o.Raw(sql, companyId, productId).QueryRows(&items)
+	return
+}
+
+// AddCompanyProductUpdatePermissionLog 客户产品变更品种详情日志
+func AddCompanyProductUpdatePermissionLog(item *CompanyProductTryOutUpdatePermissionLog) (lastId int64, err error) {
+	o := orm.NewOrm()
+	lastId, err = o.Insert(item)
+	return
+}
+
+// GetCompanyProductTryOutUpdateLogTotal 获取客户转试用的记录列表
+func GetCompanyProductTryOutUpdateLogTotal(companyId, productId int) (total int, err error) {
+	o := orm.NewOrm()
+	var pars []interface{}
+	sql := `select count(1) total from company_product_try_out_update_log a join company_product b on a.company_id=b.company_id and a.product_id=b.product_id where a.company_id = ? `
+	pars = append(pars, companyId)
+
+	//如果有传入产品id，那么只查询该类型下的产品
+	if productId != 0 {
+		sql += ` and a.product_id = ? `
+		pars = append(pars, productId)
+	}
+	err = o.Raw(sql, pars...).QueryRow(&total)
+	return
+}
+
+// GetCompanyProductTryOutUpdateLogList 获取客户转试用的记录列表
+func GetCompanyProductTryOutUpdateLogList(companyId, productId, startSize, pageSize int) (items []*CompanyProductTryOutUpdateLog, err error) {
+	o := orm.NewOrm()
+	var pars []interface{}
+	sql := `select a.* from company_product_try_out_update_log a join company_product b on a.company_id=b.company_id and a.product_id=b.product_id where a.company_id = ? `
+	pars = append(pars, companyId)
+
+	//如果有传入产品id，那么只查询该类型下的产品
+	if productId != 0 {
+		sql += ` and a.product_id = ? `
+		pars = append(pars, productId)
+	}
+	sql += ` order by a.create_time desc  LIMIT ?,? `
+	pars = append(pars, startSize, pageSize)
+	_, err = o.Raw(sql, pars...).QueryRows(&items)
+	return
+}
+
+// GetCompanyProductTryOutPermissionUpdateNoStopListByEndDate 获取当前正在进行中的客户变更试用品种的数据（未停止的）
+func GetCompanyProductTryOutPermissionUpdateNoStopListByEndDate(companyProductUpdateLogId int) (items []*CompanyProductTryOutUpdatePermissionLog, err error) {
+	o := orm.NewOrm()
+	sql := ` select * from company_product_try_out_update_permission_log Where company_product_update_log_id=? AND is_stop=0 order by end_date desc;`
+	_, err = o.Raw(sql, companyProductUpdateLogId).QueryRows(&items)
+	return
+}

models/company_report_permission.go

@@ -20,7 +20,6 @@ type CompanyReportPermission struct {
 	EndDate                   string    `description:"合同结束日期"`
 	ModifyTime                string    `description:"更新时间"`
 	IsUpgrade                 int       `description:"是否升级，1是，0否"`
-
 }
 
 func GetCompanyReportPermission(companyId int) (items []*CompanyReportPermission, err error) {
@@ -101,3 +100,10 @@ func GetCompanyReportPermissionCount(companyId, productId int) (count int, err e
 	err = o.Raw(countSql, startDate, companyId, productId).QueryRow(&count)
 	return
 }
+
+func GetCompanyReportPermissionByStatus(companyId, productId int, status string) (items []*CompanyReportPermission, err error) {
+	o := orm.NewOrm()
+	sql := `SELECT * FROM company_report_permission WHERE company_id=? AND product_id=? AND status = ? `
+	_, err = o.Raw(sql, companyId, productId, status).QueryRows(&items)
+	return
+}

services/company_product.go

@@ -1,12 +1,13 @@
 package services
 
 import (
+	"context"
 	"fmt"
 	"hongze/hongze_task/models"
 	"hongze/hongze_task/utils"
 	"strconv"
+	"strings"
 	"time"
-	"context"
 )
 
 //客户自动冻结->试用两个月结束后，进入冻结
@@ -465,3 +466,115 @@ func CompanyRemind1Day(seller *models.Sellers) {
 		}
 	}
 }
+
+// StaticCompanyTryDay 定时任务每天更新客户试用总天数
+func StaticCompanyTryDay(cont context.Context) (err error) {
+	errList := make([]string, 0)
+	defer func() {
+		if len(errList) > 0 {
+			go utils.SendEmail(utils.APPNAME+"【"+utils.RunMode+"】"+"失败提醒", "定时任务每天更新客户试用总天数 ErrMsg:"+strings.Join(errList, "\n"), utils.EmailSendToUsers)
+		}
+	}()
+
+	list, err := models.GetCompanyProductTryOutUpdateNoStopGroup()
+	if err != nil {
+		fmt.Println("获取客户变更数据失败：", err)
+		return
+	}
+
+	for _, v := range list {
+		isAdd := false //是否要增加一天，默认不加
+		permissionList, tmpErr := models.GetCompanyReportPermissionByStatus(v.CompanyId, v.ProductId, "试用")
+		if tmpErr == nil {
+			errList = append(errList, fmt.Sprint("company_id:", v.CompanyId, ";product_id:", v.ProductId, "；寻找对应的试用品种失败，Err:", tmpErr))
+			continue
+		}
+		currPermissionIdList := make([]int, 0) //当前试用的品种
+		for _, permission := range permissionList {
+			currPermissionIdList = append(currPermissionIdList, permission.ChartPermissionId)
+		}
+
+		//获取所有未停止的记录列表
+		logList, tmpErr := models.GetCompanyProductTryOutUpdateNoStopListByEndDate(v.CompanyId, v.ProductId)
+		if tmpErr != nil {
+			errList = append(errList, fmt.Sprint("company_id:", v.CompanyId, ";product_id:", v.ProductId, "；寻找客户的试用日志失败，Err:", tmpErr))
+			continue
+		}
+
+		logPermissionIdList := make([]int, 0) //当前日志中的试用的品种
+		for _, log := range logList {
+			//获取所有未停止的品种试用记录列表
+			logPermissionList, tmpErr := models.GetCompanyProductTryOutPermissionUpdateNoStopListByEndDate(log.Id)
+			if tmpErr != nil {
+				errList = append(errList, fmt.Sprint("company_id:", v.CompanyId, ";product_id:", v.ProductId, ";logId:", v.Id, "；寻找客户的试用品种日志失败，Err:", tmpErr))
+				continue
+			}
+			lenLogPermissionList := len(logPermissionList) //当前日志存在试用的品种数量
+			stopPermission := 0                            //当前日志需要停止的品种数量
+
+			currTime := time.Now() //当前时间
+			for _, logPermission := range logPermissionList {
+				if utils.InArrayByInt(logPermissionIdList, logPermission.ChartPermissionId) {
+					// 如果已经被其他记录使用了，那么就将当前记录给标记停止
+					logPermission.IsStop = 1
+					// 如果当前时间晚于结束日期，那么实际结束日期就是今天之前一天，否则就是当天
+					if currTime.After(logPermission.EndDate) {
+						logPermission.RealEndDate = currTime.AddDate(0, 0, -1)
+					} else {
+						logPermission.RealEndDate = time.Now()
+					}
+					logPermission.Update([]string{"IsStop", "RealEndDate"})
+					stopPermission++
+					continue
+				} else if !utils.InArrayByInt(currPermissionIdList, logPermission.ChartPermissionId) {
+					// 如果该品种不在当前客户的品种里面，那么也要将当前记录给标记停止
+					logPermission.IsStop = 1
+					// 如果当前时间晚于结束日期，那么实际结束日期就是今天之前一天，否则就是当天
+					if currTime.After(logPermission.EndDate) {
+						logPermission.RealEndDate = currTime.AddDate(0, 0, -1)
+					} else {
+						logPermission.RealEndDate = time.Now()
+					}
+					logPermission.Update([]string{"IsStop", "RealEndDate"})
+					stopPermission++
+					continue
+				}
+
+				// 说明当天还是试用状态，需要变更实际结束日期
+				logPermission.RealEndDate = currTime
+				logPermission.Update([]string{"RealEndDate"})
+
+				// 剩下的说明还处于试用状态，需要添加1天试用期，且需要把该品种加入到当前日志中的试用的品种列表
+				isAdd = true
+				logPermissionIdList = append(logPermissionIdList, logPermission.ChartPermissionId)
+			}
+
+			//如果当前日志存在试用的品种数量 == 当前日志需要停止的品种数量
+			// 那么当前日志也是处于停用状态
+			if lenLogPermissionList == stopPermission {
+				log.IsStop = 1
+				// 如果当前时间晚于结束日期，那么实际结束日期就是今天之前一天，否则就是当天
+				if currTime.After(log.EndDate) {
+					log.RealEndDate = currTime.AddDate(0, 0, -1)
+				} else {
+					log.RealEndDate = time.Now()
+				}
+				log.Update([]string{"IsStop", "RealEndDate"})
+			}
+
+		}
+
+		// 如果需要添加，那么将该客户品种添加1天
+		if isAdd {
+
+		}
+		// 更新客户产品的试用天数
+		tmpErr = models.AddCompanyProductTryOutDayTotal(v.CompanyId, v.ProductId)
+		if tmpErr != nil {
+			errList = append(errList, fmt.Sprint("company_id:", v.CompanyId, ";product_id:", v.ProductId, ";logId:", v.Id, "；更新客户产品的试用天数，Err:", tmpErr))
+			continue
+		}
+	}
+
+	return
+}

services/task.go

@@ -47,8 +47,12 @@ func Task() {
 	companyReportPermissionClose := task.NewTask("companyReportPermissionClose", "0 35 2 * * *", CompanyReportPermissionClose)
 	task.AddTask("用户产品权限试用-->关闭", companyReportPermissionClose)
 
+	//用户产品权限试用-->关闭
+	staticCompanyTryDay := task.NewTask("StaticCompanyTryDay", "0 40 2 * * *", StaticCompanyTryDay)
+	task.AddTask("定时任务每天更新客户试用总天数", staticCompanyTryDay)
+
 	// 存量客户数据统计
-	stackCompanyStatistic := task.NewTask("stackCompanyStatistic", "0 35 2 * * *", StackCompanyStatistic)
+	stackCompanyStatistic := task.NewTask("stackCompanyStatistic", "0 40 2 * * *", StackCompanyStatistic)
 	task.AddTask("存量客户数据统计", stackCompanyStatistic)
 
 	// 定时往同花顺推送报告

utils/constants.go

@@ -145,25 +145,5 @@ const (
 
 //模板消息推送类型
 const (
-	TEMPLATE_MSG_REPORT                = iota + 1 //日度点评报告推送
-	TEMPLATE_MSG_INDEX                            //指标更新
-	TEMPLATE_MSG_APPLY                            //审批通知
-	TEMPLATE_MSG_RECEIVE                          //销售领取客户通知
-	TEMPLATE_MSG_CYGX_ACTIVITY_CACLE              //查研观向活动取消通知
-	TEMPLATE_MSG_CYGX_ACTIVITY_UPDATE             //查研观向活动更改时间通知
-	TEMPLATE_MSG_CYGX_ARTICLE                     //关注的作者发布报告通知
-	TEMPLATE_MSG_CYGX_DAY_REPORT                  //发送日报（周报、双周报、月报）
-	TEMPLATE_MSG_ACTIVITY_APPOINTMENT             //活动预约/报名时间通知
-	TEMPLATE_MSG_RECEIVE_PERMISSION               //销售"添加客户/领取客户"权限变更通知
-	TEMPLATE_MSG_YB_COMMUNITY_QUESTION            //研报问答社区
-	TEMPLATE_MSG_CYGX_ACTIVITY_ADD                //查研观向小程序活动更新推送 12
-	TEMPLATE_MSG_CYGX_ARTICLE_ADD                 //查研观向报告更新推送  13
-	TEMPLATE_MSG_YB_PRICE_DRIVEN                  //研报价格驱动 14
-	TEMPLATE_MSG_SEAL_FINISHED                    //用印申请已迁回通知 15
-	TEMPLATE_MSG_YB_COMMUNITY_VIDEO               //研报视频社区 16
-	TEMPLATE_MSG_MOVE_SELLER                      //销售"移动/删除联系人"权限变更通知 17
-	TEMPLATE_MSG_YB_VOICE_BROADCAST               //研报语音播报 18
-	TEMPLATE_MSG_YB_VIDEO_UPLOAD                  //研报视频上传 19
-	TEMPLATE_MSG_YB_ROAD_VIDEO                    //研报线上路演 20
-	TEMPLATE_MSG_CYGX_ROADSHOW_VIDEO              //查研观向微路演视频 21
+	TEMPLATE_MSG_YB_VOICE_BROADCAST = 20 //研报语音播报
 )