golang用户名密码认证

admin

0
文章

0
评论

2024-10-10 16:28:31 编程来源：ZONE.CI 全球网 0 阅读模式

在现代互联网应用中，用户认证是一个非常重要的功能。无论是网站、移动应用还是后端接口，都需要对用户进行验证。而用户名密码认证是最为常用和普遍的一种认证方式。本文将介绍如何使用Golang实现用户名密码认证。

准备工作

在开始之前，我们需要安装并配置好Golang开发环境。确保你已经正确安装了Golang，并设置好了相关的环境变量。如果你还没有安装Golang，请前往官方网站下载并安装最新版本。

另外，我们还需要一个数据库来存储用户的信息和密码。这里我们选择MySQL作为示例数据库。确保你已经安装了MySQL，并且能够连接到数据库。

创建数据库表

首先，我们需要创建一个名为users的表，用于存储用户的信息。表的结构如下：

CREATE TABLE users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    username VARCHAR(50) NOT NULL,
    password VARCHAR(100) NOT NULL
);

编写代码

接下来，我们开始编写Go代码。首先，我们需要导入必要的依赖包。在Go中，我们可以使用database/sql标准库来操作数据库。同时，我们还需要选择合适的数据库驱动，这里我们选择go-sql-driver/mysql。

import (
    "database/sql"
    _ "github.com/go-sql-driver/mysql"
)

接下来，我们需要建立与数据库的连接。通过sql.Open()函数，我们可以使用MySQL驱动来创建一个数据库连接。

db, err := sql.Open("mysql", "root:password@tcp(127.0.0.1:3306)/dbname")
if err != nil {
    log.Fatal(err)
}
defer db.Close()

在实际应用中，我们可以将连接参数配置在配置文件中，通过读取配置文件来获取连接参数。

现在，我们可以开始编写认证的逻辑了。首先，我们需要编写一个处理登录请求的函数。

func handleLogin(w http.ResponseWriter, r *http.Request) {
    // 解析POST请求的用户名和密码
    err := r.ParseForm()
    if err != nil {
        http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
        return
    }
    username := r.PostForm.Get("username")
    password := r.PostForm.Get("password")

    // 查询数据库，验证用户名和密码
    var storedPassword string
    err = db.QueryRow("SELECT password FROM users WHERE username = ?", username).Scan(&storedPassword)
    if err != nil {
        if err == sql.ErrNoRows {
            http.Error(w, "Invalid username or password", http.StatusUnauthorized)
        } else {
            http.Error(w, http.StatusText(http.StatusInternalServerError), http.StatusInternalServerError)
        }
        return
    }

    // 验证密码
    err = bcrypt.CompareHashAndPassword([]byte(storedPassword), []byte(password))
    if err != nil {
        http.Error(w, "Invalid username or password", http.StatusUnauthorized)
        return
    }

    // 验证通过，生成Token
    token := generateToken(username)

    // 返回Token给客户端
    w.Write([]byte(token))
}