用户登录和接口鉴权

internal/storage/db/postgres.go

@@ -49,25 +49,25 @@ func NewPostgresStorage(connectionString string, maxOpenConns, maxIdleConns, con
 // Connect 建立与PostgreSQL数据库的连接
 // 使用GORM建立数据库连接
 func (ps *PostgresStorage) Connect() error {
-	ps.logger.Info("Connecting to PostgreSQL database")
+	ps.logger.Info("正在连接PostgreSQL数据库")
 
 	var err error
 	ps.db, err = gorm.Open(postgres.Open(ps.connectionString), &gorm.Config{})
 	if err != nil {
-		ps.logger.Error(fmt.Sprintf("Failed to connect to database: %v", err))
-		return fmt.Errorf("failed to connect to database: %v", err)
+		ps.logger.Error(fmt.Sprintf("数据库连接失败: %v", err))
+		return fmt.Errorf("数据库连接失败: %v", err)
 	}
 
 	// 测试连接
 	sqlDB, err := ps.db.DB()
 	if err != nil {
-		ps.logger.Error(fmt.Sprintf("Failed to get database instance: %v", err))
-		return fmt.Errorf("failed to get database instance: %v", err)
+		ps.logger.Error(fmt.Sprintf("获取数据库实例失败: %v", err))
+		return fmt.Errorf("获取数据库实例失败: %v", err)
 	}
 
 	if err = sqlDB.Ping(); err != nil {
-		ps.logger.Error(fmt.Sprintf("Failed to ping database: %v", err))
-		return fmt.Errorf("failed to ping database: %v", err)
+		ps.logger.Error(fmt.Sprintf("数据库连接测试失败: %v", err))
+		return fmt.Errorf("数据库连接测试失败: %v", err)
 	}
 
 	// 设置连接池参数
@@ -75,7 +75,7 @@ func (ps *PostgresStorage) Connect() error {
 	sqlDB.SetMaxIdleConns(ps.maxIdleConns)
 	sqlDB.SetConnMaxLifetime(time.Duration(ps.connMaxLifetime) * time.Second)
 
-	ps.logger.Info("Successfully connected to PostgreSQL database")
+	ps.logger.Info("PostgreSQL数据库连接成功")
 	return nil
 }
 
@@ -83,19 +83,19 @@ func (ps *PostgresStorage) Connect() error {
 // 安全地关闭所有数据库连接
 func (ps *PostgresStorage) Disconnect() error {
 	if ps.db != nil {
-		ps.logger.Info("Disconnecting from PostgreSQL database")
+		ps.logger.Info("正在断开PostgreSQL数据库连接")
 
 		sqlDB, err := ps.db.DB()
 		if err != nil {
-			ps.logger.Error(fmt.Sprintf("Failed to get database instance: %v", err))
-			return fmt.Errorf("failed to get database instance: %v", err)
+			ps.logger.Error(fmt.Sprintf("获取数据库实例失败: %v", err))
+			return fmt.Errorf("获取数据库实例失败: %v", err)
 		}
 
 		if err := sqlDB.Close(); err != nil {
-			ps.logger.Error(fmt.Sprintf("Failed to close database connection: %v", err))
-			return fmt.Errorf("failed to close database connection: %v", err)
+			ps.logger.Error(fmt.Sprintf("关闭数据库连接失败: %v", err))
+			return fmt.Errorf("关闭数据库连接失败: %v", err)
 		}
-		ps.logger.Info("Successfully disconnected from PostgreSQL database")
+		ps.logger.Info("PostgreSQL数据库连接已断开")
 	}
 	return nil
 }

internal/storage/repository/user.go

@@ -0,0 +1,84 @@
+// Package repository 提供数据访问层实现
+// 包含各种数据实体的仓库接口和实现
+package repository
+
+import (
+	"fmt"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/model"
+	"golang.org/x/crypto/bcrypt"
+	"gorm.io/gorm"
+)
+
+// UserRepo 用户仓库接口
+type UserRepo interface {
+	// CreateUser 创建新用户
+	CreateUser(username, password string) (*model.User, error)
+
+	// FindByUsername 根据用户名查找用户
+	FindByUsername(username string) (*model.User, error)
+
+	// FindByID 根据ID查找用户
+	FindByID(id uint) (*model.User, error)
+}
+
+// userRepo 用户仓库实现
+type userRepo struct {
+	db *gorm.DB
+}
+
+// NewUserRepo 创建用户仓库实例
+func NewUserRepo(db *gorm.DB) UserRepo {
+	return &userRepo{
+		db: db,
+	}
+}
+
+// CreateUser 创建新用户
+func (r *userRepo) CreateUser(username, password string) (*model.User, error) {
+	// 检查用户是否已存在
+	var existingUser model.User
+	result := r.db.Where("username = ?", username).First(&existingUser)
+	if result.Error == nil {
+		return nil, fmt.Errorf("用户已存在")
+	}
+
+	// 对密码进行哈希处理
+	hashedPassword, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
+	if err != nil {
+		return nil, fmt.Errorf("密码加密失败: %v", err)
+	}
+
+	// 创建新用户
+	user := &model.User{
+		Username:     username,
+		PasswordHash: string(hashedPassword),
+	}
+
+	result = r.db.Create(user)
+	if result.Error != nil {
+		return nil, fmt.Errorf("用户创建失败: %v", result.Error)
+	}
+
+	return user, nil
+}
+
+// FindByUsername 根据用户名查找用户
+func (r *userRepo) FindByUsername(username string) (*model.User, error) {
+	var user model.User
+	result := r.db.Where("username = ?", username).First(&user)
+	if result.Error != nil {
+		return nil, result.Error
+	}
+	return &user, nil
+}
+
+// FindByID 根据ID查找用户
+func (r *userRepo) FindByID(id uint) (*model.User, error) {
+	var user model.User
+	result := r.db.First(&user, id)
+	if result.Error != nil {
+		return nil, result.Error
+	}
+	return &user, nil
+}