352 lines
8.1 KiB
Go
352 lines
8.1 KiB
Go
package repositories
|
|
|
|
import (
|
|
"database/sql"
|
|
"log"
|
|
|
|
"github.com/niangaodev/art-code/config"
|
|
"github.com/niangaodev/art-code/models"
|
|
)
|
|
|
|
// GetPosts 获取所有博客文章
|
|
func GetPosts() ([]models.Post, error) {
|
|
query := "SELECT id, title, category, date, excerpt, content, read_count, is_published, created_at, updated_at FROM posts WHERE is_published = 1 ORDER BY date DESC"
|
|
rows, err := config.DB.Query(query)
|
|
if err != nil {
|
|
log.Printf("Error querying posts: %v", err)
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var posts []models.Post
|
|
for rows.Next() {
|
|
var post models.Post
|
|
if err := rows.Scan(
|
|
&post.ID,
|
|
&post.Title,
|
|
&post.Category,
|
|
&post.Date,
|
|
&post.Excerpt,
|
|
&post.Content,
|
|
&post.ReadCount,
|
|
&post.IsPublished,
|
|
&post.CreatedAt,
|
|
&post.UpdatedAt,
|
|
); err != nil {
|
|
log.Printf("Error scanning post: %v", err)
|
|
continue
|
|
}
|
|
posts = append(posts, post)
|
|
}
|
|
|
|
return posts, nil
|
|
}
|
|
|
|
// GetPostByID 根据ID获取博客文章
|
|
func GetPostByID(id string) (*models.Post, error) {
|
|
query := "SELECT id, title, category, date, excerpt, content, read_count, is_published, created_at, updated_at FROM posts WHERE id = ? AND is_published = 1"
|
|
row := config.DB.QueryRow(query, id)
|
|
|
|
var post models.Post
|
|
if err := row.Scan(
|
|
&post.ID,
|
|
&post.Title,
|
|
&post.Category,
|
|
&post.Date,
|
|
&post.Excerpt,
|
|
&post.Content,
|
|
&post.ReadCount,
|
|
&post.IsPublished,
|
|
&post.CreatedAt,
|
|
&post.UpdatedAt,
|
|
); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
log.Printf("Error scanning post by ID: %v", err)
|
|
return nil, err
|
|
}
|
|
|
|
// 更新阅读量
|
|
updateReadCountQuery := "UPDATE posts SET read_count = read_count + 1 WHERE id = ?"
|
|
if _, err := config.DB.Exec(updateReadCountQuery, id); err != nil {
|
|
log.Printf("Error updating post read count: %v", err)
|
|
}
|
|
|
|
return &post, nil
|
|
}
|
|
|
|
// GetAllPosts 获取所有博客文章（包括未发布的）
|
|
func GetAllPosts() ([]models.Post, error) {
|
|
query := "SELECT id, title, category, date, excerpt, content, read_count, is_published, created_at, updated_at FROM posts ORDER BY date DESC"
|
|
rows, err := config.DB.Query(query)
|
|
if err != nil {
|
|
log.Printf("Error querying all posts: %v", err)
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var posts []models.Post
|
|
for rows.Next() {
|
|
var post models.Post
|
|
if err := rows.Scan(
|
|
&post.ID,
|
|
&post.Title,
|
|
&post.Category,
|
|
&post.Date,
|
|
&post.Excerpt,
|
|
&post.Content,
|
|
&post.ReadCount,
|
|
&post.IsPublished,
|
|
&post.CreatedAt,
|
|
&post.UpdatedAt,
|
|
); err != nil {
|
|
log.Printf("Error scanning post: %v", err)
|
|
continue
|
|
}
|
|
posts = append(posts, post)
|
|
}
|
|
|
|
return posts, nil
|
|
}
|
|
|
|
// CreatePost 创建博客文章
|
|
func CreatePost(post *models.Post) error {
|
|
query := `
|
|
INSERT INTO posts (id, title, category, date, excerpt, content, is_published, created_at, updated_at)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, NOW(), NOW())
|
|
`
|
|
_, err := config.DB.Exec(
|
|
query,
|
|
post.ID,
|
|
post.Title,
|
|
post.Category,
|
|
post.Date,
|
|
post.Excerpt,
|
|
post.Content,
|
|
post.IsPublished,
|
|
)
|
|
if err != nil {
|
|
log.Printf("Error creating post: %v", err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UpdatePost 更新博客文章
|
|
func UpdatePost(post *models.Post) error {
|
|
query := `
|
|
UPDATE posts SET title = ?, category = ?, date = ?, excerpt = ?, content = ?, is_published = ?, updated_at = NOW()
|
|
WHERE id = ?
|
|
`
|
|
_, err := config.DB.Exec(
|
|
query,
|
|
post.Title,
|
|
post.Category,
|
|
post.Date,
|
|
post.Excerpt,
|
|
post.Content,
|
|
post.IsPublished,
|
|
post.ID,
|
|
)
|
|
if err != nil {
|
|
log.Printf("Error updating post: %v", err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// DeletePost 删除博客文章
|
|
func DeletePost(id string) error {
|
|
query := "DELETE FROM posts WHERE id = ?"
|
|
_, err := config.DB.Exec(query, id)
|
|
if err != nil {
|
|
log.Printf("Error deleting post: %v", err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetPostCount 获取文章总数
|
|
func GetPostCount() (int, error) {
|
|
var count int
|
|
query := "SELECT COUNT(*) FROM posts"
|
|
row := config.DB.QueryRow(query)
|
|
|
|
err := row.Scan(&count)
|
|
if err != nil {
|
|
log.Printf("Error getting post count: %v", err)
|
|
return 0, err
|
|
}
|
|
|
|
return count, nil
|
|
}
|
|
|
|
// BuildPostResponse 构建博客文章响应
|
|
func BuildPostResponse(post *models.Post, includeContent bool) *models.PostResponse {
|
|
response := &models.PostResponse{
|
|
ID: post.ID,
|
|
Title: post.Title,
|
|
Category: post.Category,
|
|
Date: post.Date.Format("2006-01-02"),
|
|
Excerpt: post.Excerpt,
|
|
}
|
|
|
|
if includeContent {
|
|
response.Content = post.Content
|
|
}
|
|
|
|
return response
|
|
}
|
|
|
|
// BuildPostsResponse 构建博客文章列表响应
|
|
func BuildPostsResponse(posts []models.Post) []models.PostResponse {
|
|
var responses []models.PostResponse
|
|
for _, post := range posts {
|
|
responses = append(responses, *BuildPostResponse(&post, false))
|
|
}
|
|
return responses
|
|
}
|
|
|
|
// SavePostHistory 保存文章历史记录
|
|
func SavePostHistory(post *models.Post, modifiedBy uint) error {
|
|
// 获取当前最大版本号
|
|
var maxVersion uint
|
|
query := "SELECT COALESCE(MAX(version), 0) FROM post_history WHERE post_id = ?"
|
|
if err := config.DB.QueryRow(query, post.ID).Scan(&maxVersion); err != nil {
|
|
log.Printf("Error getting max version: %v", err)
|
|
return err
|
|
}
|
|
|
|
// 插入新的历史记录
|
|
insertQuery := `
|
|
INSERT INTO post_history (
|
|
post_id, version, title, category, date, excerpt, content,
|
|
is_published, modified_by, modified_at
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, NOW())
|
|
`
|
|
_, err := config.DB.Exec(
|
|
insertQuery,
|
|
post.ID,
|
|
maxVersion+1,
|
|
post.Title,
|
|
post.Category,
|
|
post.Date,
|
|
post.Excerpt,
|
|
post.Content,
|
|
post.IsPublished,
|
|
modifiedBy,
|
|
)
|
|
if err != nil {
|
|
log.Printf("Error saving post history: %v", err)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetPostHistory 获取文章历史记录
|
|
func GetPostHistory(postID string) ([]models.PostHistory, error) {
|
|
query := `
|
|
SELECT id, post_id, version, title, category, date, excerpt, content,
|
|
is_published, modified_by, modified_at, created_at
|
|
FROM post_history
|
|
WHERE post_id = ?
|
|
ORDER BY version DESC
|
|
`
|
|
rows, err := config.DB.Query(query, postID)
|
|
if err != nil {
|
|
log.Printf("Error querying post history: %v", err)
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var history []models.PostHistory
|
|
for rows.Next() {
|
|
var h models.PostHistory
|
|
if err := rows.Scan(
|
|
&h.ID,
|
|
&h.PostID,
|
|
&h.Version,
|
|
&h.Title,
|
|
&h.Category,
|
|
&h.Date,
|
|
&h.Excerpt,
|
|
&h.Content,
|
|
&h.IsPublished,
|
|
&h.ModifiedBy,
|
|
&h.ModifiedAt,
|
|
&h.CreatedAt,
|
|
); err != nil {
|
|
log.Printf("Error scanning post history: %v", err)
|
|
continue
|
|
}
|
|
history = append(history, h)
|
|
}
|
|
|
|
return history, nil
|
|
}
|
|
|
|
// GetPostHistoryByVersion 获取指定版本的文章历史记录
|
|
func GetPostHistoryByVersion(postID string, version uint) (*models.PostHistory, error) {
|
|
query := `
|
|
SELECT id, post_id, version, title, category, date, excerpt, content,
|
|
is_published, modified_by, modified_at, created_at
|
|
FROM post_history
|
|
WHERE post_id = ? AND version = ?
|
|
`
|
|
row := config.DB.QueryRow(query, postID, version)
|
|
|
|
var h models.PostHistory
|
|
if err := row.Scan(
|
|
&h.ID,
|
|
&h.PostID,
|
|
&h.Version,
|
|
&h.Title,
|
|
&h.Category,
|
|
&h.Date,
|
|
&h.Excerpt,
|
|
&h.Content,
|
|
&h.IsPublished,
|
|
&h.ModifiedBy,
|
|
&h.ModifiedAt,
|
|
&h.CreatedAt,
|
|
); err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
log.Printf("Error scanning post history by version: %v", err)
|
|
return nil, err
|
|
}
|
|
|
|
return &h, nil
|
|
}
|
|
|
|
// BuildPostHistoryResponse 构建文章历史记录响应
|
|
func BuildPostHistoryResponse(history *models.PostHistory) *models.PostHistoryResponse {
|
|
return &models.PostHistoryResponse{
|
|
ID: history.ID,
|
|
PostID: history.PostID,
|
|
Version: history.Version,
|
|
Title: history.Title,
|
|
Category: history.Category,
|
|
Date: history.Date.Format("2006-01-02"),
|
|
IsPublished: history.IsPublished,
|
|
ModifiedBy: history.ModifiedBy,
|
|
ModifiedAt: history.ModifiedAt.Format("2006-01-02 15:04:05"),
|
|
CreatedAt: history.CreatedAt.Format("2006-01-02 15:04:05"),
|
|
}
|
|
}
|
|
|
|
// BuildPostHistoryResponses 构建文章历史记录列表响应
|
|
func BuildPostHistoryResponses(history []models.PostHistory) []models.PostHistoryResponse {
|
|
var responses []models.PostHistoryResponse
|
|
for _, h := range history {
|
|
responses = append(responses, *BuildPostHistoryResponse(&h))
|
|
}
|
|
return responses
|
|
}
|