用户登录和接口鉴权
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user