refactor: reorganize packages into a domain-based layered structure
This commit is contained in:
15
cmd/main.go
15
cmd/main.go
@@ -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)
|
||||||
|
|||||||
@@ -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"`
|
||||||
@@ -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
|
||||||
}
|
}
|
||||||
@@ -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 {
|
||||||
@@ -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 {
|
|
||||||
}
|
|
||||||
@@ -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 {
|
||||||
@@ -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
|
||||||
@@ -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)
|
||||||
Reference in New Issue
Block a user