-
Notifications
You must be signed in to change notification settings - Fork 0
/
Blog.go
395 lines (336 loc) · 11.4 KB
/
Blog.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
package models
import (
"bytes"
"fmt"
"github.com/PuerkitoBio/goquery"
"github.com/astaxie/beego"
"github.com/astaxie/beego/orm"
"github.com/lifei6671/mindoc/cache"
"github.com/lifei6671/mindoc/conf"
"github.com/lifei6671/mindoc/utils"
"strings"
"time"
)
//博文表
type Blog struct {
BlogId int `orm:"pk;auto;unique;column(blog_id)" json:"blog_id"`
//文章标题
BlogTitle string `orm:"column(blog_title);size(500)" json:"blog_title"`
//文章标识
BlogIdentify string `orm:"column(blog_identify);size(100);unique" json:"blog_identify"`
//排序序号
OrderIndex int `orm:"column(order_index);type(int);default(0)" json:"order_index"`
//所属用户
MemberId int `orm:"column(member_id);type(int);default(0);index" json:"member_id"`
//用户头像
MemberAvatar string `orm:"-" json:"member_avatar"`
//文章类型:0 普通文章/1 链接文章
BlogType int `orm:"column(blog_type);type(int);default(0)" json:"blog_type"`
//链接到的项目中的文档ID
DocumentId int `orm:"column(document_id);type(int);default(0)" json:"document_id"`
//文章的标识
DocumentIdentify string `orm:"-" json:"document_identify"`
//关联文档的项目标识
BookIdentify string `orm:"-" json:"book_identify"`
//关联文档的项目ID
BookId int `orm:"-" json:"book_id"`
//文章摘要
BlogExcerpt string `orm:"column(blog_excerpt);size(1500)" json:"blog_excerpt"`
//文章内容
BlogContent string `orm:"column(blog_content);type(text);null" json:"blog_content"`
//发布后的文章内容
BlogRelease string `orm:"column(blog_release);type(text);null" json:"blog_release"`
//文章当前的状态,枚举enum(’publish’,’draft’,’password’)值,publish为已 发表,draft为草稿,password 为私人内容(不会被公开) 。默认为publish。
BlogStatus string `orm:"column(blog_status);size(100);default(publish)" json:"blog_status"`
//文章密码,varchar(100)值。文章编辑才可为文章设定一个密码,凭这个密码才能对文章进行重新强加或修改。
Password string `orm:"column(password);size(100)" json:"-"`
//最后修改时间
Modified time.Time `orm:"column(modify_time);type(datetime);auto_now" json:"modify_time"`
//修改人id
ModifyAt int `orm:"column(modify_at);type(int)" json:"-"`
ModifyRealName string `orm:"-" json:"modify_real_name"`
//创建时间
Created time.Time `orm:"column(create_time);type(datetime);auto_now_add" json:"create_time"`
CreateName string `orm:"-" json:"create_name"`
//版本号
Version int64 `orm:"type(bigint);column(version)" json:"version"`
//附件列表
AttachList []*Attachment `orm:"-" json:"attach_list"`
}
// 多字段唯一键
func (b *Blog) TableUnique() [][]string {
return [][]string{
{"blog_id", "blog_identify"},
}
}
// TableName 获取对应数据库表名.
func (b *Blog) TableName() string {
return "blogs"
}
// TableEngine 获取数据使用的引擎.
func (b *Blog) TableEngine() string {
return "INNODB"
}
func (b *Blog) TableNameWithPrefix() string {
return conf.GetDatabasePrefix() + b.TableName()
}
func NewBlog() *Blog {
return &Blog{
BlogStatus: "public",
}
}
//根据文章ID查询文章
func (b *Blog) Find(blogId int) (*Blog, error) {
o := orm.NewOrm()
err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).One(b)
if err != nil {
beego.Error("查询文章时失败 -> ", err)
return nil, err
}
return b.Link()
}
//从缓存中读取文章
func (b *Blog) FindFromCache(blogId int) (blog *Blog, err error) {
key := fmt.Sprintf("blog-id-%d", blogId);
var temp Blog
err = cache.Get(key, &temp);
if err == nil {
b = &temp
b.Link()
beego.Debug("从缓存读取文章成功 ->", key)
return b, nil
} else {
beego.Error("读取缓存失败 ->", err)
}
blog, err = b.Find(blogId)
if err == nil {
//默认一个小时
if err := cache.Put(key, blog, time.Hour*1); err != nil {
beego.Error("将文章存入缓存失败 ->", err)
}
}
return
}
//查找指定用户的指定文章
func (b *Blog) FindByIdAndMemberId(blogId, memberId int) (*Blog, error) {
o := orm.NewOrm()
err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).Filter("member_id", memberId).One(b)
if err != nil {
beego.Error("查询文章时失败 -> ", err)
return nil, err
}
return b.Link()
}
//根据文章标识查询文章
func (b *Blog) FindByIdentify(identify string) (*Blog, error) {
o := orm.NewOrm()
err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_identify", identify).One(b)
if err != nil {
beego.Error("查询文章时失败 -> ", err)
return nil, err
}
return b, nil
}
//获取指定文章的链接内容
func (b *Blog) Link() (*Blog, error) {
o := orm.NewOrm()
//如果是链接文章,则需要从链接的项目中查找文章内容
if b.BlogType == 1 && b.DocumentId > 0 {
doc := NewDocument()
if err := o.QueryTable(doc.TableNameWithPrefix()).Filter("document_id", b.DocumentId).One(doc, "release", "markdown", "identify", "book_id"); err != nil {
beego.Error("查询文章链接对象时出错 -> ", err)
} else {
b.DocumentIdentify = doc.Identify
b.BlogRelease = doc.Release
//目前仅支持markdown文档进行链接
b.BlogContent = doc.Markdown
book := NewBook()
if err := o.QueryTable(book.TableNameWithPrefix()).Filter("book_id", doc.BookId).One(book, "identify"); err != nil {
beego.Error("查询关联文档的项目时出错 ->", err)
} else {
b.BookIdentify = book.Identify
b.BookId = doc.BookId
}
//处理链接文档存在源文档修改时间的问题
if content, err := goquery.NewDocumentFromReader(bytes.NewBufferString(b.BlogRelease)); err == nil {
content.Find(".wiki-bottom").Remove()
if html,err := content.Html();err == nil {
b.BlogRelease = html
} else {
beego.Error("处理文章失败 ->",err)
}
}else {
beego.Error("处理文章失败 ->",err)
}
}
}
if b.ModifyAt > 0 {
member := NewMember()
if err := o.QueryTable(member.TableNameWithPrefix()).Filter("member_id", b.ModifyAt).One(member, "real_name", "account"); err == nil {
if member.RealName != "" {
b.ModifyRealName = member.RealName
} else {
b.ModifyRealName = member.Account
}
}
}
if b.MemberId > 0 {
member := NewMember()
if err := o.QueryTable(member.TableNameWithPrefix()).Filter("member_id", b.MemberId).One(member, "real_name", "account", "avatar"); err == nil {
if member.RealName != "" {
b.CreateName = member.RealName
} else {
b.CreateName = member.Account
}
b.MemberAvatar = member.Avatar
}
}
return b, nil
}
//判断指定的文章标识是否存在
func (b *Blog) IsExist(identify string) bool {
o := orm.NewOrm()
return o.QueryTable(b.TableNameWithPrefix()).Filter("blog_identify", identify).Exist()
}
//保存文章
func (b *Blog) Save(cols ...string) error {
o := orm.NewOrm()
if b.OrderIndex <= 0 {
blog := NewBlog()
if err := o.QueryTable(blog.TableNameWithPrefix()).OrderBy("-blog_id").Limit(1).One(blog, "blog_id"); err == nil {
b.OrderIndex = blog.BlogId + 1;
} else {
c, _ := o.QueryTable(b.TableNameWithPrefix()).Count()
b.OrderIndex = int(c) + 1
}
}
var err error
b.Processor().Version = time.Now().Unix()
if b.BlogId > 0 {
b.Modified = time.Now()
_, err = o.Update(b, cols...)
key := fmt.Sprintf("blog-id-%d", b.BlogId)
cache.Delete(key)
} else {
b.Created = time.Now()
_, err = o.Insert(b)
}
return err
}
//过滤文章的危险标签,处理文章外链以及图片.
func (b *Blog) Processor() *Blog {
b.BlogRelease = utils.SafetyProcessor(b.BlogRelease)
//解析文档中非本站的链接,并设置为新窗口打开
if content, err := goquery.NewDocumentFromReader(bytes.NewBufferString(b.BlogRelease)); err == nil {
content.Find("a").Each(func(i int, contentSelection *goquery.Selection) {
if src, ok := contentSelection.Attr("href"); ok {
if strings.HasPrefix(src, "http://") || strings.HasPrefix(src, "https://") {
//beego.Info(src,conf.BaseUrl,strings.HasPrefix(src,conf.BaseUrl))
if conf.BaseUrl != "" && !strings.HasPrefix(src, conf.BaseUrl) {
contentSelection.SetAttr("target", "_blank")
if html, err := content.Html(); err == nil {
b.BlogRelease = html
}
}
}
}
})
//设置图片为CDN地址
if cdnimg := beego.AppConfig.String("cdnimg"); cdnimg != "" {
content.Find("img").Each(func(i int, contentSelection *goquery.Selection) {
if src, ok := contentSelection.Attr("src"); ok && strings.HasPrefix(src, "/uploads/") {
contentSelection.SetAttr("src", utils.JoinURI(cdnimg, src))
}
})
}
}
return b
}
//分页查询文章列表
func (b *Blog) FindToPager(pageIndex, pageSize int, memberId int, status string) (blogList []*Blog, totalCount int, err error) {
o := orm.NewOrm()
offset := (pageIndex - 1) * pageSize
query := o.QueryTable(b.TableNameWithPrefix())
if memberId > 0 {
query = query.Filter("member_id", memberId)
}
if status != "" {
query = query.Filter("blog_status", status)
}
_, err = query.OrderBy("-order_index", "-blog_id").Offset(offset).Limit(pageSize).All(&blogList)
if err != nil {
if err == orm.ErrNoRows {
err = nil
}
beego.Error("获取文章列表时出错 ->", err)
return
}
count, err := query.Count()
if err != nil {
beego.Error("获取文章数量时出错 ->", err)
return nil, 0, err
}
totalCount = int(count)
for _, blog := range blogList {
if blog.BlogType == 1 {
blog.Link()
}
}
return
}
//删除文章
func (b *Blog) Delete(blogId int) error {
o := orm.NewOrm()
_, err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).Delete()
if err != nil {
beego.Error("删除文章失败 ->", err)
}
return err
}
//查询下一篇文章
func (b *Blog) QueryNext(blogId int) (*Blog, error) {
o := orm.NewOrm()
blog := NewBlog()
if err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).One(blog, "order_index"); err != nil {
beego.Error("查询文章时出错 ->", err)
return b, err
}
err := o.QueryTable(b.TableNameWithPrefix()).Filter("order_index__gte", blog.OrderIndex).Filter("blog_id__gt", blogId).OrderBy("order_index", "blog_id").One(blog)
if err != nil && err != orm.ErrNoRows {
beego.Error("查询文章时出错 ->", err)
}
return blog, err
}
//查询下一篇文章
func (b *Blog) QueryPrevious(blogId int) (*Blog, error) {
o := orm.NewOrm()
blog := NewBlog()
if err := o.QueryTable(b.TableNameWithPrefix()).Filter("blog_id", blogId).One(blog, "order_index"); err != nil {
beego.Error("查询文章时出错 ->", err)
return b, err
}
err := o.QueryTable(b.TableNameWithPrefix()).Filter("order_index__lte", blog.OrderIndex).Filter("blog_id__lt", blogId).OrderBy("-order_index", "-blog_id").One(blog)
if err != nil && err != orm.ErrNoRows {
beego.Error("查询文章时出错 ->", err)
}
return blog, err
}
//关联文章附件
func (b *Blog) LinkAttach() (err error) {
o := orm.NewOrm()
var attachList []*Attachment
//当不是关联文章时,用文章ID去查询附件
if b.BlogType != 1 || b.DocumentId <= 0 {
_, err = o.QueryTable(NewAttachment().TableNameWithPrefix()).Filter("document_id", b.BlogId).Filter("book_id", 0).All(&attachList)
if err != nil && err != orm.ErrNoRows {
beego.Error("查询文章附件时出错 ->", err)
}
} else {
_, err = o.QueryTable(NewAttachment().TableNameWithPrefix()).Filter("document_id", b.DocumentId).Filter("book_id", b.BookId).All(&attachList)
if err != nil && err != orm.ErrNoRows {
beego.Error("查询文章附件时出错 ->", err)
}
}
b.AttachList = attachList
return
}