eta_mobile/services/english_report.go

package services

import (
	"encoding/json"
	"errors"
	"eta/eta_mobile/models"
	"eta/eta_mobile/models/system"
	"eta/eta_mobile/services/alarm_msg"
	"eta/eta_mobile/utils"
	"fmt"
	"html"
	"strconv"
	"strings"
	"time"
)

// CreateNewEnglishReport 生成英文研报
func CreateNewEnglishReport(req models.AddEnglishReportReq, adminInfo *system.Admin) (newReportId int64, reportCode string, err error) {
	var contentSub string
	if req.Content != "" {
		content, e := FilterReportContentBr(req.Content)
		if e != nil {
			err = errors.New("内容去除前后空格失败, Err: " + e.Error())
			return
		}
		req.Content = content

		contentSub, e = GetReportContentSub(req.Content)
		if e != nil {
			go alarm_msg.SendAlarmMsg("ContentSub 失败，Err:"+e.Error(), 3)
			err = errors.New("获取报告内容前几段失败, Err: " + e.Error())
			return
		}
	}
	maxStage, e := models.GetEnglishReportStage(req.ClassifyIdFirst, req.ClassifyIdSecond)
	if e != nil {
		err = errors.New("期数获取失败, Err: " + e.Error())
		return
	}

	item := new(models.EnglishReport)
	item.AddType = req.AddType
	item.ClassifyIdFirst = req.ClassifyIdFirst
	item.ClassifyNameFirst = req.ClassifyNameFirst
	item.ClassifyIdSecond = req.ClassifyIdSecond
	item.ClassifyNameSecond = req.ClassifyNameSecond
	item.Title = req.Title
	item.Abstract = req.Abstract
	item.Author = req.Author
	item.Frequency = req.Frequency
	item.State = req.State
	item.Content = html.EscapeString(req.Content)
	item.Stage = maxStage + 1
	item.ContentSub = html.EscapeString(contentSub)
	item.CreateTime = req.CreateTime
	item.ModifyTime = time.Now()
	item.AdminId = adminInfo.AdminId
	item.AdminRealName = adminInfo.RealName
	newReportId, e = models.AddEnglishReport(item)
	if e != nil {
		err = errors.New("新增报告失败, Err: " + e.Error())
		return
	}
	reportCode = utils.MD5(strconv.Itoa(int(newReportId)))
	//修改唯一编码
	{
		go models.ModifyEnglishReportCode(newReportId, reportCode)
	}
	return
}

// IEnglishEmailSend 英文研报-邮件推送接口
type IEnglishEmailSend interface {
	NewClient() (err error)
	SendEmail(item *EnglishReportSendEmailRequest) (ok bool, result string, err error)
	BatchSendEmail(list []*EnglishReportSendEmailRequest) (results []*EnglishReportSendEmailResult, err error)
}

// EnglishReportSendEmailRequest 英文研报-推送邮件请求体
type EnglishReportSendEmailRequest struct {
	ReportId        int    `description:"英文报告ID"`
	EmailId         int    `description:"邮箱ID"`
	Email           string `description:"邮箱地址"`
	Subject         string `description:"邮件主题"`
	FromAlias       string `description:"发信人昵称"`
	ReportTitle     string `description:"报告标题"`
	ReportAbstract  string `description:"报告摘要"`
	ReportContent   string `description:"报告内容"`
	ReportShareLink string `description:"报告分享链接"`
	ReportTime      string `description:"报告时间"`
	HtmlBody        string `description:"模板内容主体"`
}

// EnglishReportSendEmailResult 英文研报-推送邮件响应体
type EnglishReportSendEmailResult struct {
	ReportId   int    `description:"英文报告ID"`
	EmailId    int    `description:"邮箱ID"`
	Email      string `description:"邮箱地址"`
	Ok         bool   `description:"是否推送成功"`
	SendData   string `description:"请求数据-JSON"`
	ResultData string `description:"推送结果-JSON"`
	Source     int    `description:"服务来源：1-阿里云；2-腾讯云"`
}

