eta_report/services/knowledge_resource.go

package services

import (
	"eta_gn/eta_report/models"
	"eta_gn/eta_report/utils"
	"html"
	"strconv"
	"strings"
	"time"
)

func ReportKnowledgeResourceAdd(title string, fileUrl string, adminId int, adminRealName string) (msg string, err error) {
	classifyObj := new(models.KnowledgeClassify)
	childClassifyId, err := classifyObj.GetChildClassifyIdByNamePath("市场研究报告", "内部报告", "日常报告")
	if err != nil {
		if utils.IsErrNoRow(err) {
			msg = "分类不存在，请检查知识资源-报告库中是否存在-市场研究报告/内部报告/日常报告分类"
			return
		}
		msg = "报告库添加失败"
		return
	}

	obj := new(models.KnowledgeResource)
	obj.ResourceType = models.KnowledgeResourceTypeReport
	obj.ClassifyId = childClassifyId
	obj.Title = title
	obj.CreateTime = time.Now()
	obj.ModifyTime = time.Now()
	obj.State = models.KnowledgeResourceStatePublished
	timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
	obj.ResourceCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
	obj.AdminId = adminId
	obj.AdminRealName = adminRealName
	obj.SourceFrom = "市场信息研究所"
	currentTime := time.Now()
	obj.StartTime = &currentTime
	// 此处传入的url会在入库时存入knowledge_resource_file表中
	obj.FileUrl = fileUrl
	if fileUrl != "" {
		obj.IsFile = 1
	}
	err = obj.Create()
	if err != nil {
		return
	}
	return
}

func ReportKnowledgeResourceClassifyCheckAndSave(reportClassifyId int) (classifyId int, err error) {
	classifyObj := new(models.Classify)
	classifyObj, err = classifyObj.GetItemById(reportClassifyId)
	if err != nil {
		if !utils.IsErrNoRow(err) {
			return
		}
	}
	levelPath := classifyObj.LevelPath
	classifyIdStrs := strings.Split(levelPath, ",")
	classifyIds := make([]int, 0, len(classifyIdStrs))
	for _, classifyIdStr := range classifyIdStrs {
		classifyId, _ := strconv.Atoi(classifyIdStr)
		classifyIds = append(classifyIds, classifyId)
	}
	classifyList, err := classifyObj.GetItemsByIds(classifyIds)
	if err != nil {
		return
	}
	knowledgeClassifyObj := new(models.KnowledgeClassify)
	var parentId int
	for i, classify := range classifyList {
		knowledgeClassifyObj, err = knowledgeClassifyObj.GetClassifyByNameTypeAndParentId(classify.ClassifyName, models.KnowledgeResourceTypeOpinion, parentId)
		if err != nil {
			if !utils.IsErrNoRow(err) {
				return
			}
			for j := i; j < len(classifyList); j++ {
				tmp := &models.KnowledgeClassify{
					ClassifyName: classifyList[j].ClassifyName,
					Sort:         1,
					ParentId:     parentId,
					CreateTime:   time.Now(),
					ModifyTime:   time.Now(),
					Enabled:      1,
					Level:        classifyList[j].Level,
					ResourceType: models.KnowledgeResourceTypeOpinion,
				}
				err = tmp.Create()
				if err != nil {
					return
				}
				parentId = tmp.ClassifyId
			}
			classifyId = parentId
			return
		}
		parentId = knowledgeClassifyObj.ClassifyId
	}
	classifyId = parentId
	return
}

// ViewPointSave 知识资源观点库保存
func ViewPointSave(outId, adminId int, adminRealName, title, content, sourceFrom, classifyName string, classifyId int, startTime *time.Time) (msg string, err error) {
	if classifyId <= 0 {
		tmpClassifyId, er := ViewPointKnowledgeResourceClassifyCheckAndSave(classifyName, models.KnowledgeResourceTypeOpinion)
		if er != nil {
			msg = "观点保存失败"
			err = er
			return
		}
		classifyId = tmpClassifyId
	}
	obj := new(models.KnowledgeResource)
	knowledgeResource, err := obj.GetKnowledgeResourceByOutId(outId)
	if err != nil {
		if !utils.IsErrNoRow(err) {
			msg = "观点保存失败"
			return
		}
		// 创建新的知识资源库
		err = nil
		obj.ResourceType = models.KnowledgeResourceTypeOpinion
		obj.ClassifyId = classifyId
		obj.Title = title
		obj.CreateTime = time.Now()
		obj.ModifyTime = time.Now()
		obj.State = models.KnowledgeResourceStateApproved
		obj.Content = html.EscapeString(content)
		timestamp := strconv.FormatInt(time.Now().UnixNano(), 10)
		obj.ResourceCode = utils.MD5(utils.CHART_PREFIX + "_" + timestamp)
		obj.AdminId = adminId
		if adminRealName == "" || obj.AdminId == 0 {
			obj.AdminRealName = "无"
		} else {
			obj.AdminRealName = adminRealName
		}
		obj.SourceFrom = sourceFrom
		obj.StartTime = startTime
		obj.IsFile = 0
		obj.OutId = outId
		err = obj.Create()
		if err != nil {
			msg = "观点保存失败"
			return
		}
	} else {
		// 更新知识资源库的状态
		var updateCols []string
		if knowledgeResource.Title != title {
			knowledgeResource.Title = title
			updateCols = append(updateCols, "title")
		}
		if knowledgeResource.Content != content {
			knowledgeResource.Content = content
			updateCols = append(updateCols, "content")
		}
		if !knowledgeResource.StartTime.Equal(*startTime) {
			knowledgeResource.StartTime = startTime
			updateCols = append(updateCols, "start_time")
		}
		if knowledgeResource.SourceFrom != sourceFrom {
			knowledgeResource.SourceFrom = sourceFrom
			updateCols = append(updateCols, "source_from")
		}

		if len(updateCols) > 0 {
			knowledgeResource.ModifyTime = time.Now()
			updateCols = append(updateCols, "modify_time")
			err = knowledgeResource.Update(updateCols)
			if err != nil {
				msg = "观点保存失败"
				return
			}
		}
	}
	return
}

func ViewPointKnowledgeResourceClassifyCheckAndSave(classifyName string, resourceType int) (classifyId int, err error) {
	obj := new(models.KnowledgeClassify)
	obj, err = obj.GetClassifyByNameTypeAndParentId(classifyName, resourceType, 0)
	if err != nil {
		if !utils.IsErrNoRow(err) {
			return
		}
		// 创建新的分类
		err = nil
		obj.ClassifyName = classifyName
		obj.ResourceType = resourceType
		obj.CreateTime = time.Now()
		obj.ModifyTime = time.Now()
		obj.Enabled = 1
		obj.Level = 1
		obj.Sort = 1
		err = obj.Create()
		if err != nil {
			return
		}
		classifyId = obj.ClassifyId
	} else {
		classifyId = obj.ClassifyId
	}
	return
}