Golang实现简单登录注册 忘记密码

业界资讯 2020-06-19 00:18:25 阅读次数: 0

引入需要的包以及数据库连接

// 按需引入需要的包
	"database/sql"
	"fmt"
	"github.com/gin-gonic/gin" //gin
	_ "github.com/go-sql-driver/mysql" //mysql
	"strings"

// 连接数据库参数
const (
	userName = "用户名"
	password = "密码"
	ip = "ip地址"
	port = "端口"
	dbName = "数据库名"
)

var DB *sql.DB
// 连接数据库
func initDB() {
	// 构建连接："用户名:密码@tcp(IP:端口)/数据库?charset=utf8"
	path := strings.Join([]string{userName, ":", password, "@tcp(",ip, ":", port, ")/", dbName, "?charset=utf8"}, "")
	// 打开数据库 （"驱动名"，连接）
	DB,_ = sql.Open("mysql",path)
	// 设置数据库最大连接数
	DB.SetConnMaxLifetime(100)
	// 设置数据库最大闲置数
	DB.SetMaxIdleConns(10)
	// 验证连接
	if err := DB.Ping(); err != nil {
		fmt.Println("连接数据库失败")
		return
	}
	fmt.Println("连接数据库成功")
}

声明用户结构体

// User
type User struct {
	id int //id
	username string //用户名
	password string //密码
	tel string //手机号
}

登录

func userLogin(c *gin.Context) {
	userName := c.Request.URL.Query().Get("username") //获取登录用户名
	passWord := c.Request.URL.Query().Get("password") //获取登录密码
	//查询列表
	rows,err := DB.Query("SELECT * FROM voro_user") //执行sql语句查询表中数据
	if err != nil {
		fmt.Println("查询失败")
	}
	var s User
	for rows.Next() {
		err = rows.Scan(&s.id, &s.username, &s.password, &s.tel)
		if err != nil {
			fmt.Println(err)
		}
	}
	if userName != s.username { // 判断获取到的登录用户名是否在数据库中存在
		// 无此用户(用户名不存在)
		c.JSON(200,gin.H{
			"success":false,
			"code":400,
			"msg":"无此用户",
		})
	} else {
		// 获取当前用户名密码
		// 获取登录用户名的密码 查询是否匹配
		us,_ := DB.Query("SELECT password FROM voro_user where username='" + userName + "'")
		for us.Next(){
			var u passWd
			err = us.Scan(&u.password)
			if err != nil {
				fmt.Println(err)
			}
			// 密码是否匹配
			if passWord != u.password{ //密码不一致
				c.JSON(200,gin.H{
					"success":false,
					"code":400,
					"msg":"密码错误",
				})
			} else {
				c.JSON(200,gin.H{ //用户名存在且密码匹配
					"success":true,
					"code":200,
					"msg":"登录成功",
				})
			}
		}
	}
	rows.Close()
}

注册

func userRegister(c *gin.Context){
	userName := c.Request.URL.Query().Get("username") //注册用户名
	passWord := c.Request.URL.Query().Get("password") //注册密码
	userTel := c.Request.URL.Query().Get("tel") //注册手机号
	//查询列表
	rows,err := DB.Query("SELECT * FROM voro_user") //查询用户名是否已存在
	if err != nil {
		fmt.Println("查询失败")
	}
	//遍历返回结果
	for rows.Next() {
		var s User
		err = rows.Scan(&s.id,&s.username,&s.password,&s.tel)
		if err != nil{
			fmt.Println(err)
		}
		fmt.Println(s.username)
		if userName != s.username{ //数据库中不存在此用户(可以注册)
			// 执行插入
			result, err := DB.Exec("INSERT INTO voro_user(username,password,tel)VALUES (?,?,?)",userName,passWord,userTel)
			if err != nil {
				fmt.Println("执行失败")
				return
			} else {
				rows,_ := result.RowsAffected() //输出执行的行数
				if rows != 1 { //不为1 表示插入失败(注册失败)
					c.JSON(200,gin.H{
						"success":false,
					})
				} else {
					c.JSON(200,gin.H{ //注册成功
						"success":true,
						"username":userName,
						"tel":userTel,
					})
				}
			}
		} else {
			fmt.Println("用户名已被注册") //用户名已存在(注册失败 用户名已被注册)
			c.JSON(200,gin.H{
				"code":400,
				"success":false,
				"msg":"用户名已被注册",
			})
		}
	}
	rows.Close()
}

忘记密码

