all_platform/api_conn/authMiddleware.go

91 lines
3.2 KiB
Go
Raw Normal View History

2025-03-16 23:57:25 +08:00
package api_conn
import (
"all_platform/dbpool"
"all_platform/tool"
"fmt"
"log"
"net/http"
"strconv"
"time"
"github.com/gin-gonic/gin"
machine_code "github.com/super-l/machine-code"
)
// 定义一个路由组,应用于特定的中间件
// 校验用户路由中间件
func authMiddleware(c *gin.Context) {
if c.Request.Header.Get("key") != "" {
if key := tool.Keyprin(c.Request.Header.Get("key")); key.Status == "1" {
a, _ := strconv.ParseInt(key.End_time, 10, 64)
duration := time.Unix(a, 0).Sub(time.Unix(tool.Timestamp("second"), 0))
days := int(duration / (24 * time.Hour)) //天
hours := int(duration % (24 * time.Hour) / time.Hour) //小时
minutes := int(duration / time.Minute % 60) //分钟
secod := int(duration / time.Second % 60) //秒
if days <= 0 && hours <= 0 && minutes <= 0 {
c.JSON(http.StatusOK, gin.H{"error": fmt.Sprintf("授权已到期,剩余时间:%d天 %d小时 %d分钟 %d秒", days, hours, minutes, secod)})
log.Printf("授权已到期,剩余时间:%d天 %d小时 %d分钟 %d秒", days, hours, minutes, secod)
c.Abort()
} else {
log.Printf("授权未到期,剩余时间:%d天 %d小时 %d分钟 %d秒", days, hours, minutes, secod)
c.Next()
}
} else {
c.JSON(http.StatusOK, gin.H{"error": "类型不匹配api请求类型为1"})
log.Println("类型不匹配api请求类型为1")
c.Abort()
}
} else {
c.JSON(http.StatusOK, gin.H{"error": "密钥为空"})
log.Println("密钥为空")
c.Abort()
}
}
func Auth_main(mobile_phone_number, status string) map[string]string {
var auth_s dbpool.Douy_info
fmt.Println("电脑序列号:", machine_code.Machine.BoardSerialNumber)
if machine_code.Machine.BoardSerialNumber != "" {
//查询sql 获取用户信息
rows, err := dbpool.Query("SELECT * FROM `douy`.`douy_info` WHERE mobile_phone_number = ? and status = ?;", mobile_phone_number, status)
if err != nil {
log.Println(err)
}
defer rows.Close()
// var typecho_userss []dbpool.Typecho_users
for rows.Next() {
if err := rows.Scan(&auth_s.Mobile_phone_number, &auth_s.SerialNumber, &auth_s.Username, &auth_s.Initial_time, &auth_s.End_time, &auth_s.Surplus_time, &auth_s.Key, &auth_s.Status); err != nil {
log.Println(err)
}
// typecho_userss = append(typecho_userss, users)
}
if err := rows.Err(); err != nil {
log.Fatal(err)
}
// log.Println("查询到的结果:", auth_s)
} else {
log.Println("该设备未申请授权,该设备码为:", machine_code.Machine.BoardSerialNumber)
}
aa := map[string]string{
"mobile_phone_number": auth_s.Mobile_phone_number.String,
"serialNumber": auth_s.SerialNumber.String,
"username": auth_s.Username.String,
"initial_time": auth_s.Initial_time.String,
"end_time": auth_s.End_time.String,
"surplus_time": auth_s.Surplus_time.String,
"key": auth_s.Key.String,
}
return aa
}
// 认证参数
type Auth struct {
Msg string `json:"msg"` //输入手机号
BoardSerialNumber string `json:"boardSerialNumber"` //输入主板序列号
Status string `json:"status"` //输入1是api2是客户端
}