go的gin和gorm框架实现切换身份的接口

使用go的gin和gorm框架实现切换身份的接口,接收前端发送的JSON对象,查询数据库并更新,返回前端信息

接收前端发来的JSON对象,包含由openid和登陆状态组成的一个string和要切换的身份码int型

后端接收后判断要切换的身份是否低于该用户身份,是则更新数据库的登录状态为要切换的身份码,返回由openid和新的登录状态组成的string,否则返回错误码和权限不足的错误信息

测试代码

创建两个结构体,一个用来查询数据库,一个用来接收前端发送的JSON对象

连接数据库,接收前端POST请求,提取数据,查库,处理,返回信息

package main

import (
	"github.com/gin-gonic/gin"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"log"
	"net/http"
	"strconv"
)

type User struct {
	Openid     string `json:"openid" gorm:"primaryKey"`
	IdCode     int    `json:"idcode"`
	LoginState int    `json:"login_state"`
}
type RequestData struct {
	OpenidAndLoginStatus string `json:"openidAndLoginState"`
	IdCodeToChange       int    `json:"idcodeToChange"`
}

func main() {
	dsn := "username:password@tcp(host:port)/database?charset=utf8mb4&parseTime=True&loc=Local"
	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		log.Fatal(err)
	}

	err = db.AutoMigrate(&User{})
	if err != nil {
		log.Fatal(err)
	}

	router := gin.Default()
	// 接口测试地址 192.168.160.128:8080/update
	router.POST("/update", func(c *gin.Context) {
		var requestData RequestData
		if err := c.ShouldBindJSON(&requestData); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid JSON data"})
			return
		}
		openidAndLoginState := requestData.OpenidAndLoginStatus
		openID := openidAndLoginState[:len(openidAndLoginState)-1]
		loginStatus, _ := strconv.Atoi(openidAndLoginState[len(openidAndLoginState)-1:])
		idCode := requestData.IdCodeToChange
		var user User
		result := db.Where("openid = ?", openID).First(&user)
		if result.Error != nil {
			c.JSON(http.StatusOK, gin.H{"error": "Invalid openid"})
			return
		}

		if user.IdCode >= idCode {
			user.LoginState = loginStatus
			result = db.Save(&user)
			if result.Error != nil {
				c.JSON(http.StatusInternalServerError, gin.H{"error": "Database save error"})
				return
			}
		} else {
			c.JSON(http.StatusOK, gin.H{"error": "Insufficient privileges"})
			return
		}

		response := openID + strconv.Itoa(idCode)
		c.JSON(http.StatusOK, gin.H{"response": response})
	})

	err = router.Run(":8080")
	if err != nil {
		return
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

@YeMaolin

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值