update curd & code logic
This commit is contained in:
Sakurasan
@@ -1,172 +1,46 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strconv"
|
||||
"chat/opencatd/router"
|
||||
_ "chat/pkg/opencatd/db"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
type User struct {
|
||||
IsDelete bool `json:"IsDelete"`
|
||||
ID int `json:"id"`
|
||||
UpdatedAt string `json:"updatedAt"`
|
||||
Name string `json:"name"`
|
||||
Token string `json:"token"`
|
||||
CreatedAt string `json:"createdAt"`
|
||||
}
|
||||
|
||||
type Key struct {
|
||||
ID int `json:"id"`
|
||||
Key string `json:"key"`
|
||||
UpdatedAt string `json:"updatedAt"`
|
||||
Name string `json:"name"`
|
||||
CreatedAt string `json:"createdAt"`
|
||||
}
|
||||
|
||||
var users = []User{
|
||||
{false, 1, "2023-03-28T12:36:03.907590825Z", "root", "4930f144-7ffc-469b-8f2d-71a267358d87", "2023-03-28T11:52:45.593940257Z"},
|
||||
}
|
||||
|
||||
var keys = []Key{
|
||||
{1, "1234567890", "2023-03-28T12:53:41.124319079Z", "key1", "2023-03-28T12:53:41.124318928Z"},
|
||||
}
|
||||
|
||||
func main() {
|
||||
router := gin.Default()
|
||||
|
||||
r := gin.Default()
|
||||
group := r.Group("/1")
|
||||
{
|
||||
group.Use(router.AuthMiddleware())
|
||||
|
||||
// 获取当前用户信息
|
||||
group.GET("/me", router.HandleMe)
|
||||
|
||||
// 获取所有Key
|
||||
group.GET("/keys", router.HandleKeys)
|
||||
|
||||
// 获取所有用户信息
|
||||
group.GET("/users", router.HandleUsers)
|
||||
|
||||
// 添加Key
|
||||
group.POST("/keys", router.HandleAddKey)
|
||||
|
||||
// 删除Key
|
||||
group.DELETE("/keys/:id", router.HandleDelKey)
|
||||
|
||||
// 添加用户
|
||||
group.POST("/users", router.HandleAddUser)
|
||||
|
||||
// 删除用户
|
||||
group.DELETE("/users/:id", router.HandleDelUser)
|
||||
|
||||
// 重置用户Token
|
||||
group.POST("/users/:id/reset", router.HandleResetUserToken)
|
||||
}
|
||||
|
||||
// 初始化用户
|
||||
router.POST("/1/users/init", func(c *gin.Context) {
|
||||
resJSON := User{
|
||||
false,
|
||||
1,
|
||||
"2023-03-28T14:13:54.969089197Z",
|
||||
"root",
|
||||
"3260dd53-463f-45d0-afa4-57acd04ee036",
|
||||
"2023-03-28T14:13:54.969089107Z",
|
||||
}
|
||||
r.POST("/1/users/init", router.Handleinit)
|
||||
|
||||
c.JSON(http.StatusOK, resJSON)
|
||||
})
|
||||
|
||||
// 获取当前用户信息
|
||||
router.GET("/1/me", func(c *gin.Context) {
|
||||
resJSON := users[0]
|
||||
|
||||
c.JSON(http.StatusOK, resJSON)
|
||||
})
|
||||
|
||||
// 获取所有Key
|
||||
router.GET("/1/keys", func(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, keys)
|
||||
})
|
||||
|
||||
// 获取所有用户信息
|
||||
router.GET("/1/users", func(c *gin.Context) {
|
||||
c.JSON(http.StatusOK, users)
|
||||
})
|
||||
|
||||
// 添加Key
|
||||
router.POST("/1/keys", func(c *gin.Context) {
|
||||
var body Key
|
||||
if err := c.BindJSON(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
keys = append(keys, body)
|
||||
c.JSON(http.StatusCreated, body)
|
||||
})
|
||||
|
||||
// 删除Key
|
||||
router.DELETE("/1/keys/:id", func(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
var keyIdx int = -1
|
||||
|
||||
for idx, key := range keys {
|
||||
if strconv.Itoa(key.ID) == id {
|
||||
keyIdx = idx
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if keyIdx == -1 {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": "Key not found"})
|
||||
return
|
||||
}
|
||||
|
||||
keys = append(keys[:keyIdx], keys[keyIdx+1:]...)
|
||||
c.JSON(http.StatusOK, gin.H{"message": "ok"})
|
||||
})
|
||||
|
||||
// 添加用户
|
||||
router.POST("/1/users", func(c *gin.Context) {
|
||||
var body User
|
||||
if err := c.BindJSON(&body); err != nil {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
newUser := User{
|
||||
false,
|
||||
len(users) + 1,
|
||||
"2023-03-28T13:05:02.103135082Z",
|
||||
body.Name,
|
||||
GenerateToken(),
|
||||
"2023-03-28T12:38:57.513105431Z",
|
||||
}
|
||||
|
||||
users = append(users, newUser)
|
||||
c.JSON(http.StatusCreated, newUser)
|
||||
})
|
||||
|
||||
// 删除用户
|
||||
router.DELETE("/1/users/:id", func(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
var userIdx int = -1
|
||||
|
||||
for idx, user := range users {
|
||||
if strconv.Itoa(user.ID) == id {
|
||||
userIdx = idx
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if userIdx == -1 {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
|
||||
return
|
||||
}
|
||||
|
||||
users = append(users[:userIdx], users[userIdx+1:]...)
|
||||
c.JSON(http.StatusOK, gin.H{"message": "ok"})
|
||||
})
|
||||
|
||||
// 重置用户Token
|
||||
router.POST("/1/users/:id/reset", func(c *gin.Context) {
|
||||
id := c.Param("id")
|
||||
var userIdx int = -1
|
||||
|
||||
for idx, user := range users {
|
||||
if strconv.Itoa(user.ID) == id {
|
||||
userIdx = idx
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if userIdx == -1 {
|
||||
c.JSON(http.StatusNotFound, gin.H{"error": "User not found"})
|
||||
return
|
||||
}
|
||||
|
||||
users[userIdx].Token = GenerateToken()
|
||||
c.JSON(http.StatusOK, users[userIdx])
|
||||
})
|
||||
|
||||
router.Run(":8080")
|
||||
}
|
||||
|
||||
func GenerateToken() string {
|
||||
token := uuid.New()
|
||||
return token.String()
|
||||
r.Run(":80")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user