121 lines
4.3 KiB
Go
121 lines
4.3 KiB
Go
package models
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/beego/beego/v2/client/orm"
|
|
)
|
|
|
|
// CmsArticleCategory CMS 文章分类 yz_cms_article_category
|
|
type CmsArticleCategory struct {
|
|
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
|
Tid uint64 `orm:"column(tid);default(0)" json:"tid"`
|
|
Cid uint64 `orm:"column(cid);default(0)" json:"cid"`
|
|
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);auto_now_add;type(datetime)" json:"create_time"`
|
|
UpdateTime *time.Time `orm:"column(update_time);type(datetime);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"
|
|
}
|
|
|
|
// CmsArticle CMS 文章 yz_cms_article
|
|
type CmsArticle struct {
|
|
ID uint64 `orm:"column(id);pk;auto" json:"id"`
|
|
Tid uint64 `orm:"column(tid);default(0)" 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(mediumtext);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(0)" json:"status"`
|
|
Top int8 `orm:"column(top);default(0)" json:"top"`
|
|
Recommend int8 `orm:"column(recommend);default(0)" json:"recommend"`
|
|
Views int `orm:"column(views);default(0)" json:"views"`
|
|
Likes int `orm:"column(likes);default(0)" json:"likes"`
|
|
PublisherID *uint64 `orm:"column(publisher_id);null" json:"publisher_id"`
|
|
PublishTime *time.Time `orm:"column(publish_time);type(datetime);null" json:"publish_time"`
|
|
CreateTime time.Time `orm:"column(create_time);auto_now_add;type(datetime)" json:"create_time"`
|
|
UpdateTime *time.Time `orm:"column(update_time);type(datetime);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"
|
|
}
|
|
|
|
func CmsCategoryNameMap(tid uint64, ids []uint64) map[uint64]string {
|
|
out := make(map[uint64]string)
|
|
if len(ids) == 0 {
|
|
return out
|
|
}
|
|
var rows []CmsArticleCategory
|
|
qs := Orm.QueryTable(new(CmsArticleCategory)).
|
|
Filter("id__in", ids).
|
|
Filter("delete_time__isnull", true)
|
|
cond := orm.NewCondition().Or("tid", 0).Or("tid", tid)
|
|
_, _ = qs.SetCond(cond).All(&rows, "ID", "Name")
|
|
for _, r := range rows {
|
|
out[r.ID] = r.Name
|
|
}
|
|
return out
|
|
}
|
|
|
|
func CmsCategoryImageMap(tid uint64, ids []uint64) map[uint64]string {
|
|
out := make(map[uint64]string)
|
|
if len(ids) == 0 {
|
|
return out
|
|
}
|
|
var rows []CmsArticleCategory
|
|
qs := Orm.QueryTable(new(CmsArticleCategory)).
|
|
Filter("id__in", ids).
|
|
Filter("delete_time__isnull", true)
|
|
cond := orm.NewCondition().Or("tid", 0).Or("tid", tid)
|
|
_, _ = qs.SetCond(cond).All(&rows, "ID", "Image")
|
|
for _, r := range rows {
|
|
out[r.ID] = r.Image
|
|
}
|
|
return out
|
|
}
|
|
|
|
func CmsFormatTime(t *time.Time) string {
|
|
if t == nil {
|
|
return ""
|
|
}
|
|
return t.Format("2006-01-02 15:04:05")
|
|
}
|
|
|
|
func CmsSimilarArticles(tid uint64, title string, limit int) ([]orm.Params, error) {
|
|
if limit <= 0 {
|
|
limit = 5
|
|
}
|
|
var rows []CmsArticle
|
|
_, err := Orm.QueryTable(new(CmsArticle)).
|
|
Filter("tid", tid).
|
|
Filter("delete_time__isnull", true).
|
|
Filter("title__icontains", title).
|
|
Limit(limit).
|
|
All(&rows, "ID", "Title")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
out := make([]orm.Params, 0, len(rows))
|
|
for _, r := range rows {
|
|
out = append(out, orm.Params{
|
|
"id": r.ID,
|
|
"title": r.Title,
|
|
"similarity": 80,
|
|
})
|
|
}
|
|
return out, nil
|
|
}
|