// BatchSendAliEnglishReportEmail 批量推送英文研报邮件
func BatchSendAliEnglishReportEmail(list []*EnglishReportSendEmailRequest) (err error) {
	defer func() {
		if err != nil {
			go alarm_msg.SendAlarmMsg("阿里云群发英文研报邮件失败, Err: "+err.Error(), 3)
		}
	}()
	if len(list) == 0 {
		return
	}
	requestMap := make(map[int]*EnglishReportSendEmailRequest, 0)
	for i := range list {
		requestMap[list[i].EmailId] = list[i]
	}

	// 请求阿里云接口批量推送
	aliEmail := new(AliyunEmail)
	resultList, e := aliEmail.BatchSendEmail(list)
	if e != nil {
		err = e
		return
	}

	// 返回的结果更新日志
	resendList := make([]*EnglishReportSendEmailRequest, 0)
	failLogIds := make([]int, 0)
	updateCols := []string{"SendData", "Result", "SendStatus", "ErrMsg"}
	for i := range resultList {
		var cond string
		var pars []interface{}
		cond = ` AND is_deleted = 0 AND report_id = ? AND email_id = ? AND source = ? AND send_status = ?`
		pars = append(pars, resultList[i].ReportId, resultList[i].EmailId, models.EnglishReportEmailLogSourceAli, models.EnglishReportEmailLogStatusIng)
		l, e := models.GetEnglishReportEmailLog(cond, pars)
		if e != nil {
			continue
		}
		l.SendData = resultList[i].SendData
		l.Result = resultList[i].ResultData
		if resultList[i].Ok {
			l.SendStatus = models.EnglishReportEmailLogStatusSuccess
		} else {
			l.SendStatus = models.EnglishReportEmailLogStatusFail
			failLogIds = append(failLogIds, l.Id)
			if requestMap[resultList[i].EmailId] != nil {
				resendList = append(resendList, requestMap[resultList[i].EmailId])
			}
			// 取出错误信息
			r := new(AliyunEmailResult)
			if e = json.Unmarshal([]byte(resultList[i].ResultData), &r); e != nil {
				continue
			}
			rd := new(AliyunEmailResultData)
			res := strings.Replace(r.Data, `\`, ``, -1)
			if e = json.Unmarshal([]byte(res), &rd); e != nil {
				continue
			}
			l.ErrMsg = rd.Message
		}
		if e = l.Update(updateCols); e != nil {
			continue
		}
	}

	// 推送失败的重新腾讯云, 若腾讯云也失败将不再自动重推, 用户手动去重推
	if len(resendList) > 0 && len(failLogIds) > 0 {
		_ = ResendTencentEnglishReportEmail(resendList, failLogIds)
	}
	return
}

// ResendTencentEnglishReportEmail 腾讯云邮件重新推送
func ResendTencentEnglishReportEmail(resendList []*EnglishReportSendEmailRequest, failLogIds []int) (err error) {
	defer func() {
		if err != nil {
			go alarm_msg.SendAlarmMsg("腾讯云重发英文研报邮件失败, Err: "+err.Error(), 3)
		}
	}()
	if len(resendList) == 0 || len(failLogIds) == 0 {
		return
	}

	// 标记原有日志为已删除
	if len(failLogIds) > 0 {
		if e := models.DeleteEnglishReportEmailLogByIds(failLogIds); e != nil {
			err = errors.New("删除原邮件日志失败, Err: " + e.Error())
			return
		}
	}

	// 写入新的日志
	nowTime := time.Now().Local()
	logData := make([]*models.EnglishReportEmailLog, 0)
	for i := range resendList {
		sendByte, e := json.Marshal(resendList[i])
		if e != nil {
			err = errors.New("sendByte json.Marshal Err, Err: " + e.Error())
			return
		}

		logData = append(logData, &models.EnglishReportEmailLog{
			ReportId:   resendList[i].ReportId,
			EmailId:    resendList[i].EmailId,
			Email:      resendList[i].Email,
			SendData:   string(sendByte),
			Source:     models.EnglishReportEmailLogSourceTencent,
			SendStatus: models.EnglishReportEmailLogStatusIng,
			CreateTime: nowTime,
		})
	}
	emailLog := new(models.EnglishReportEmailLog)
	if e := emailLog.InsertMulti(logData); e != nil {
		err = errors.New("批量写入群发邮件日志失败, Err: " + e.Error())
		return
	}

	// 请求腾讯云
	tecentEmail := new(TencentEmail)
	resultList, e := tecentEmail.BatchSendEmail(resendList)
	if e != nil {
		err = e
		return
	}
	updateCols := []string{"SendData", "Result", "SendStatus", "ErrMsg"}
	for i := range resultList {
		var cond string
		var pars []interface{}
		cond = ` AND is_deleted = 0 AND report_id = ? AND email_id = ? AND source = ? AND send_status = ?`
		pars = append(pars, resultList[i].ReportId, resultList[i].EmailId, models.EnglishReportEmailLogSourceTencent, models.EnglishReportEmailLogStatusIng)
		l, e := models.GetEnglishReportEmailLog(cond, pars)
		if e != nil {
			continue
		}
		l.SendData = resultList[i].SendData
		l.Result = resultList[i].ResultData
		if resultList[i].Ok {
			l.SendStatus = models.EnglishReportEmailLogStatusSuccess
		} else {
			l.SendStatus = models.EnglishReportEmailLogStatusFail
			r := new(TencentEmailResult)
			if e = json.Unmarshal([]byte(resultList[i].ResultData), &r); e != nil {
				continue
			}
			l.ErrMsg = r.Message
		}
		if e = l.Update(updateCols); e != nil {
			continue
		}
	}
	return
}

// UpdateEnglishReportEs 更新英文报告/章节Es
func UpdateEnglishReportEs(reportId int, publishState int) (err error) {
	if reportId <= 0 {
		return
	}
	reportInfo, err := models.GetEnglishReportById(reportId)
	if err != nil {
		return
	}
	// 新增报告ES
	esReport := &models.ElasticEnglishReportDetail{
		Id:                 strconv.Itoa(reportInfo.Id),
		ReportId:           reportInfo.Id,
		Title:              reportInfo.Title,
		Abstract:           reportInfo.Abstract,
		BodyContent:        utils.TrimHtml(html.UnescapeString(reportInfo.Content)),
		PublishTime:        reportInfo.PublishTime,
		CreateTime:         reportInfo.CreateTime,
		ReportCode:         reportInfo.ReportCode,
		PublishState:       publishState,
		Author:             reportInfo.Author,
		Frequency:          reportInfo.Frequency,
		ClassifyIdFirst:    reportInfo.ClassifyIdFirst,
		ClassifyNameFirst:  reportInfo.ClassifyNameFirst,
		ClassifyIdSecond:   reportInfo.ClassifyIdSecond,
		ClassifyNameSecond: reportInfo.ClassifyNameSecond,
		StageStr:           strconv.Itoa(reportInfo.Stage),
		Overview:           utils.TrimHtml(html.UnescapeString(reportInfo.Overview)),
		ContentSub:         utils.TrimHtml(html.UnescapeString(reportInfo.ContentSub)),
	}
	docId := fmt.Sprintf("%d", reportInfo.Id)
	if err = EsAddOrEditEnglishReport(utils.EsEnglishReportIndexName, docId, esReport); err != nil {
		return
	}
	return
}

// UpdateEnglishVideoEs 更新英文线上路演Es
func UpdateEnglishVideoEs(videoId int, publishState int) (err error) {
	if videoId <= 0 {
		return
	}
	videoInfo, err := models.GetEnglishVideoById(videoId)
	if err != nil {
		return
	}
	// 新增报告ES
	esReport := &models.ElasticEnglishReportDetail{
		Id:                 "v" + strconv.Itoa(videoInfo.Id),
		VideoId:            videoInfo.Id,
		Title:              videoInfo.Title,
		Abstract:           videoInfo.Abstract,
		BodyContent:        "",
		PublishTime:        videoInfo.PublishTime,
		CreateTime:         videoInfo.CreateTime,
		ReportCode:         videoInfo.VideoCode,
		PublishState:       publishState,
		Author:             videoInfo.Author,
		Frequency:          "",
		ClassifyIdFirst:    videoInfo.ClassifyIdFirst,
		ClassifyNameFirst:  videoInfo.ClassifyNameFirst,
		ClassifyIdSecond:   videoInfo.ClassifyIdSecond,
		ClassifyNameSecond: videoInfo.ClassifyNameSecond,
		StageStr:           "",
		Overview:           utils.TrimHtml(html.UnescapeString(videoInfo.Overview)),
		ContentSub:         "",
	}
	docId := fmt.Sprintf("v%d", videoInfo.Id)
	if err = EsAddOrEditEnglishReport(utils.EsEnglishReportIndexName, docId, esReport); err != nil {
		return
	}
	return
}

func UpdateEnglishReportClassifyId(oldItem, newItem, newParent *models.EnglishClassify, classifyId int) (err error) {
	//一级分类改为二级分类
	defer func() {
		if err != nil {
			go alarm_msg.SendAlarmMsg("英文报告分类改名-同步更新报告表字段及权限表关键词失败3, Err："+err.Error(), 3)
		}
	}()
	//如果二级分类变更为一级分类
	//如果二级分类更换了父级分类，则更新报告中的父级分类ID
	//如果一级分类更换了名称，则更新所有报告中的一级分类名称
	//如果二级分类更换了名称，则更新所有报告中的二级分类名称
	if oldItem.ParentId != newItem.ParentId {
		parentClassifyName := ""
		parentId := 0

		//二级分类改为一级分类, 或者二级分类的一级分类有变更
		if newItem.ParentId > 0 {
			parentClassifyName = newParent.ClassifyName
			parentId = newParent.Id
		}

		// 更新报告表分类字段
		var condition string
		var pars []interface{}

		condition += ` AND classify_id_second = ? `
		pars = append(pars, classifyId)
		list, e := models.GetEnglishReportByCondition(condition, pars)
		if e != nil {
			err = e
			return
		}
		if len(list) > 0 {
			//二级分类改为一级分类
			if oldItem.ParentId > 0 && newItem.ParentId == 0 {
				//查询该二级分类下是否存在关联报告，如果存在则不允许变更
				err = fmt.Errorf("该分类有关联的报告，不允许变更为一级分类")
				return
			}
		}
		var idSlice []string
		for _, report := range list {
			idSlice = append(idSlice, strconv.Itoa(report.Id))
		}
		ids := strings.Join(idSlice, ",")
		if ids != "" {
			if err = models.UpdateEnglishReportByClassifyId(parentClassifyName, newItem.ClassifyName, parentId, oldItem.Id, ids); err != nil {
				return
			}
		}
	} else if oldItem.ClassifyName != newItem.ClassifyName {
		if oldItem.ParentId > 0 {
			//只对二级分类名称作了修改
			// 更新报告表分类字段
			if err = models.UpdateEnglishReportSecondClassifyNameByClassifyId(classifyId, newItem.ClassifyName); err != nil {
				return
			}
		} else if oldItem.ParentId == 0 {
			//只对一级分类名称作了修改
			// 更新报告表分类字段
			if err = models.UpdateEnglishReportFirstClassifyNameByClassifyId(classifyId, newItem.ClassifyName); err != nil {
				return
			}
		}
	}
	return
}

func UpdateEnglishVideoClassifyId(oldItem, newItem, newParent *models.EnglishClassify, classifyId int) (err error) {
	//一级分类改为二级分类
	defer func() {
		if err != nil {
			go alarm_msg.SendAlarmMsg("英文报告分类改名-同步更新报告表字段及权限表关键词失败3, Err："+err.Error(), 3)
		}
	}()
	//如果二级分类变更为一级分类
	//如果二级分类更换了父级分类，则更新报告中的父级分类ID
	//如果一级分类更换了名称，则更新所有报告中的一级分类名称
	//如果二级分类更换了名称，则更新所有报告中的二级分类名称
	if oldItem.ParentId != newItem.ParentId {
		parentClassifyName := ""
		parentId := 0
		//二级分类改为一级分类, 或者二级分类的一级分类有变更
		if newItem.ParentId > 0 {
			parentClassifyName = newParent.ClassifyName
			parentId = newParent.Id
		}

		// 更新报告表分类字段
		var condition string
		var pars []interface{}

		condition += ` AND classify_id_second = ? `
		pars = append(pars, classifyId)
		list, e := models.GetEnglishVideoByCondition(condition, pars)
		if e != nil {
			err = e
			return
		}
		if len(list) > 0 {
			//二级分类改为一级分类
			if oldItem.ParentId > 0 && newItem.ParentId == 0 {
				//查询该二级分类下是否存在关联报告，如果存在则不允许变更
				err = fmt.Errorf("该分类有关联的视频，不允许变更为一级分类")
				return
			}
		}
		var idSlice []string
		for _, report := range list {
			idSlice = append(idSlice, strconv.Itoa(report.Id))
		}
		ids := strings.Join(idSlice, ",")
		if ids != "" {
			if err = models.UpdateEnglishVideoByClassifyId(parentClassifyName, newItem.ClassifyName, parentId, oldItem.Id, ids); err != nil {
				return
			}
		}
	} else if oldItem.ClassifyName != newItem.ClassifyName {
		if oldItem.ParentId > 0 {
			//只对二级分类名称作了修改
			// 更新报告表分类字段
			if err = models.UpdateEnglishVideoSecondClassifyNameByClassifyId(classifyId, newItem.ClassifyName); err != nil {
				return
			}
		} else if oldItem.ParentId == 0 {
			//只对一级分类名称作了修改
			// 更新报告表分类字段
			if err = models.UpdateEnglishVideoFirstClassifyNameByClassifyId(classifyId, newItem.ClassifyName); err != nil {
				return
			}
		}
	}
	return
}

// UpdateAllPublishedEnglishReportToEs 更新所有已发布的报告ES
func UpdateAllPublishedEnglishReportToEs() (err error) {
	// 获取所有已发布的报告
	var condition string
	var pars []interface{}
	condition = ` AND state = 2 `
	reportList, err := models.GetEnglishReportByCondition(condition, pars)
	count := 0
	failCount := 0
	for i := 0; i < len(reportList); i++ {
		if err = UpdateEnglishReportEs(reportList[i].Id, 2); err != nil {
			fmt.Printf("更新失败, report_id: %d, Err: %s\n", reportList[i].Id, err.Error())
			failCount += 1
		} else {
			count += 1
		}
	}
	fmt.Printf("报告总数：%d, 更新成功数: %d, 更新失败数: %d", len(reportList), count, failCount)

	return
}

// UpdateEnReportEditMark 更新英文研报当前更新状态
// status 枚举值 1：编辑中，0：完成编辑, 2：只做查询
func UpdateEnReportEditMark(reportId, nowUserId, status int, nowUserName string) (ret models.MarkReportResp, err error) {
	//更新标记key
	key := fmt.Sprint(`crm:enReport:edit:`, reportId)
	opUserId, e := utils.Rc.RedisInt(key)
	var opUser models.MarkReportItem
	if e != nil {
		opUserInfoStr, tErr := utils.Rc.RedisString(key)
		if tErr == nil {
			tErr = json.Unmarshal([]byte(opUserInfoStr), &opUser)
			if tErr == nil {
				opUserId = opUser.AdminId
			}
		}
	}
	if opUserId > 0 && opUserId != nowUserId {
		editor := opUser.Editor
		if editor == "" {
			//查询账号的用户姓名
			otherInfo, e := system.GetSysAdminById(opUserId)
			if e != nil {
				err = fmt.Errorf("查询其他编辑者信息失败")
				return
			}
			editor = otherInfo.RealName
		}

		ret.Status = 1
		ret.Msg = fmt.Sprintf("当前%s正在编辑报告", editor)
		ret.Editor = editor
		return
	}
	if status == 1 {
		nowUser := &models.MarkReportItem{AdminId: nowUserId, Editor: nowUserName}
		bt, e := json.Marshal(nowUser)
		if e != nil {
			err = fmt.Errorf("格式化编辑者信息失败")
			return
		}
		if opUserId > 0 {
			utils.Rc.Do("SETEX", key, int64(180), string(bt)) //3分钟缓存
		} else {
			utils.Rc.SetNX(key, string(bt), time.Second*60*3) //3分钟缓存
		}
	} else if status == 0 {
		//清除编辑缓存
		_ = utils.Rc.Delete(key)
	}
	return
}

// 取消英文策略报告一键同步
func EnglishPolicyReportSyncCancel(reportInfo *models.EnglishReportDetail) (err error) {
	var policyReport *models.EnglishPolicyReportDetail
	if reportInfo.FromReportId > 0 {
		//查询报告是否存在
		policyReport, err = models.GetEnglishPolicyReportById(reportInfo.FromReportId)
		if err != nil {
			if err.Error() == utils.ErrNoRow() {
				err = fmt.Errorf("报告不存在！")
				return
			}
			err = fmt.Errorf("获取报告信息失败,Err:%v", err)
			return
		}
		//判断报告是否已同步
		if policyReport.State == 1 {
			err = fmt.Errorf("报告已取消同步，无需重复取消")
			return
		}

		if err = models.SyncCancelEnglishPolicyReport(policyReport.Id); err != nil {
			err = fmt.Errorf("报告已取消同步失败 Err %v", err)
			return
		}
	}
	return
}