CREATE TABLE `user` (
  `user_id` int NOT NULL AUTO_INCREMENT,
  `user_name` varchar(100) DEFAULT NULL,
  `user_code` varchar(100) NOT NULL,
  `password` varchar(100) NOT NULL,
  PRIMARY KEY (`user_id`),
  UNIQUE KEY `user_user_code_IDX` (`user_code`) USING BTREE
) ENGINE=InnoDB AUTO_INCREMENT=15 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;

go mod init fa.com/wms

go get github.com/go-sql-driver/mysql@latest

package db

import (
	"database/sql"
	"fmt"
	"time"

	//注冊驅動器 _下劃線表示執(zhí)行驅動中的init函數，不使用其他函數
	_ "github.com/go-sql-driver/mysql"
)
// 創(chuàng)建數據庫連接
func ConnMySQL() *sql.DB {
	// 數據源名
	driverName := "mysql"
	// 用戶名root
	// 密碼1234
	// tcp協(xié)議連接
	// 數據庫地址
	// 數據庫 wms
	dataSourceName := "root" + ":" + "1234" + "@" + "tcp" + "(" + "127.0.0.1:3306" + ")" + "/" + "wms"
	db, err := sql.Open(driverName, dataSourceName)
	if err != nil {
		panic(err)
	}

	// 數據庫設置
	db.SetConnMaxLifetime(time.Minute * 10)
	db.SetConnMaxIdleTime(time.Minute * 10)
	db.SetMaxOpenConns(10)
	db.SetMaxIdleConns(10)

	// 連接測試
	err = db.Ping()
	if err != nil {
		fmt.Println("數據庫連接失敗")
		panic(err)
	}
	return db
}

// 包名
package user

import (
	"log"
	// 引入mysql數據庫連接包
	"fa.com/wms/db"
)
// 定義user類型結構體
type User struct {
	UserId   int
	UserName string
	UserCode string
	Password string
}

// 向數據庫新增一個User數據
func AddUser(u *User) {
	// 調用db包ConnMySQL()
	db := db.ConnMySQL()
	// 預編譯保存sql創(chuàng)建 statement
	stmt, err := db.Prepare("INSERT INTO `user`	(user_name, user_code, password) VALUES (?, ?, ?)")
	// err 不等空則statement創(chuàng)建失敗
	if err != nil {
		// 打印錯誤信息
		log.Fatal(err)
		// 拋出錯誤信息，阻止程序繼續(xù)運行
		panic(err)
	}
	// 在函數運行結束時關閉statement
	defer stmt.Close()
	// 執(zhí)行保存sql
	// _下劃線 表示忽略函數返回的sql.Result值
	_, err = stmt.Exec(u.UserName, u.UserCode, u.Password)
	if err != nil {
		// 打印錯誤信息
		log.Fatal(err)
		// 拋出錯誤信息，阻止程序繼續(xù)運行
		panic(err)
	}
}

func DeleteById(userId int) {
	db := db.ConnMySQL()
	// 預編譯刪除sql創(chuàng)建 statement
	stmt, err := db.Prepare("delete from `user`  where user_id = ?")
	// err 不等空則statement創(chuàng)建失敗
	if err != nil {
		// 打印錯誤信息
		log.Fatal(err)
		// 拋出錯誤信息，阻止程序繼續(xù)運行
		panic(err)
	}
	// 在函數運行結束時關閉statement
	defer stmt.Close()
	// 執(zhí)行刪除sql
	// _下劃線 表示忽略函數返回的sql.Result值
	_, err = stmt.Exec(userId)
	if err != nil {
		// 打印錯誤信息
		log.Fatal(err)
		// 拋出錯誤信息，阻止程序繼續(xù)運行
		panic(err)
	}
}

func UpdateUser(u *User) {
	db := db.ConnMySQL()
	// 預編譯更新sql創(chuàng)建 statement
	stmt, err := db.Prepare("UPDATE `user` SET user_name=?, user_code=?, password=? WHERE user_id=?")
	// err 不等空則statement創(chuàng)建失敗
	if err != nil {
		// 打印錯誤信息
		log.Fatal(err)
		// 拋出錯誤信息，阻止程序繼續(xù)運行
		panic(err)
	}
	// 在函數運行結束時關閉statement
	defer stmt.Close()
	// 執(zhí)行更新sql
	// _下劃線 表示忽略函數返回的sql.Result值
	_, err = stmt.Exec(u.UserName, u.UserCode, u.Password, u.UserId)
	if err != nil {
		// 打印錯誤信息
		log.Fatal(err)
		// 拋出錯誤信息，阻止程序繼續(xù)運行
		panic(err)
	}
}

func QueryUserByCode(code string) []User {
	// 調用db包ConnMySQL()
	db := db.ConnMySQL()
	// 預編譯查詢sql創(chuàng)建 statement
	stmt, err := db.Prepare("SELECT user_id, user_name, user_code, password FROM `user` WHERE user_code= ?")
	if err != nil {
		log.Fatal(err)
		panic(err)
	}
	defer stmt.Close()
	// 執(zhí)行查詢sql，返回查詢結果rows
	rows, err := stmt.Query(code)
	if err != nil {
		// 打印錯誤信息
		log.Fatal(err)
		// 拋出錯誤信息，阻止程序繼續(xù)運行
		panic(err)
	}
	// 定義User切片
	s := make([]User, 0)
	// 遍歷rows
	for rows.Next() {
		u := User{}
		// 掃描rows的每一列并保存數據到User對應字段
		err := rows.Scan(&u.UserId, &u.UserName, &u.UserCode, &u.Password)
		if err != nil {
			// 打印錯誤信息
			log.Fatal(err)
			// 拋出錯誤信息，阻止程序繼續(xù)運行
			panic(err)
		}，
		// 掃描后的user加入到切片
		s = append(s, u)
	}
	return s
}