项目初始化

dao/db.go

@@ -0,0 +1,53 @@
+package dao
+
+import (
+	"RelationshipManagement-backend/config"
+	"fmt"
+	"gorm.io/driver/mysql"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
+
+var DB *gorm.DB
+
+// InitDB 初始化数据库连接
+func InitDB(cfg *config.Config) error {
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		cfg.DB.User,
+		cfg.DB.Password,
+		cfg.DB.Host,
+		cfg.DB.Port,
+		cfg.DB.DBName,
+	)
+
+	var err error
+	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{
+		Logger: logger.Default.LogMode(logger.Info),
+	})
+
+	if err != nil {
+		return fmt.Errorf("failed to connect database: %w", err)
+	}
+
+	// 自动迁移（开发环境使用，生产环境建议使用迁移工具）
+	// AutoMigrate()
+
+	return nil
+}
+
+// AutoMigrate 自动迁移数据库表
+func AutoMigrate() error {
+	// 在这里添加需要自动迁移的模型
+	// return DB.AutoMigrate(&model.User{})
+	return nil
+}
+
+// CloseDB 关闭数据库连接
+func CloseDB() error {
+	sqlDB, err := DB.DB()
+	if err != nil {
+		return err
+	}
+	return sqlDB.Close()
+}
+

dao/user_dao.go

@@ -0,0 +1,69 @@
+package dao
+
+import (
+	"RelationshipManagement-backend/model"
+	"gorm.io/gorm"
+)
+
+// UserDAO 用户数据访问对象
+type UserDAO struct {
+	db *gorm.DB
+}
+
+// NewUserDAO 创建用户DAO实例
+func NewUserDAO(db *gorm.DB) *UserDAO {
+	return &UserDAO{db: db}
+}
+
+// Create 创建用户
+func (dao *UserDAO) Create(user *model.User) error {
+	return dao.db.Create(user).Error
+}
+
+// GetByID 根据ID获取用户
+func (dao *UserDAO) GetByID(id uint) (*model.User, error) {
+	var user model.User
+	err := dao.db.First(&user, id).Error
+	if err != nil {
+		return nil, err
+	}
+	return &user, nil
+}
+
+// GetByUsername 根据用户名获取用户
+func (dao *UserDAO) GetByUsername(username string) (*model.User, error) {
+	var user model.User
+	err := dao.db.Where("username = ?", username).First(&user).Error
+	if err != nil {
+		return nil, err
+	}
+	return &user, nil
+}
+
+// GetByEmail 根据邮箱获取用户
+func (dao *UserDAO) GetByEmail(email string) (*model.User, error) {
+	var user model.User
+	err := dao.db.Where("email = ?", email).First(&user).Error
+	if err != nil {
+		return nil, err
+	}
+	return &user, nil
+}
+
+// Update 更新用户
+func (dao *UserDAO) Update(user *model.User) error {
+	return dao.db.Save(user).Error
+}
+
+// Delete 删除用户
+func (dao *UserDAO) Delete(id uint) error {
+	return dao.db.Delete(&model.User{}, id).Error
+}
+
+// List 获取用户列表
+func (dao *UserDAO) List(offset, limit int) ([]*model.User, error) {
+	var users []*model.User
+	err := dao.db.Offset(offset).Limit(limit).Find(&users).Error
+	return users, err
+}
+