用golang框架+mysql实现登录注册
时间: 2024-03-22 16:42:58 浏览: 136
好的,以下是使用 Golang web 框架 Gin 和 MySQL 数据库实现登录注册的示例代码:
```go
package main
import (
"database/sql"
"fmt"
"log"
"github.com/gin-gonic/gin"
_ "github.com/go-sql-driver/mysql"
)
type User struct {
ID uint `json:"id" gorm:"primary_key"`
Username string `json:"username"`
Password string `json:"password"`
}
func main() {
// 连接 MySQL 数据库
db, err := sql.Open("mysql", "root:password@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local")
if err != nil {
log.Fatal(err)
}
defer db.Close()
// 初始化 Gin 引擎
r := gin.Default()
// 注册路由
r.POST("/register", func(c *gin.Context) {
var user User
if err := c.ShouldBindJSON(&user); err != nil {
c.JSON(400, gin.H{"error": err.Error()})
return
}
if user.Username == "" || user.Password == "" {
c.JSON(400, gin.H{"error": "username and password cannot be empty"})
return
}
// 查询用户名是否已经存在
var count int
if err := db.QueryRow("SELECT COUNT(*) FROM users WHERE username = ?", user.Username).Scan(&count); err != nil {
c.JSON(500, gin.H{"error": err.Error()})
return
}
if count > 0 {
c.JSON(400, gin.H{"error": "username already exists"})
return
}
// 插入新用户信息
result, err := db.Exec("INSERT INTO users (username, password) VALUES (?, ?)", user.Username, user.Password)
if err != nil {
c.JSON(500, gin.H{"error": err.Error()})
return
}
// 返回新用户的 ID
id, _ := result.LastInsertId()
c.JSON(200, gin.H{"id": id})
})
r.POST("/login", func(c *gin.Context) {
var user User
if err := c.ShouldBindJSON(&user); err != nil {
c.JSON(400, gin.H{"error": err.Error()})
return
}
if user.Username == "" || user.Password == "" {
c.JSON(400, gin.H{"error": "username and password cannot be empty"})
return
}
// 查询用户信息
var u User
if err := db.QueryRow("SELECT id, username, password FROM users WHERE username = ?", user.Username).Scan(&u.ID, &u.Username, &u.Password); err != nil {
c.JSON(400, gin.H{"error": "invalid username or password"})
return
}
// 检查密码是否正确
if u.Password != user.Password {
c.JSON(400, gin.H{"error": "invalid username or password"})
return
}
c.JSON(200, gin.H{"id": u.ID, "username": u.Username})
})
// 启动服务
if err := r.Run(":8080"); err != nil {
log.Fatal(err)
}
}
```
在上面的代码中,我们定义了一个 `User` 结构体来表示用户信息,包含 ID、用户名和密码。首先,我们连接了 MySQL 数据库,并使用 Gin 框架注册了两个路由:`/register` 和 `/login`。`/register` 路由用于新用户注册,会检查用户名是否已经存在,如果不存在则将用户信息插入到数据库中,并返回新用户的 ID。`/login` 路由用于用户登录,会查询用户名是否存在,并检查密码是否正确,如果正确则返回用户信息。
你需要注意修改上面代码中数据库连接的用户名和密码,并且需要创建一个名为 `users` 的表,包含 `id`、`username` 和 `password` 三个字段。
你可以将以上代码保存为 `main.go` 文件,然后使用以下命令在本地启动服务器:
```
go run main.go
```
然后在浏览器或者使用 curl 工具测试 `/register` 和 `/login` 接口,例如:
```
curl -X POST -H "Content-Type: application/json" -d '{"username": "test", "password": "123456"}' http://localhost:8080/register
curl -X POST -H "Content-Type: application/json" -d '{"username": "test", "password": "123456"}' http://localhost:8080/login
```
以上命令中的 `test` 和 `123456` 分别为用户名和密码。