39a42695d3
主要变更: - 采用 go-zero 框架替代 Gin，提升开发效率 - 重构项目结构，API 文件模块化组织 - 将 model 移至 api/internal/model 目录 - 移除 common 包，改为标准 pkg 目录结构 - 实现统一的仓储模式，支持配置驱动数据库切换 - 简化测试策略，专注 API 集成测试 - 更新 CLAUDE.md 文档，提供详细的开发指导 技术栈更新: - 框架: Gin → go-zero v1.6.0+ - 代码生成: 引入 goctl 工具 - 架构模式: 四层架构 → go-zero 三层架构 (Handler→Logic→Model) - 项目布局: 遵循 Go 社区标准和 go-zero 最佳实践
78 lines
1.6 KiB
Go
78 lines
1.6 KiB
Go
package postgres
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
"gorm.io/gorm"
|
|
"gorm.io/driver/postgres"
|
|
"photography-backend/internal/config"
|
|
"photography-backend/internal/model/entity"
|
|
)
|
|
|
|
// Database 数据库连接
|
|
type Database struct {
|
|
DB *gorm.DB
|
|
}
|
|
|
|
// NewDatabase 创建数据库连接
|
|
func NewDatabase(cfg *config.DatabaseConfig) (*Database, error) {
|
|
dsn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=%s",
|
|
cfg.Host,
|
|
cfg.Port,
|
|
cfg.Username,
|
|
cfg.Password,
|
|
cfg.Database,
|
|
cfg.SSLMode,
|
|
)
|
|
|
|
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to connect to database: %w", err)
|
|
}
|
|
|
|
// 获取底层sql.DB实例配置连接池
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get sql.DB instance: %w", err)
|
|
}
|
|
|
|
// 设置连接池参数
|
|
sqlDB.SetMaxOpenConns(cfg.MaxOpenConns)
|
|
sqlDB.SetMaxIdleConns(cfg.MaxIdleConns)
|
|
sqlDB.SetConnMaxLifetime(time.Duration(cfg.ConnMaxLifetime) * time.Second)
|
|
|
|
// 测试连接
|
|
if err := sqlDB.Ping(); err != nil {
|
|
return nil, fmt.Errorf("failed to ping database: %w", err)
|
|
}
|
|
|
|
return &Database{DB: db}, nil
|
|
}
|
|
|
|
// AutoMigrate 自动迁移数据库表结构
|
|
func (d *Database) AutoMigrate() error {
|
|
return d.DB.AutoMigrate(
|
|
&entity.User{},
|
|
&entity.Category{},
|
|
&entity.Tag{},
|
|
&entity.Photo{},
|
|
)
|
|
}
|
|
|
|
// Close 关闭数据库连接
|
|
func (d *Database) Close() error {
|
|
sqlDB, err := d.DB.DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return sqlDB.Close()
|
|
}
|
|
|
|
// Health 检查数据库健康状态
|
|
func (d *Database) Health() error {
|
|
sqlDB, err := d.DB.DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return sqlDB.Ping()
|
|
} |