193 lines
7.4 KiB
Go
193 lines
7.4 KiB
Go
package models
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/beego/beego/v2/client/orm"
|
|
)
|
|
|
|
// CmsArticle CMS文章表: yz_cms_article
|
|
type CmsArticle struct {
|
|
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
|
Tid uint64 `orm:"column(tid)" json:"tid"`
|
|
Title string `orm:"column(title);size(255)" json:"title"`
|
|
Author string `orm:"column(author);size(100);default()" json:"author"`
|
|
CateID uint64 `orm:"column(cate_id);default(0)" json:"cate_id"`
|
|
Content string `orm:"column(content);type(text);null" json:"content"`
|
|
Desc string `orm:"column(desc);size(500);default()" json:"desc"`
|
|
Image string `orm:"column(image);size(500);default()" json:"image"`
|
|
IsTrans int8 `orm:"column(is_trans);default(0)" json:"is_trans"`
|
|
TransURL *string `orm:"column(transurl);size(500);null" json:"transurl"`
|
|
Status int8 `orm:"column(status);default(1)" json:"status"` // 1草稿 2已发布 3已下架
|
|
Views int64 `orm:"column(views);default(0)" json:"views"`
|
|
Likes int64 `orm:"column(likes);default(0)" json:"likes"`
|
|
Top int8 `orm:"column(top);default(0)" json:"top"`
|
|
Recommend int8 `orm:"column(recommend);default(0)" json:"recommend"`
|
|
PublishTime *time.Time `orm:"column(publish_time);type(datetime);null" json:"publish_time"`
|
|
PublisherID uint64 `orm:"column(publisher_id);default(0)" json:"publisher_id"`
|
|
CreateTime time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
|
|
UpdateTime *time.Time `orm:"column(update_time);type(datetime);auto_now;null" json:"update_time"`
|
|
DeleteTime *time.Time `orm:"column(delete_time);type(datetime);null" json:"delete_time"`
|
|
}
|
|
|
|
func (m *CmsArticle) TableName() string {
|
|
return "yz_cms_article"
|
|
}
|
|
|
|
// CmsArticleCategory CMS文章分类表: yz_cms_article_category
|
|
type CmsArticleCategory struct {
|
|
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
|
Tid uint64 `orm:"column(tid)" json:"tid"`
|
|
Cid uint64 `orm:"column(cid);default(0)" json:"cid"` // 父级分类ID
|
|
Name string `orm:"column(name);size(100)" json:"name"`
|
|
Image string `orm:"column(image);size(500);default()" json:"image"`
|
|
Desc string `orm:"column(desc);size(500);default()" json:"desc"`
|
|
Sort int `orm:"column(sort);default(0)" json:"sort"`
|
|
Status int8 `orm:"column(status);default(1)" json:"status"`
|
|
CreateTime time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
|
|
UpdateTime *time.Time `orm:"column(update_time);type(datetime);auto_now;null" json:"update_time"`
|
|
DeleteTime *time.Time `orm:"column(delete_time);type(datetime);null" json:"delete_time"`
|
|
}
|
|
|
|
func (m *CmsArticleCategory) TableName() string {
|
|
return "yz_cms_article_category"
|
|
}
|
|
|
|
// EnsureCmsArticleTables 确保 CMS 文章相关表存在。
|
|
func EnsureCmsArticleTables() error {
|
|
sqls := []string{
|
|
`CREATE TABLE IF NOT EXISTS ` + "`yz_cms_article_category`" + ` (
|
|
` + "`id`" + ` bigint unsigned NOT NULL AUTO_INCREMENT,
|
|
` + "`tid`" + ` bigint unsigned NOT NULL DEFAULT 0,
|
|
` + "`cid`" + ` bigint unsigned NOT NULL DEFAULT 0,
|
|
` + "`name`" + ` varchar(100) NOT NULL DEFAULT '',
|
|
` + "`image`" + ` varchar(500) NOT NULL DEFAULT '',
|
|
` + "`desc`" + ` varchar(500) NOT NULL DEFAULT '',
|
|
` + "`sort`" + ` int NOT NULL DEFAULT 0,
|
|
` + "`status`" + ` tinyint NOT NULL DEFAULT 1,
|
|
` + "`create_time`" + ` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
` + "`update_time`" + ` datetime NULL DEFAULT NULL,
|
|
` + "`delete_time`" + ` datetime NULL DEFAULT NULL,
|
|
PRIMARY KEY (` + "`id`" + `),
|
|
KEY ` + "`idx_tid`" + ` (` + "`tid`" + `),
|
|
KEY ` + "`idx_tid_cid`" + ` (` + "`tid`" + `,` + "`cid`" + `),
|
|
KEY ` + "`idx_delete_time`" + ` (` + "`delete_time`" + `)
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci`,
|
|
`CREATE TABLE IF NOT EXISTS ` + "`yz_cms_article`" + ` (
|
|
` + "`id`" + ` bigint unsigned NOT NULL AUTO_INCREMENT,
|
|
` + "`tid`" + ` bigint unsigned NOT NULL DEFAULT 0,
|
|
` + "`title`" + ` varchar(255) NOT NULL DEFAULT '',
|
|
` + "`author`" + ` varchar(100) NOT NULL DEFAULT '',
|
|
` + "`cate_id`" + ` bigint unsigned NOT NULL DEFAULT 0,
|
|
` + "`content`" + ` longtext NULL,
|
|
` + "`desc`" + ` varchar(500) NOT NULL DEFAULT '',
|
|
` + "`image`" + ` varchar(500) NOT NULL DEFAULT '',
|
|
` + "`is_trans`" + ` tinyint NOT NULL DEFAULT 0,
|
|
` + "`transurl`" + ` varchar(500) NULL DEFAULT NULL,
|
|
` + "`status`" + ` tinyint NOT NULL DEFAULT 1,
|
|
` + "`views`" + ` bigint NOT NULL DEFAULT 0,
|
|
` + "`likes`" + ` bigint NOT NULL DEFAULT 0,
|
|
` + "`top`" + ` tinyint NOT NULL DEFAULT 0,
|
|
` + "`recommend`" + ` tinyint NOT NULL DEFAULT 0,
|
|
` + "`publish_time`" + ` datetime NULL DEFAULT NULL,
|
|
` + "`publisher_id`" + ` bigint unsigned NOT NULL DEFAULT 0,
|
|
` + "`create_time`" + ` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
` + "`update_time`" + ` datetime NULL DEFAULT NULL,
|
|
` + "`delete_time`" + ` datetime NULL DEFAULT NULL,
|
|
PRIMARY KEY (` + "`id`" + `),
|
|
KEY ` + "`idx_tid`" + ` (` + "`tid`" + `),
|
|
KEY ` + "`idx_tid_cate`" + ` (` + "`tid`" + `,` + "`cate_id`" + `),
|
|
KEY ` + "`idx_tid_status`" + ` (` + "`tid`" + `,` + "`status`" + `),
|
|
KEY ` + "`idx_delete_time`" + ` (` + "`delete_time`" + `)
|
|
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci`,
|
|
}
|
|
|
|
for _, sqlStr := range sqls {
|
|
if _, err := Orm.Raw(sqlStr).Exec(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CmsFormatTime 格式化 CMS 可空时间。
|
|
func CmsFormatTime(t *time.Time) string {
|
|
if t == nil || t.IsZero() {
|
|
return ""
|
|
}
|
|
return t.Format("2006-01-02 15:04:05")
|
|
}
|
|
|
|
// CmsCategoryNameMap 批量获取分类ID到分类名称的映射。
|
|
func CmsCategoryNameMap(tid uint64, cateIDs []uint64) map[uint64]string {
|
|
result := make(map[uint64]string)
|
|
if tid == 0 || len(cateIDs) == 0 {
|
|
return result
|
|
}
|
|
|
|
seen := make(map[uint64]bool)
|
|
ids := make([]uint64, 0, len(cateIDs))
|
|
for _, id := range cateIDs {
|
|
if id == 0 || seen[id] {
|
|
continue
|
|
}
|
|
seen[id] = true
|
|
ids = append(ids, id)
|
|
}
|
|
if len(ids) == 0 {
|
|
return result
|
|
}
|
|
|
|
var rows []CmsArticleCategory
|
|
_, err := Orm.QueryTable(new(CmsArticleCategory)).
|
|
Filter("tid", tid).
|
|
Filter("id__in", ids).
|
|
Filter("delete_time__isnull", true).
|
|
All(&rows, "id", "name")
|
|
if err != nil && err != orm.ErrNoRows {
|
|
return result
|
|
}
|
|
for _, row := range rows {
|
|
result[row.ID] = row.Name
|
|
}
|
|
return result
|
|
}
|
|
|
|
// CmsSimilarArticles 根据标题查找相似文章，用于创建文章时提示重复内容。
|
|
func CmsSimilarArticles(tid uint64, title string, limit int) ([]map[string]interface{}, error) {
|
|
title = strings.TrimSpace(title)
|
|
if tid == 0 || title == "" {
|
|
return []map[string]interface{}{}, nil
|
|
}
|
|
if limit <= 0 {
|
|
limit = 5
|
|
}
|
|
|
|
var rows []CmsArticle
|
|
_, err := Orm.QueryTable(new(CmsArticle)).
|
|
Filter("tid", tid).
|
|
Filter("title__icontains", title).
|
|
Filter("delete_time__isnull", true).
|
|
OrderBy("-id").
|
|
Limit(limit).
|
|
All(&rows, "id", "title", "cate_id", "status", "create_time")
|
|
if err != nil && err != orm.ErrNoRows {
|
|
return nil, err
|
|
}
|
|
|
|
out := make([]map[string]interface{}, 0, len(rows))
|
|
for _, row := range rows {
|
|
out = append(out, map[string]interface{}{
|
|
"id": row.ID,
|
|
"title": row.Title,
|
|
"cate_id": row.CateID,
|
|
"status": row.Status,
|
|
"create_time": row.CreateTime.Format("2006-01-02 15:04:05"),
|
|
"similarity": fmt.Sprintf("标题包含“%s”", title),
|
|
})
|
|
}
|
|
return out, nil
|
|
}
|