feat: 添加仪表板功能，整合统计数据并优化后台管理界面

2025-09-26 11:06:23 +00:00 · 2025-09-24 12:09:16 +08:00
parent 2bcaad716d
commit 636b4d5ea0
13 changed files with 430 additions and 220 deletions
--- a/internal/service/admin.go
+++ b/internal/service/admin.go
@ -1,77 +1,110 @@
 package service

 import (
-	"github.com/snowykami/neo-blog/internal/dto"
-	"github.com/snowykami/neo-blog/internal/model"
-	"github.com/snowykami/neo-blog/internal/repo"
-	"github.com/snowykami/neo-blog/pkg/errs"
-	"gorm.io/gorm"
+  "github.com/snowykami/neo-blog/internal/dto"
+  "github.com/snowykami/neo-blog/internal/model"
+  "github.com/snowykami/neo-blog/internal/repo"
+  "github.com/snowykami/neo-blog/pkg/errs"
+  "gorm.io/gorm"
 )

 type AdminService struct{}

 func NewAdminService() *AdminService {
-	return &AdminService{}
+  return &AdminService{}
+}
+
+func (c *AdminService) GetDashboard() (map[string]any, error) {
+  var (
+    postCount, commentCount, userCount, viewCount int64
+    err                                           error
+    mustCount                                     = func(q *gorm.DB, dest *int64) {
+      if err == nil {
+        err = q.Count(dest).Error
+      }
+    }
+    mustScan = func(q *gorm.DB, dest *int64) {
+      if err == nil {
+        err = q.Scan(dest).Error
+      }
+    }
+  )
+  db := repo.GetDB()
+
+  mustCount(db.Model(&model.Comment{}), &commentCount)
+  mustCount(db.Model(&model.Post{}), &postCount)
+  mustCount(db.Model(&model.User{}), &userCount)
+  mustScan(db.Model(&model.Post{}).Select("SUM(view_count)"), &viewCount)
+
+  if err != nil {
+    return nil, err
+  }
+  return map[string]any{
+    "total_comments": commentCount,
+    "total_posts":    postCount,
+    "total_users":    userCount,
+    "total_views":    viewCount,
+  }, nil
 }

 func (c *AdminService) CreateOidcConfig(req *dto.AdminOidcConfigDto) error {
-	oidcConfig := &model.OidcConfig{
-		Name:             req.Name,
-		DisplayName:      req.DisplayName,
-		Icon:             req.Icon,
-		ClientID:         req.ClientID,
-		ClientSecret:     req.ClientSecret,
-		OidcDiscoveryUrl: req.OidcDiscoveryUrl,
-		Enabled:          req.Enabled,
-		Type:             req.Type,
-	}
-	return repo.Oidc.CreateOidcConfig(oidcConfig)
+  oidcConfig := &model.OidcConfig{
+    Name:             req.Name,
+    DisplayName:      req.DisplayName,
+    Icon:             req.Icon,
+    ClientID:         req.ClientID,
+    ClientSecret:     req.ClientSecret,
+    OidcDiscoveryUrl: req.OidcDiscoveryUrl,
+    Enabled:          req.Enabled,
+    Type:             req.Type,
+  }
+  return repo.Oidc.CreateOidcConfig(oidcConfig)
 }

 func (c *AdminService) DeleteOidcConfig(id string) error {
-	if id == "" {
-		return errs.ErrBadRequest
-	}
-	return repo.Oidc.DeleteOidcConfig(id)
+  if id == "" {
+    return errs.ErrBadRequest
+  }
+  return repo.Oidc.DeleteOidcConfig(id)
 }

 func (c *AdminService) GetOidcConfigByID(id string) (*dto.AdminOidcConfigDto, error) {
-	if id == "" {
-		return nil, errs.ErrBadRequest
-	}
-	config, err := repo.Oidc.GetOidcConfigByID(id)
-	if err != nil {
-		return nil, err
-	}
-	return config.ToAdminDto(), nil
+  if id == "" {
+    return nil, errs.ErrBadRequest
+  }
+  config, err := repo.Oidc.GetOidcConfigByID(id)
+  if err != nil {
+    return nil, err
+  }
+  return config.ToAdminDto(), nil
 }

 func (c *AdminService) ListOidcConfigs(onlyEnabled bool) ([]*dto.AdminOidcConfigDto, error) {
-	configs, err := repo.Oidc.ListOidcConfigs(onlyEnabled)
-	if err != nil {
-		return nil, err
-	}
-	var dtos []*dto.AdminOidcConfigDto
-	for _, config := range configs {
-		dtos = append(dtos, config.ToAdminDto())
-	}
-	return dtos, nil
+  configs, err := repo.Oidc.ListOidcConfigs(onlyEnabled)
+  if err != nil {
+    return nil, err
+  }
+  var dtos []*dto.AdminOidcConfigDto
+  for _, config := range configs {
+    dtos = append(dtos, config.ToAdminDto())
+  }
+  return dtos, nil
 }

 func (c *AdminService) UpdateOidcConfig(req *dto.AdminOidcConfigDto) error {
-	if req.ID == 0 {
-		return errs.ErrBadRequest
-	}
-	oidcConfig := &model.OidcConfig{
-		Model:            gorm.Model{ID: req.ID},
-		Name:             req.Name,
-		DisplayName:      req.DisplayName,
-		Icon:             req.Icon,
-		ClientID:         req.ClientID,
-		ClientSecret:     req.ClientSecret,
-		OidcDiscoveryUrl: req.OidcDiscoveryUrl,
-		Enabled:          req.Enabled,
-		Type:             req.Type,
-	}
-	return repo.Oidc.UpdateOidcConfig(oidcConfig)
+  if req.ID == 0 {
+    return errs.ErrBadRequest
+  }
+  oidcConfig := &model.OidcConfig{
+    Model:            gorm.Model{ID: req.ID},
+    Name:             req.Name,
+    DisplayName:      req.DisplayName,
+    Icon:             req.Icon,
+    ClientID:         req.ClientID,
+    ClientSecret:     req.ClientSecret,
+    OidcDiscoveryUrl: req.OidcDiscoveryUrl,
+    Enabled:          req.Enabled,
+    Type:             req.Type,
+  }
+  return repo.Oidc.UpdateOidcConfig(oidcConfig)
 }
