162 lines
4.5 KiB
Go
162 lines
4.5 KiB
Go
package main
|
||
|
||
import (
|
||
"database/sql"
|
||
"fmt"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
_ "github.com/lib/pq"
|
||
"golang.org/x/crypto/bcrypt"
|
||
)
|
||
|
||
// LoginRequest 登录请求参数结构
|
||
type LoginRequest struct {
|
||
Account string `json:"account" binding:"required"`
|
||
Password string `json:"password" binding:"required"`
|
||
}
|
||
|
||
// LoginResponse 登录响应结构
|
||
type LoginResponse struct {
|
||
Success bool `json:"success"`
|
||
Message string `json:"message"`
|
||
Data struct {
|
||
UserID string `json:"user_id,omitempty"`
|
||
} `json:"data,omitempty"`
|
||
}
|
||
|
||
var db *sql.DB
|
||
|
||
func main() {
|
||
log.Println("开始初始化应用程序")
|
||
|
||
// 初始化Gin引擎
|
||
r := gin.Default()
|
||
log.Println("Gin引擎初始化完成")
|
||
|
||
// 从环境变量获取数据库配置
|
||
dbHost := os.Getenv("DB_HOST")
|
||
dbPort := os.Getenv("DB_PORT")
|
||
dbUser := os.Getenv("DB_USER")
|
||
dbPassword := os.Getenv("DB_PASSWORD")
|
||
dbName := os.Getenv("DB_NAME")
|
||
log.Printf("获取数据库配置: host=%s, port=%s, user=%s, dbname=%s", dbHost, dbPort, dbUser, dbName)
|
||
|
||
// 构建数据库连接字符串
|
||
connStr := fmt.Sprintf(
|
||
"host=%s port=%s user=%s password=%s dbname=%s sslmode=disable",
|
||
dbHost, dbPort, dbUser, dbPassword, dbName,
|
||
)
|
||
log.Printf("数据库连接字符串构建完成: %s", connStr)
|
||
|
||
var err error
|
||
db, err = sql.Open("postgres", connStr)
|
||
if err != nil {
|
||
log.Fatalf("无法连接数据库: %v", err)
|
||
panic(fmt.Sprintf("无法连接数据库: %v", err))
|
||
}
|
||
defer db.Close()
|
||
log.Println("数据库连接对象创建成功")
|
||
|
||
// 验证数据库连接
|
||
if err := db.Ping(); err != nil {
|
||
log.Fatalf("数据库连接失败: %v", err)
|
||
panic(fmt.Sprintf("数据库连接失败: %v", err))
|
||
}
|
||
log.Println("数据库连接验证成功")
|
||
|
||
// 登录接口
|
||
r.POST("/user/login", loginHandler)
|
||
log.Println("登录接口注册完成: POST /user/login")
|
||
|
||
// 启动服务，监听80端口
|
||
log.Println("服务启动在80端口")
|
||
r.Run(":80")
|
||
}
|
||
|
||
// loginHandler 处理登录逻辑
|
||
func loginHandler(c *gin.Context) {
|
||
reqID := c.Request.Header.Get("X-Request-ID")
|
||
if reqID == "" {
|
||
reqID = fmt.Sprintf("%d", gin.Mode()) // 简单生成一个请求标识
|
||
}
|
||
log.Printf("[%s] 收到登录请求 from %s", reqID, c.ClientIP())
|
||
|
||
var req LoginRequest
|
||
// 绑定并验证请求参数
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
log.Printf("[%s] 请求参数绑定失败: %v, 请求体: %+v", reqID, err, c.Request.Body)
|
||
c.JSON(http.StatusBadRequest, LoginResponse{
|
||
Success: false,
|
||
Message: "账号或密码不能为空",
|
||
})
|
||
return
|
||
}
|
||
log.Printf("[%s] 请求参数绑定成功: 账号=%s", reqID, req.Account)
|
||
|
||
// 1. 判断账号和密码是否为空（双重保险）
|
||
if req.Account == "" || req.Password == "" {
|
||
log.Printf("[%s] 账号或密码为空: 账号=%s", reqID, req.Account)
|
||
c.JSON(http.StatusBadRequest, LoginResponse{
|
||
Success: false,
|
||
Message: "账号或密码不能为空",
|
||
})
|
||
return
|
||
}
|
||
|
||
// 2. 查询账号对应的密码和用户ID
|
||
var storedPassword string
|
||
var userID string
|
||
query := `
|
||
SELECT password, user_id
|
||
FROM user_account_password_view
|
||
WHERE account = $1 AND deleted = false
|
||
`
|
||
log.Printf("[%s] 执行查询: %s, 参数: %s", reqID, query, req.Account)
|
||
err := db.QueryRow(query, req.Account).Scan(&storedPassword, &userID)
|
||
switch {
|
||
case err == sql.ErrNoRows:
|
||
// 账号不存在或已被删除
|
||
log.Printf("[%s] 账号不存在或已删除: %s", reqID, req.Account)
|
||
c.JSON(http.StatusOK, LoginResponse{
|
||
Success: false,
|
||
Message: "账号不存在",
|
||
})
|
||
return
|
||
case err != nil:
|
||
log.Printf("[%s] 查询账号信息失败: %v, 账号=%s", reqID, err, req.Account)
|
||
c.JSON(http.StatusInternalServerError, LoginResponse{
|
||
Success: false,
|
||
Message: "查询账号信息失败",
|
||
})
|
||
return
|
||
}
|
||
log.Printf("[%s] 查询账号信息成功: 账号=%s, userID=%s", reqID, req.Account, userID)
|
||
|
||
// 3. 验证密码（使用bcrypt比较原始密码和存储的加密密码）
|
||
err = bcrypt.CompareHashAndPassword([]byte(storedPassword), []byte(req.Password))
|
||
if err != nil {
|
||
// 密码不匹配
|
||
log.Printf("[%s] 密码验证失败: 账号=%s, 错误=%v", reqID, req.Account, err)
|
||
c.JSON(http.StatusOK, LoginResponse{
|
||
Success: false,
|
||
Message: "密码错误",
|
||
})
|
||
return
|
||
}
|
||
log.Printf("[%s] 密码验证成功: 账号=%s", reqID, req.Account)
|
||
|
||
// 4. 登录成功
|
||
log.Printf("[%s] 登录成功: 账号=%s, userID=%s", reqID, req.Account, userID)
|
||
c.JSON(http.StatusOK, LoginResponse{
|
||
Success: true,
|
||
Message: "登录成功",
|
||
Data: struct {
|
||
UserID string `json:"user_id,omitempty"`
|
||
}{
|
||
UserID: userID,
|
||
},
|
||
})
|
||
} |