-
Notifications
You must be signed in to change notification settings - Fork 2
/
utils.go
75 lines (67 loc) · 1.82 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
package main
import (
"github.com/Missuo0o/goBank/model"
"github.com/gin-contrib/sessions"
"github.com/gin-gonic/gin"
"golang.org/x/crypto/bcrypt"
"gorm.io/gorm"
"math/rand"
"net/http"
"time"
)
func hashPassword(password string) string {
bytes, _ := bcrypt.GenerateFromPassword([]byte(password), 14)
return string(bytes)
}
func checkPasswordHash(password, hash string) bool {
err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
return err == nil
}
// RoleAuthMiddleware is a middleware that checks if the user is logged in and has the correct role
func RoleAuthMiddleware(allowRole string) gin.HandlerFunc {
return func(c *gin.Context) {
session := sessions.Default(c)
username := session.Get("username")
role := session.Get("role")
if username == nil {
c.JSON(http.StatusUnauthorized, gin.H{
"message": "Not logged in",
})
c.Abort() // Prevent subsequent handlers from being called
return
} else if role != allowRole {
c.JSON(http.StatusUnauthorized, gin.H{
"message": "Unauthorized",
})
c.Abort() // Prevent subsequent handlers from being called
return
}
// If authentication is passed, continue processing the request
c.Next()
}
}
// generateRandomNumber
func generateRandomNumber() int64 {
rand.Seed(time.Now().UnixNano()) // 设置随机种子
// 生成一个在1000000000到9999999999之间的随机数字
minNumber := int64(1000000000)
maxNumber := int64(9999999999)
return rand.Int63n(maxNumber-minNumber+1) + minNumber
}
func generateUniqueRandomNumberString(db *gorm.DB) int64 {
var number []int64
db.Model(&model.Account{}).Select("number").Find(&number)
for {
newNumber := generateRandomNumber()
isUnique := true
for _, num := range number {
if num == newNumber {
isUnique = false
break
}
}
if isUnique {
return newNumber
}
}
}