--- a/internal/service/post.go
+++ b/internal/service/post.go
@ -1,155 +1,157 @@
 package service

 import (
-  "context"
-  "strconv"
+	"context"
+	"strconv"

-  "github.com/snowykami/neo-blog/internal/ctxutils"
-  "github.com/snowykami/neo-blog/internal/dto"
-  "github.com/snowykami/neo-blog/internal/model"
-  "github.com/snowykami/neo-blog/internal/repo"
-  "github.com/snowykami/neo-blog/pkg/errs"
+	"github.com/snowykami/neo-blog/internal/ctxutils"
+	"github.com/snowykami/neo-blog/internal/dto"
+	"github.com/snowykami/neo-blog/internal/model"
+	"github.com/snowykami/neo-blog/internal/repo"
+	"github.com/snowykami/neo-blog/pkg/errs"
 )

 type PostService struct{}

 func NewPostService() *PostService {
-  return &PostService{}
+	return &PostService{}
 }

 func (p *PostService) CreatePost(ctx context.Context, req *dto.CreateOrUpdatePostReq) (uint, error) {
-  currentUser, ok := ctxutils.GetCurrentUser(ctx)
-  if !ok {
-    return 0, errs.ErrUnauthorized
-  }
-  post := &model.Post{
-    Title:   req.Title,
-    Content: req.Content,
-    UserID:  currentUser.ID,
-    Labels: func() []model.Label {
-      labelModels := make([]model.Label, 0)
-      for _, labelID := range req.Labels {
-        labelModel, err := repo.Label.GetLabelByID(strconv.Itoa(int(labelID)))
-        if err == nil {
-          labelModels = append(labelModels, *labelModel)
-        }
-      }
-      return labelModels
-    }(),
-    IsPrivate: req.IsPrivate,
-  }
-  if err := repo.Post.CreatePost(post); err != nil {
-    return 0, err
-  }
-  return post.ID, nil
+	currentUser, ok := ctxutils.GetCurrentUser(ctx)
+	if !ok {
+		return 0, errs.ErrUnauthorized
+	}
+	post := &model.Post{
+		UserID: currentUser.ID,
+		PostBase: model.PostBase{
+			Title:   req.Title,
+			Content: req.Content,
+			Labels: func() []model.Label {
+				labelModels := make([]model.Label, 0)
+				for _, labelID := range req.Labels {
+					labelModel, err := repo.Label.GetLabelByID(strconv.Itoa(int(labelID)))
+					if err == nil {
+						labelModels = append(labelModels, *labelModel)
+					}
+				}
+				return labelModels
+			}(),
+			IsPrivate: req.IsPrivate,
+		},
+	}
+	if err := repo.Post.CreatePost(post); err != nil {
+		return 0, err
+	}
+	return post.ID, nil
 }

 func (p *PostService) DeletePost(ctx context.Context, id string) error {
-  currentUser, ok := ctxutils.GetCurrentUser(ctx)
-  if !ok {
-    return errs.ErrUnauthorized
-  }
-  if id == "" {
-    return errs.ErrBadRequest
-  }
-  post, err := repo.Post.GetPostByID(id)
-  if err != nil {
-    return errs.New(errs.ErrNotFound.Code, "post not found", err)
-  }
-  if post.UserID != currentUser.ID {
-    return errs.ErrForbidden
-  }
-  if err := repo.Post.DeletePost(id); err != nil {
-    return errs.ErrInternalServer
-  }
-  return nil
+	currentUser, ok := ctxutils.GetCurrentUser(ctx)
+	if !ok {
+		return errs.ErrUnauthorized
+	}
+	if id == "" {
+		return errs.ErrBadRequest
+	}
+	post, err := repo.Post.GetPostByID(id)
+	if err != nil {
+		return errs.New(errs.ErrNotFound.Code, "post not found", err)
+	}
+	if post.UserID != currentUser.ID {
+		return errs.ErrForbidden
+	}
+	if err := repo.Post.DeletePost(id); err != nil {
+		return errs.ErrInternalServer
+	}
+	return nil
 }

 func (p *PostService) GetPost(ctx context.Context, id string) (*dto.PostDto, error) {
-  if id == "" {
-    return nil, errs.ErrBadRequest
-  }
-  post, err := repo.Post.GetPostByID(id)
-  if err != nil {
-    return nil, errs.New(errs.ErrNotFound.Code, "post not found", err)
-  }
-  currentUser, ok := ctxutils.GetCurrentUser(ctx)
-  if post.IsPrivate && (!ok || post.UserID != currentUser.ID) {
-    return nil, errs.ErrForbidden
-  }
+	if id == "" {
+		return nil, errs.ErrBadRequest
+	}
+	post, err := repo.Post.GetPostByID(id)
+	if err != nil {
+		return nil, errs.New(errs.ErrNotFound.Code, "post not found", err)
+	}
+	currentUser, ok := ctxutils.GetCurrentUser(ctx)
+	if post.IsPrivate && (!ok || post.UserID != currentUser.ID) {
+		return nil, errs.ErrForbidden
+	}

-  return post.ToDto(), nil
+	return post.ToDto(), nil
 }

 func (p *PostService) UpdatePost(ctx context.Context, id string, req *dto.CreateOrUpdatePostReq) (uint, error) {
-  currentUser, ok := ctxutils.GetCurrentUser(ctx)
-  if !ok {
-    return 0, errs.ErrUnauthorized
-  }
-  if id == "" {
-    return 0, errs.ErrBadRequest
-  }
-  post, err := repo.Post.GetPostByID(id)
-  if err != nil {
-    return 0, errs.New(errs.ErrNotFound.Code, "post not found", err)
-  }
-  if post.UserID != currentUser.ID {
-    return 0, errs.ErrForbidden
-  }
-  post.Title = req.Title
-  post.Content = req.Content
-  post.IsPrivate = req.IsPrivate
-  post.Labels = func() []model.Label {
-    labelModels := make([]model.Label, len(req.Labels))
-    for _, labelID := range req.Labels {
-      labelModel, err := repo.Label.GetLabelByID(strconv.Itoa(int(labelID)))
-      if err == nil {
-        labelModels = append(labelModels, *labelModel)
-      }
-    }
-    return labelModels
-  }()
-  if err := repo.Post.UpdatePost(post); err != nil {
-    return 0, errs.ErrInternalServer
-  }
-  return post.ID, nil
+	currentUser, ok := ctxutils.GetCurrentUser(ctx)
+	if !ok {
+		return 0, errs.ErrUnauthorized
+	}
+	if id == "" {
+		return 0, errs.ErrBadRequest
+	}
+	post, err := repo.Post.GetPostByID(id)
+	if err != nil {
+		return 0, errs.New(errs.ErrNotFound.Code, "post not found", err)
+	}
+	if post.UserID != currentUser.ID {
+		return 0, errs.ErrForbidden
+	}
+	post.Title = req.Title
+	post.Content = req.Content
+	post.IsPrivate = req.IsPrivate
+	post.Labels = func() []model.Label {
+		labelModels := make([]model.Label, len(req.Labels))
+		for _, labelID := range req.Labels {
+			labelModel, err := repo.Label.GetLabelByID(strconv.Itoa(int(labelID)))
+			if err == nil {
+				labelModels = append(labelModels, *labelModel)
+			}
+		}
+		return labelModels
+	}()
+	if err := repo.Post.UpdatePost(post); err != nil {
+		return 0, errs.ErrInternalServer
+	}
+	return post.ID, nil
 }

 func (p *PostService) ListPosts(ctx context.Context, req *dto.ListPostReq) ([]*dto.PostDto, int64, error) {
-  postDtos := make([]*dto.PostDto, 0)
-  currentUserID, _ := ctxutils.GetCurrentUserID(ctx)
-  posts, total, err := repo.Post.ListPosts(currentUserID, req.Keywords, req.Labels, req.LabelRule, req.Page, req.Size, req.OrderBy, req.Desc)
-  if err != nil {
-    return nil, total, errs.New(errs.ErrInternalServer.Code, "failed to list posts", err)
-  }
-  for _, post := range posts {
-    postDtos = append(postDtos, post.ToDtoWithShortContent(100))
-  }
-  return postDtos, total, nil
+	postDtos := make([]*dto.PostDto, 0)
+	currentUserID, _ := ctxutils.GetCurrentUserID(ctx)
+	posts, total, err := repo.Post.ListPosts(currentUserID, req.Keywords, req.Labels, req.LabelRule, req.Page, req.Size, req.OrderBy, req.Desc)
+	if err != nil {
+		return nil, total, errs.New(errs.ErrInternalServer.Code, "failed to list posts", err)
+	}
+	for _, post := range posts {
+		postDtos = append(postDtos, post.ToDtoWithShortContent(100))
+	}
+	return postDtos, total, nil
 }

 func (p *PostService) ToggleLikePost(ctx context.Context, id string) (bool, error) {
-  currentUser, ok := ctxutils.GetCurrentUser(ctx)
-  if !ok {
-    return false, errs.ErrUnauthorized
-  }
-  if id == "" {
-    return false, errs.ErrBadRequest
-  }
-  post, err := repo.Post.GetPostByID(id)
-  if err != nil {
-    return false, errs.New(errs.ErrNotFound.Code, "post not found", err)
-  }
-  if post.UserID == currentUser.ID {
-    return false, errs.ErrForbidden
-  }
-  idInt, err := strconv.ParseUint(id, 10, 64)
-  if err != nil {
-    return false, errs.New(errs.ErrBadRequest.Code, "invalid post ID", err)
-  }
-  liked, err := repo.Post.ToggleLikePost(uint(idInt), currentUser.ID)
-  if err != nil {
-    return false, errs.ErrInternalServer
-  }
-  return liked, nil
+	currentUser, ok := ctxutils.GetCurrentUser(ctx)
+	if !ok {
+		return false, errs.ErrUnauthorized
+	}
+	if id == "" {
+		return false, errs.ErrBadRequest
+	}
+	post, err := repo.Post.GetPostByID(id)
+	if err != nil {
+		return false, errs.New(errs.ErrNotFound.Code, "post not found", err)
+	}
+	if post.UserID == currentUser.ID {
+		return false, errs.ErrForbidden
+	}
+	idInt, err := strconv.ParseUint(id, 10, 64)
+	if err != nil {
+		return false, errs.New(errs.ErrBadRequest.Code, "invalid post ID", err)
+	}
+	liked, err := repo.Post.ToggleLikePost(uint(idInt), currentUser.ID)
+	if err != nil {
+		return false, errs.ErrInternalServer
+	}
+	return liked, nil
 }