refactor: reorganize packages into a domain-based layered structure

This commit is contained in:
Leon
2025-12-05 09:59:57 +08:00
parent 34b05ad83f
commit c1862b5475
8 changed files with 83 additions and 94 deletions

View File

@@ -2,8 +2,8 @@ package main
import ( import (
"feedsystem_video_go/internal/config" "feedsystem_video_go/internal/config"
"feedsystem_video_go/internal/database" "feedsystem_video_go/internal/db"
"feedsystem_video_go/internal/router" apphttp "feedsystem_video_go/internal/http"
"log" "log"
) )
@@ -14,19 +14,20 @@ func main() {
if err != nil { if err != nil {
log.Fatalf("Failed to load config: %v", err) log.Fatalf("Failed to load config: %v", err)
} }
// Connect database // Connect database
log.Printf("Database config: %v", cfg.Database) log.Printf("Database config: %v", cfg.Database)
sqlDB, err := db.NewDB(cfg.Database)
db, err := database.NewDB(cfg.Database)
if err != nil { if err != nil {
log.Fatalf("Failed to connect database: %v", err) log.Fatalf("Failed to connect database: %v", err)
} }
if err := database.AutoMigrate(db); err != nil { if err := db.AutoMigrate(sqlDB); err != nil {
log.Fatalf("Failed to auto migrate database: %v", err) log.Fatalf("Failed to auto migrate database: %v", err)
} }
defer database.CloseDB(db) defer db.CloseDB(sqlDB)
// Set router // Set router
r := router.SetRouter(db) r := apphttp.SetRouter(sqlDB)
log.Printf("Server is running on port %s", cfg.Server.Port) log.Printf("Server is running on port %s", cfg.Server.Port)
if err := r.Run(":" + cfg.Server.Port); err != nil { if err := r.Run(":" + cfg.Server.Port); err != nil {
log.Fatalf("Failed to run server: %v", err) log.Fatalf("Failed to run server: %v", err)

View File

@@ -1,4 +1,4 @@
package model package account
type User struct { type User struct {
ID uint `gorm:"primaryKey" json:"id"` ID uint `gorm:"primaryKey" json:"id"`

View File

@@ -1,8 +1,6 @@
package repository package account
import ( import (
"feedsystem_video_go/internal/model"
"gorm.io/gorm" "gorm.io/gorm"
) )
@@ -14,7 +12,7 @@ func NewUserRepository(db *gorm.DB) *UserRepository {
return &UserRepository{db: db} return &UserRepository{db: db}
} }
func (ur *UserRepository) CreateUser(user *model.User) error { func (ur *UserRepository) CreateUser(user *User) error {
if err := ur.db.Create(user).Error; err != nil { if err := ur.db.Create(user).Error; err != nil {
return err return err
} }
@@ -22,29 +20,29 @@ func (ur *UserRepository) CreateUser(user *model.User) error {
} }
func (ur *UserRepository) RenameByID(id uint, newUsername string) error { func (ur *UserRepository) RenameByID(id uint, newUsername string) error {
if err := ur.db.Model(&model.User{}).Where("id = ?", id).Update("username", newUsername).Error; err != nil { if err := ur.db.Model(&User{}).Where("id = ?", id).Update("username", newUsername).Error; err != nil {
return err return err
} }
return nil return nil
} }
func (ur *UserRepository) ChangePassword(id uint, newPassword string) error { func (ur *UserRepository) ChangePassword(id uint, newPassword string) error {
if err := ur.db.Model(&model.User{}).Where("id = ?", id).Update("password", newPassword).Error; err != nil { if err := ur.db.Model(&User{}).Where("id = ?", id).Update("password", newPassword).Error; err != nil {
return err return err
} }
return nil return nil
} }
func (ur *UserRepository) FindByID(id uint) (*model.User, error) { func (ur *UserRepository) FindByID(id uint) (*User, error) {
var user model.User var user User
if err := ur.db.First(&user, id).Error; err != nil { if err := ur.db.First(&user, id).Error; err != nil {
return nil, err return nil, err
} }
return &user, nil return &user, nil
} }
func (ur *UserRepository) FindByUsername(username string) (*model.User, error) { func (ur *UserRepository) FindByUsername(username string) (*User, error) {
var user model.User var user User
if err := ur.db.Where("username = ?", username).First(&user).Error; err != nil { if err := ur.db.Where("username = ?", username).First(&user).Error; err != nil {
return nil, err return nil, err
} }

View File

@@ -1,19 +1,14 @@
package service package account
import (
"feedsystem_video_go/internal/model"
"feedsystem_video_go/internal/repository"
)
type UserService struct { type UserService struct {
userRepository *repository.UserRepository userRepository *UserRepository
} }
func NewUserService(userRepository *repository.UserRepository) *UserService { func NewUserService(userRepository *UserRepository) *UserService {
return &UserService{userRepository: userRepository} return &UserService{userRepository: userRepository}
} }
func (us *UserService) CreateUser(user *model.User) error { func (us *UserService) CreateUser(user *User) error {
if err := us.userRepository.CreateUser(user); err != nil { if err := us.userRepository.CreateUser(user); err != nil {
return err return err
} }
@@ -34,7 +29,7 @@ func (us *UserService) ChangePassword(id uint, newPassword string) error {
return nil return nil
} }
func (us *UserService) FindByID(id uint) (*model.User, error) { func (us *UserService) FindByID(id uint) (*User, error) {
if user, err := us.userRepository.FindByID(id); err != nil { if user, err := us.userRepository.FindByID(id); err != nil {
return nil, err return nil, err
} else { } else {
@@ -42,7 +37,7 @@ func (us *UserService) FindByID(id uint) (*model.User, error) {
} }
} }
func (us *UserService) FindByUsername(username string) (*model.User, error) { func (us *UserService) FindByUsername(username string) (*User, error) {
if user, err := us.userRepository.FindByUsername(username); err != nil { if user, err := us.userRepository.FindByUsername(username); err != nil {
return nil, err return nil, err
} else { } else {

View File

@@ -1,43 +0,0 @@
package api
type CreateUserRequest struct {
Username string `json:"username"`
Password string `json:"password"`
}
type CreateUserResponse struct {
}
type RenameByIDRequest struct {
ID uint `json:"id"`
NewUsername string `json:"new_username"`
}
type RenameByIDResponse struct {
}
type FindByIDRequest struct {
ID uint `json:"id"`
}
type FindByIDResponse struct {
ID uint `json:"id"`
Username string `json:"username"`
}
type FindByUsernameRequest struct {
Username string `json:"username"`
}
type FindByUsernameResponse struct {
ID uint `json:"id"`
Username string `json:"username"`
}
type ChangePasswordRequest struct {
ID uint `json:"id"`
NewPassword string `json:"new_password"`
}
type ChangePasswordResponse struct {
}

View File

@@ -1,8 +1,8 @@
package database package db
import ( import (
"feedsystem_video_go/internal/account"
"feedsystem_video_go/internal/config" "feedsystem_video_go/internal/config"
"feedsystem_video_go/internal/model"
"fmt" "fmt"
"gorm.io/driver/mysql" "gorm.io/driver/mysql"
@@ -22,7 +22,7 @@ func NewDB(dbcfg config.DatabaseConfig) (*gorm.DB, error) {
} }
func AutoMigrate(db *gorm.DB) error { func AutoMigrate(db *gorm.DB) error {
return db.AutoMigrate(&model.User{}) return db.AutoMigrate(&account.User{})
} }
func CloseDB(db *gorm.DB) error { func CloseDB(db *gorm.DB) error {

View File

@@ -1,27 +1,67 @@
package handler package account
import ( import (
"feedsystem_video_go/internal/api" "feedsystem_video_go/internal/account"
"feedsystem_video_go/internal/model"
"feedsystem_video_go/internal/service"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
type UserHandler struct { type UserHandler struct {
userService *service.UserService userService *account.UserService
} }
func NewUserHandler(userService *service.UserService) *UserHandler { type CreateUserRequest struct {
Username string `json:"username"`
Password string `json:"password"`
}
type CreateUserResponse struct {
}
type RenameByIDRequest struct {
ID uint `json:"id"`
NewUsername string `json:"new_username"`
}
type RenameByIDResponse struct {
}
type FindByIDRequest struct {
ID uint `json:"id"`
}
type FindByIDResponse struct {
ID uint `json:"id"`
Username string `json:"username"`
}
type FindByUsernameRequest struct {
Username string `json:"username"`
}
type FindByUsernameResponse struct {
ID uint `json:"id"`
Username string `json:"username"`
}
type ChangePasswordRequest struct {
ID uint `json:"id"`
NewPassword string `json:"new_password"`
}
type ChangePasswordResponse struct {
}
func NewUserHandler(userService *account.UserService) *UserHandler {
return &UserHandler{userService: userService} return &UserHandler{userService: userService}
} }
func (h *UserHandler) CreateUser(c *gin.Context) { func (h *UserHandler) CreateUser(c *gin.Context) {
var req api.CreateUserRequest var req CreateUserRequest
if err := c.ShouldBindJSON(&req); err != nil { if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(400, gin.H{"error": err.Error()}) c.JSON(400, gin.H{"error": err.Error()})
return return
} }
if err := h.userService.CreateUser(&model.User{ if err := h.userService.CreateUser(&account.User{
Username: req.Username, Username: req.Username,
Password: req.Password, Password: req.Password,
}); err != nil { }); err != nil {
@@ -32,7 +72,7 @@ func (h *UserHandler) CreateUser(c *gin.Context) {
} }
func (h *UserHandler) RenameByID(c *gin.Context) { func (h *UserHandler) RenameByID(c *gin.Context) {
var req api.RenameByIDRequest var req RenameByIDRequest
if err := c.ShouldBindJSON(&req); err != nil { if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(400, gin.H{"error": err.Error()}) c.JSON(400, gin.H{"error": err.Error()})
return return
@@ -45,7 +85,7 @@ func (h *UserHandler) RenameByID(c *gin.Context) {
} }
func (h *UserHandler) ChangePassword(c *gin.Context) { func (h *UserHandler) ChangePassword(c *gin.Context) {
var req api.ChangePasswordRequest var req ChangePasswordRequest
if err := c.ShouldBindJSON(&req); err != nil { if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(400, gin.H{"error": err.Error()}) c.JSON(400, gin.H{"error": err.Error()})
return return
@@ -58,7 +98,7 @@ func (h *UserHandler) ChangePassword(c *gin.Context) {
} }
func (h *UserHandler) FindByID(c *gin.Context) { func (h *UserHandler) FindByID(c *gin.Context) {
var req api.FindByIDRequest var req FindByIDRequest
if err := c.ShouldBindJSON(&req); err != nil { if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(400, gin.H{"error": err.Error()}) c.JSON(400, gin.H{"error": err.Error()})
return return
@@ -72,7 +112,7 @@ func (h *UserHandler) FindByID(c *gin.Context) {
} }
func (h *UserHandler) FindByUsername(c *gin.Context) { func (h *UserHandler) FindByUsername(c *gin.Context) {
var req api.FindByUsernameRequest var req FindByUsernameRequest
if err := c.ShouldBindJSON(&req); err != nil { if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(400, gin.H{"error": err.Error()}) c.JSON(400, gin.H{"error": err.Error()})
return return

View File

@@ -1,9 +1,7 @@
package router package http
import ( import (
"feedsystem_video_go/internal/handler" "feedsystem_video_go/internal/account"
"feedsystem_video_go/internal/repository"
"feedsystem_video_go/internal/service"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"gorm.io/gorm" "gorm.io/gorm"
@@ -12,9 +10,9 @@ import (
func SetRouter(db *gorm.DB) *gin.Engine { func SetRouter(db *gorm.DB) *gin.Engine {
r := gin.Default() r := gin.Default()
userRepository := repository.NewUserRepository(db) userRepository := account.NewUserRepository(db)
userService := service.NewUserService(userRepository) userService := account.NewUserService(userRepository)
userHandler := handler.NewUserHandler(userService) userHandler := NewUserHandler(userService)
userGroup := r.Group("/user") userGroup := r.Group("/user")
{ {
userGroup.POST("/register", userHandler.CreateUser) userGroup.POST("/register", userHandler.CreateUser)