func forgetPassword(c *gin.Context){
	userName := c.Request.URL.Query().Get("username") //忘记密码的用户名
	newPassWord := c.Request.URL.Query().Get("newPassWord") //需要修改的新密码
	userTel := c.Request.URL.Query().Get("tel") //用户注册手机号
	//查询列表
	rows,err := DB.Query("SELECT * FROM voro_user")
	if err != nil {
		fmt.Println("查询失败")
	}
	var s User
	for rows.Next() {
		err = rows.Scan(&s.id, &s.username, &s.password, &s.tel)
		if err != nil {
			fmt.Println(err)
		}
	}
	if userName != s.username { //不存在此用户
		// 无此用户
		c.JSON(200, gin.H{
			"success": false,
			"code":    400,
			"msg":     "无此用户",
		})
	} else { //存在此用户名
		// 获取用户注册手机号
		us,_ := DB.Query("SELECT tel FROM voro_user where username='" + userName + "'")
		for us.Next(){
			var m User
			err = us.Scan(&m.tel)
			if err != nil {
				fmt.Println(err)
			}
			// 手机号是否匹配
			if userTel != m.tel{ //注册手机号不匹配
				c.JSON(200,gin.H{
					"success":false,
					"code":400,
					"msg":"安全手机验证错误",
				})
			} else { //注册手机号一致
				results,err:=DB.Exec("UPDATE voro_user SET password='" + newPassWord + "' where username='" + userName + "' and tel='" + userTel + "'") //执行更新操作
				if err != nil {
					fmt.Println(err)
				}
				aff_count,_ := results.RowsAffected()
				if aff_count != 0{ //不为0 表示更新成功
					c.JSON(200,gin.H{
						"success":true,
						"code":200,
						"msg":"密码更新成功",
					})
				} else {
					c.JSON(200,gin.H{
						"success":false,
						"code":400,
						"msg":"密码更新失败",
					})
				}
			}
		}
	}
	rows.Close()
}

全部代码

package main

import (
	"database/sql"
	"fmt"
	"github.com/gin-gonic/gin"
	_ "github.com/go-sql-driver/mysql"
	"strings"
)
// 连接数据库参数
const (
	userName = "***"
	password = "***"
	ip = "***"
	port = "***"
	dbName = "***"
)

var DB *sql.DB
// 连接数据库
func initDB() {
	// 构建连接："用户名:密码@tcp(IP:端口)/数据库?charset=utf8"
	path := strings.Join([]string{userName, ":", password, "@tcp(",ip, ":", port, ")/", dbName, "?charset=utf8"}, "")
	// 打开数据库 （"驱动名"，连接）
	DB,_ = sql.Open("mysql",path)
	// 设置数据库最大连接数
	DB.SetConnMaxLifetime(100)
	// 设置数据库最大闲置数
	DB.SetMaxIdleConns(10)
	// 验证连接
	if err := DB.Ping(); err != nil {
		fmt.Println("连接数据库失败")
		return
	}
	fmt.Println("连接数据库成功")
}
// User
type User struct {
	id int //id
	username string //用户名
	password string //密码
	tel string //手机号
}
type passWd struct {
	password string
}
// @userLogin
func userLogin(c *gin.Context) {
	userName := c.Request.URL.Query().Get("username")
	passWord := c.Request.URL.Query().Get("password")
	//查询列表
	rows,err := DB.Query("SELECT * FROM voro_user")
	if err != nil {
		fmt.Println("查询失败")
	}
	var s User
	for rows.Next() {
		err = rows.Scan(&s.id, &s.username, &s.password, &s.tel)
		if err != nil {
			fmt.Println(err)
		}
	}
	if userName != s.username {
		// 无此用户
		c.JSON(200,gin.H{
			"success":false,
			"code":400,
			"msg":"无此用户",
		})
	} else {
		// 获取当前用户名密码
		us,_ := DB.Query("SELECT password FROM voro_user where username='" + userName + "'")
		for us.Next(){
			var u passWd
			err = us.Scan(&u.password)
			if err != nil {
				fmt.Println(err)
			}
			// 密码是否匹配
			if passWord != u.password{
				c.JSON(200,gin.H{
					"success":false,
					"code":400,
					"msg":"密码错误",
				})
			} else {
				c.JSON(200,gin.H{
					"success":true,
					"code":200,
					"msg":"登录成功",
				})
			}
		}
	}
	rows.Close()
}

