mirror of
https://github.com/eiblog/eiblog.git
synced 2026-02-04 13:52:26 +08:00
352 lines
9.4 KiB
Go
352 lines
9.4 KiB
Go
// Package cache provides ...
|
|
package cache
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/eiblog/eiblog/pkg/cache/render"
|
|
"github.com/eiblog/eiblog/pkg/cache/store"
|
|
"github.com/eiblog/eiblog/pkg/config"
|
|
"github.com/eiblog/eiblog/pkg/internal"
|
|
"github.com/eiblog/eiblog/pkg/model"
|
|
"github.com/eiblog/eiblog/tools"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
// Ei eiblog cache
|
|
Ei *Cache
|
|
|
|
// regenerate pages chan
|
|
pagesCh = make(chan string, 2)
|
|
pageSeries = "series-md"
|
|
pageArchive = "archive-md"
|
|
)
|
|
|
|
func init() {
|
|
store, err := store.NewStore(config.Conf.Database.Driver,
|
|
config.Conf.Database.Source)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// Ei init
|
|
Ei = &Cache{
|
|
Store: store,
|
|
TagArticles: make(map[string]model.SortedArticles),
|
|
ArticlesMap: make(map[string]*model.Article),
|
|
}
|
|
err = Ei.loadOrInit()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
go Ei.regeneratePages()
|
|
go Ei.timerClean()
|
|
go Ei.timerDisqus()
|
|
}
|
|
|
|
// Cache 整站缓存
|
|
type Cache struct {
|
|
store.Store
|
|
|
|
// load from db
|
|
Blogger *model.Blogger
|
|
Account *model.Account
|
|
Articles model.SortedArticles
|
|
|
|
// auto generate
|
|
PageSeries string // page
|
|
Series model.SortedSeries
|
|
PageArchives string // page
|
|
Archives model.SortedArchives
|
|
TagArticles map[string]model.SortedArticles // tagname:articles
|
|
ArticlesMap map[string]*model.Article // slug:article
|
|
}
|
|
|
|
// // LoadInsertAccount 读取或创建账户
|
|
// LoadInsertAccount(ctx context.Context, acct *model.Account) (*model.Account, error)
|
|
// // UpdateAccount 更新账户
|
|
// UpdateAccount(ctx context.Context, name string, fields map[string]interface{}) error
|
|
//
|
|
// // LoadInsertBlogger 读取或创建博客
|
|
// LoadInsertBlogger(ctx context.Context, blogger *model.Blogger) (*model.Blogger, error)
|
|
// // UpdateBlogger 更新博客
|
|
// UpdateBlogger(ctx context.Context, fields map[string]interface{}) error
|
|
//
|
|
// // InsertSeries 创建专题
|
|
// InsertSeries(ctx context.Context, series *model.Series) error
|
|
// // RemoveSeries 删除专题
|
|
// RemoveSeries(ctx context.Context, id int) error
|
|
// // UpdateSeries 更新专题
|
|
// UpdateSeries(ctx context.Context, id int, fields map[string]interface{}) error
|
|
// // LoadAllSeries 读取所有专题
|
|
// LoadAllSeries(ctx context.Context) (model.SortedSeries, error)
|
|
//
|
|
// // InsertArticle 创建文章
|
|
// InsertArticle(ctx context.Context, article *model.Article) error
|
|
// // RemoveArticle 硬删除文章
|
|
// RemoveArticle(ctx context.Context, id int) error
|
|
// // DeleteArticle 软删除文章,放入回收箱
|
|
// DeleteArticle(ctx context.Context, id int) error
|
|
// // CleanArticles 清理回收站文章
|
|
// CleanArticles(ctx context.Context) error
|
|
// // UpdateArticle 更新文章
|
|
// UpdateArticle(ctx context.Context, id int, fields map[string]interface{}) error
|
|
// // RecoverArticle 恢复文章到草稿
|
|
// RecoverArticle(ctx context.Context, id int) error
|
|
// // LoadAllArticle 读取所有文章
|
|
// LoadAllArticle(ctx context.Context) (model.SortedArticles, error)
|
|
// // LoadTrashArticles 读取回收箱
|
|
// LoadTrashArticles(ctx context.Context) (model.SortedArticles, error)
|
|
// // LoadDraftArticles 读取草稿箱
|
|
// LoadDraftArticles(ctx context.Context) (model.SortedArticles, error)
|
|
|
|
// PageArticles 文章翻页
|
|
func (c *Cache) PageArticles(page int, pageSize int) (prev,
|
|
next int, articles []*model.Article) {
|
|
|
|
var l int
|
|
for l = len(c.Articles); l > 0; l-- {
|
|
if c.Articles[l-1].ID >= config.Conf.BlogApp.General.StartID {
|
|
break
|
|
}
|
|
}
|
|
if l == 0 {
|
|
return 0, 0, nil
|
|
}
|
|
m := l / pageSize
|
|
if d := l % pageSize; d > 0 {
|
|
m++
|
|
}
|
|
if page > m {
|
|
page = m
|
|
}
|
|
if page > 1 {
|
|
prev = page - 1
|
|
}
|
|
if page < m {
|
|
next = page + 1
|
|
}
|
|
s := (page - 1) * pageSize
|
|
e := page * pageSize
|
|
if e > l {
|
|
e = l
|
|
}
|
|
articles = c.Articles[s:e]
|
|
return
|
|
}
|
|
|
|
// loadOrInit 读取数据或初始化
|
|
func (c *Cache) loadOrInit() error {
|
|
blogapp := config.Conf.BlogApp
|
|
// blogger
|
|
blogger := &model.Blogger{
|
|
BlogName: strings.Title(blogapp.Account.Username),
|
|
SubTitle: "Rome was not built in one day.",
|
|
BeiAn: "蜀ICP备xxxxxxxx号-1",
|
|
BTitle: fmt.Sprintf("%s's Blog", strings.Title(blogapp.Account.Username)),
|
|
Copyright: `本站使用「<a href="//creativecommons.org/licenses/by/4.0/">署名 4.0 国际</a>」创作共享协议，转载请注明作者及原网址。`,
|
|
}
|
|
created, err := c.LoadInsertBlogger(context.Background(), blogger)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c.Blogger = blogger
|
|
if created { // init articles: about blogroll
|
|
about := &model.Article{
|
|
ID: 1, // 固定ID
|
|
Author: blogapp.Account.Username,
|
|
Title: "关于",
|
|
Slug: "about",
|
|
}
|
|
err = c.InsertArticle(context.Background(), about)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// 推送到 disqus
|
|
go internal.ThreadCreate(about, blogger.BTitle)
|
|
blogroll := &model.Article{
|
|
ID: 2, // 固定ID
|
|
Author: blogapp.Account.Username,
|
|
Title: "友情链接",
|
|
Slug: "blogroll",
|
|
}
|
|
err = c.InsertArticle(context.Background(), blogroll)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
// account
|
|
pwd := tools.EncryptPasswd(blogapp.Account.Password,
|
|
blogapp.Account.Password)
|
|
|
|
account := &model.Account{
|
|
Username: blogapp.Account.Username,
|
|
Password: pwd,
|
|
}
|
|
_, err = c.LoadInsertAccount(context.Background(), account)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c.Account = account
|
|
// all articles
|
|
articles, err := c.LoadAllArticle(context.Background())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for i, v := range articles {
|
|
// 渲染页面
|
|
render.GenerateExcerptMarkdown(v)
|
|
|
|
c.ArticlesMap[v.Slug] = v
|
|
// 分析文章
|
|
if v.ID < blogapp.General.StartID {
|
|
continue
|
|
}
|
|
if i > 0 {
|
|
v.Prev = Ei.Articles[i-1]
|
|
}
|
|
if Ei.Articles[i+1].ID >= blogapp.General.StartID {
|
|
v.Next = Ei.Articles[i+1]
|
|
}
|
|
c.rebuildArticle(v, false)
|
|
}
|
|
Ei.Articles = articles
|
|
// 重建专题与归档
|
|
pagesCh <- pageSeries
|
|
pagesCh <- pageArchive
|
|
return nil
|
|
}
|
|
|
|
// rebuildArticle 重建缓存tag、series、archive
|
|
func (c *Cache) rebuildArticle(article *model.Article, needSort bool) {
|
|
// tag
|
|
tags := strings.Split(article.Tags, ",")
|
|
for _, tag := range tags {
|
|
c.TagArticles[tag] = append(c.TagArticles[tag], article)
|
|
if needSort {
|
|
sort.Sort(c.TagArticles[tag])
|
|
}
|
|
}
|
|
// series
|
|
for i, series := range c.Series {
|
|
if series.ID == article.SeriesID {
|
|
c.Series[i].Articles = append(c.Series[i].Articles, article)
|
|
if needSort {
|
|
sort.Sort(c.Series[i].Articles)
|
|
pagesCh <- pageSeries // 重建专题
|
|
}
|
|
}
|
|
}
|
|
// archive
|
|
y, m, _ := article.CreatedAt.Date()
|
|
for i, archive := range c.Archives {
|
|
if ay, am, _ := archive.Time.Date(); y == ay && m == am {
|
|
c.Archives[i].Articles = append(c.Archives[i].Articles, article)
|
|
}
|
|
if needSort {
|
|
sort.Sort(c.Archives[i].Articles)
|
|
pagesCh <- pageArchive // 重建归档
|
|
}
|
|
return
|
|
}
|
|
// 新建归档
|
|
c.Archives = append(c.Archives, &model.Archive{
|
|
Time: article.CreatedAt,
|
|
Articles: model.SortedArticles{article},
|
|
})
|
|
if needSort { // 重建归档
|
|
pagesCh <- pageArchive
|
|
}
|
|
}
|
|
|
|
// regeneratePages 重新生成series,archive页面
|
|
func (c *Cache) regeneratePages() {
|
|
for {
|
|
switch page := <-pagesCh; page {
|
|
case pageSeries:
|
|
sort.Sort(c.Series)
|
|
buf := bytes.Buffer{}
|
|
buf.WriteString(c.Blogger.SeriesSay)
|
|
buf.WriteString("\n\n")
|
|
for _, series := range c.Series {
|
|
buf.WriteString(fmt.Sprintf("### %s{#toc-%d}", series.Name, series.ID))
|
|
buf.WriteByte('\n')
|
|
buf.WriteString(series.Desc)
|
|
buf.WriteString("\n\n")
|
|
for _, article := range series.Articles {
|
|
//eg. * [标题一](/post/hello-world.html) <span class="date">(Man 02, 2006)</span>
|
|
str := fmt.Sprintf(`* [%s](/post/%s.html) <span class="date">(%s)</span>`,
|
|
article.Title, article.Slug, article.CreatedAt.Format("Jan 02, 2006"))
|
|
buf.WriteString(str)
|
|
}
|
|
buf.WriteString("\n\n")
|
|
}
|
|
c.PageSeries = string(render.RenderPage(buf.Bytes()))
|
|
case pageArchive:
|
|
sort.Sort(c.Archives)
|
|
buf := bytes.Buffer{}
|
|
buf.WriteString(c.Blogger.ArchiveSay + "\n")
|
|
var (
|
|
currentYear string
|
|
gt12Month = len(Ei.Archives) > 12
|
|
)
|
|
for _, archive := range c.Archives {
|
|
if gt12Month {
|
|
year := archive.Time.Format("2006 年")
|
|
if currentYear != year {
|
|
currentYear = year
|
|
buf.WriteString(fmt.Sprintf("\n### %s\n\n", archive.Time.Format("2006 年")))
|
|
}
|
|
} else {
|
|
buf.WriteString(fmt.Sprintf("\n### %s\n\n", archive.Time.Format("2006年1月")))
|
|
}
|
|
for i, article := range archive.Articles {
|
|
if i == 0 && gt12Month {
|
|
str := fmt.Sprintf(`* *[%s](/post/%s.html) <span class="date">(%s)</span>`,
|
|
article.Title, article.Slug, article.CreatedAt.Format("Jan 02, 2006"))
|
|
buf.WriteString(str)
|
|
} else {
|
|
str := fmt.Sprintf(`* [%s](/post/%s.html) <span class="date">(%s)</span>`,
|
|
article.Title, article.Slug, article.CreatedAt.Format("Jan 02, 2006"))
|
|
buf.WriteString(str)
|
|
}
|
|
buf.WriteByte('\n')
|
|
}
|
|
}
|
|
c.PageArchives = string(render.RenderPage(buf.Bytes()))
|
|
}
|
|
}
|
|
}
|
|
|
|
// timerClean 定时清理文章
|
|
func (c *Cache) timerClean() {
|
|
dur := time.Duration(config.Conf.BlogApp.General.Clean)
|
|
ticker := time.NewTicker(dur * time.Hour)
|
|
|
|
for range ticker.C {
|
|
err := c.CleanArticles(context.Background())
|
|
if err != nil {
|
|
logrus.Error("cache.timerClean.CleanArticles: ", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// timerDisqus disqus定时操作
|
|
func (c *Cache) timerDisqus() {
|
|
dur := time.Duration(config.Conf.BlogApp.Disqus.Interval)
|
|
ticker := time.NewTicker(dur * time.Hour)
|
|
|
|
for range ticker.C {
|
|
err := internal.PostsCount(c.ArticlesMap)
|
|
if err != nil {
|
|
logrus.Error("cache.timerDisqus.PostsCount: ", err)
|
|
}
|
|
}
|
|
}
|