eta_task/models/dk.go

package models

import (
	"eta/eta_task/global"
	"eta/eta_task/utils"
)

type WechatArticleTag struct {
	TagName     string
	ArticleId   string
	TextContent string
}

type WechatArticle struct {
	TextContent string
}

func GetWechatArticleTags(classifyName, startTime, endTime string, startSize, pageSize int) (list []*WechatArticleTag, err error) {
	sql := `SELECT a.tag_name,GROUP_CONCAT(DISTINCT a.article_id) AS article_id
		FROM article_dfa_tag_mapping AS a
		INNER JOIN wechat_article AS b ON a.article_id=b.wechat_article_id
		INNER JOIN dfa_tag_sensitive_word_mapping AS c ON a.tag_name=c.tag_name
		WHERE a.tag_name<>''
		AND b.article_create_time >= ?
		AND b.article_create_time <= ?
		AND a.source='wechat'
		AND c.classify_name=?
		GROUP BY a.tag_name LIMIT ?,?`

	err = global.DbMap[utils.DbNameAI].Raw(sql, startTime, endTime, classifyName, startSize, pageSize).Find(&list).Error
	return list, err
}

func GetWechatArticle(articleId, startTime, endTime string) (list []*WechatArticle, err error) {
	sql := `SELECT a.text_content
		FROM wechat_article AS a
		WHERE a.wechat_article_id IN(` + articleId + `) 
		AND a.article_create_time >= ?
		AND a.article_create_time <= ?`
	//AND a.is_summary=0`

	err = global.DbMap[utils.DbNameAI].Raw(sql, startTime, endTime).Find(&list).Error
	return list, err
}

func ModifyWechatArticleSummary(articleId string, isSummary int) (err error) {
	sql := ` UPDATE wechat_article SET is_summary=? WHERE wechat_article_id IN(` + articleId + `)`
	err = global.DbMap[utils.DbNameAI].Exec(sql, isSummary).Error
	return err
}

func GetWechatArticleTagsTotal(classifyName, startTime, endTime string) (total int, err error) {
	sql := `SELECT COUNT(1) AS total
		FROM article_dfa_tag_mapping AS a
		INNER JOIN wechat_article AS b ON a.article_id=b.wechat_article_id
		INNER JOIN dfa_tag_sensitive_word_mapping AS c ON a.tag_name=c.tag_name
		WHERE a.tag_name<>''
		AND b.article_create_time >= ?
		AND b.article_create_time <= ?
		AND a.source='wechat'
		AND c.classify_name=?
		GROUP BY a.tag_name`

	err = global.DbMap[utils.DbNameAI].Raw(sql, startTime, endTime, classifyName).Scan(&total).Error
	return total, err
}