// @userRegister
func userRegister(c *gin.Context){
	userName := c.Request.URL.Query().Get("username")
	passWord := c.Request.URL.Query().Get("password")
	userTel := c.Request.URL.Query().Get("tel")
	//查询列表
	rows,err := DB.Query("SELECT * FROM voro_user")
	if err != nil {
		fmt.Println("查询失败")
	}
	//遍历返回结果
	for rows.Next() {
		var s User
		err = rows.Scan(&s.id,&s.username,&s.password,&s.tel)
		if err != nil{
			fmt.Println(err)
		}
		fmt.Println(s.username)
		if userName != s.username{
			// 执行插入
			result, err := DB.Exec("INSERT INTO voro_user(username,password,tel)VALUES (?,?,?)",userName,passWord,userTel)
			if err != nil {
				fmt.Println("执行失败")
				return
			} else {
				rows,_ := result.RowsAffected()
				if rows != 1 {
					c.JSON(200,gin.H{
						"success":false,
					})
				} else {
					c.JSON(200,gin.H{
						"success":true,
						"username":userName,
						"tel":userTel,
					})
				}
			}
		} else {
			fmt.Println("用户名已被注册")
			c.JSON(200,gin.H{
				"code":400,
				"success":false,
				"msg":"用户名已被注册",
			})
		}
	}
	rows.Close()
}
// @forgetPassword
func forgetPassword(c *gin.Context){
	userName := c.Request.URL.Query().Get("username")
	newPassWord := c.Request.URL.Query().Get("newPassWord")
	userTel := c.Request.URL.Query().Get("tel")
	//查询列表
	rows,err := DB.Query("SELECT * FROM voro_user")
	if err != nil {
		fmt.Println("查询失败")
	}
	var s User
	for rows.Next() {
		err = rows.Scan(&s.id, &s.username, &s.password, &s.tel)
		if err != nil {
			fmt.Println(err)
		}
	}
	if userName != s.username {
		// 无此用户
		c.JSON(200, gin.H{
			"success": false,
			"code":    400,
			"msg":     "无此用户",
		})
	} else {
		// 获取用户注册手机号
		us,_ := DB.Query("SELECT tel FROM voro_user where username='" + userName + "'")
		for us.Next(){
			var m User
			err = us.Scan(&m.tel)
			if err != nil {
				fmt.Println(err)
			}
			// 手机号是否匹配
			if userTel != m.tel{
				c.JSON(200,gin.H{
					"success":false,
					"code":400,
					"msg":"安全手机验证错误",
				})
			} else {
				results,err:=DB.Exec("UPDATE voro_user SET password='" + newPassWord + "' where username='" + userName + "' and tel='" + userTel + "'")
				if err != nil {
					fmt.Println(err)
				}
				aff_count,_ := results.RowsAffected()
				if aff_count != 0{
					c.JSON(200,gin.H{
						"success":true,
						"code":200,
						"msg":"密码更新成功",
					})
				} else {
					c.JSON(200,gin.H{
						"success":false,
						"code":400,
						"msg":"密码更新失败",
					})
				}
			}
		}
	}
	rows.Close()
}

func main(){
	initDB()
	router := gin.Default()
	user := router.Group("/user")
	{
		user.POST("/login",userLogin)
		user.POST("/register",userRegister)
		user.POST("/forgetpassword",forgetPassword)
	}
	router.Run(":9000")
}

猜你喜欢

转载自blog.csdn.net/qinghuan100/article/details/106786868
今日推荐
火速冲上 GitHub 热榜 —— 开源编程语言、框架哪有这么可爱？
北京人形机器人创新中心发布全球首个纯电驱拟人奔跑的全尺寸人形机器人“天工”
LFOSSA 源来如此公开课 | 掌握云原生未来：CNCF 认证全面攻略与备考秘籍
国产云输入法——仅华为无云端数据上传安全问题
开源日报 | 工业开源项目OGG 1.0；姐姐，你要和我一起配置火狐吗；苹果AI遥遥落后？Fedora 40
开放签电子签章：停止新增，优化体验，前进更进（五一假期前工作）
开源日报 | 中学生开源前端动画引擎；全球首个Llama3 8B中文版开源模型；联想电脑恐出局；Linus讽刺AI炒作
周排行
浏览器对同一域名进行请求的最大并发连接数
React Hook之自定义Hook
【转】MyBatis缓存机制
-Java-泛型
自动化测试常用脚本-发送邮件
LeetCode#859: Buddy Strings
java、Python处理字符串
第二篇の博客
Hadoop伪分布式环境安装
SQL Server进阶（十一）临时表、表变量
每日归档
更多
2024-04-27(56)
2024-04-26(39)
2024-04-25(22)
2024-04-24(36)
2024-04-23(26)
2024-04-22(39)
2024-04-21(0)
2024-04-20(6)
2024-04-19(5)
2024-04-18(0)

代码天地 © 2018 codetd.com

境外服务器   最新电视剧2022