前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >go的gin和gorm框架实现切换身份的接口

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

作者头像
叶茂林
发布2023-08-13 10:53:59
2130
发布2023-08-13 10:53:59
举报
文章被收录于专栏:叶子的开发者社区

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

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

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

测试代码

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

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

代码语言:javascript
复制
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
	}
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2023-08-12,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档