update

handlers/student.go

@@ -1,4 +1,4 @@
-package middleware
+package handlers
 
 import (
 	"bytes"
@@ -33,7 +33,7 @@ func GetUserInfo(isStudent bool, redisClient *redis.Storage) fiber.Handler {
 			})
 		}
 		cfg := config.New()
-		token, err := getEKTPlatformToken(redisClient, cfg.APIUser, cfg.APIPassword)
+		token, err := getEKTPlatformToken(redisClient)
 		if err != nil {
 			return c.Status(http.StatusUnauthorized).JSON(fiber.Map{
 				"message": "获取token失败,",
@@ -94,7 +94,7 @@ func GetStudentInfoByParent(redisClient *redis.Storage) fiber.Handler {
 			})
 		}
 		cfg := config.New()
-		token, err := getEKTPlatformToken(redisClient, cfg.APIUser, cfg.APIPassword)
+		token, err := getEKTPlatformToken(redisClient)
 		if err != nil {
 			return c.Status(http.StatusUnauthorized).JSON(fiber.Map{
 				"message": "获取token失败,",
@@ -134,7 +134,7 @@ func GetStudentInfoByParent(redisClient *redis.Storage) fiber.Handler {
 }
 
 // Redis中 取数据
-func getEKTPlatformToken(redisClient *redis.Storage, username string, password string) (string, error) {
+func getEKTPlatformToken(redisClient *redis.Storage) (string, error) {
 	userIDBytes, err := redisClient.Get("testToken")
 	if err != nil {
 		return "", errors.New("获取token失败,redis错误")
@@ -151,7 +151,7 @@ func getEKTPlatformToken(redisClient *redis.Storage, username string, password s
 		}
 	}
 	// token无效，重新获取token
-	token, err := fetchTokenFromAPI(username, password)
+	token, err := fetchTokenFromAPI()
 	if err != nil {
 		return "", errors.New(err.Error())
 	}
@@ -159,11 +159,12 @@ func getEKTPlatformToken(redisClient *redis.Storage, username string, password s
 	return token, nil
 }
 
-func fetchTokenFromAPI(username, password string) (string, error) {
-	//url 定义在config.go中
-	cfg := config.New()
+func fetchTokenFromAPI() (string, error) {
 
+	cfg := config.New()
 	url := cfg.APITokenUrl
+	username := cfg.APIUser
+	password := cfg.APIPassword
 	reqBody := map[string]string{
 		"name":     username,
 		"password": password,

main.go

@@ -55,9 +55,9 @@ func main() {
 	app.Get("/user", middleware.AuthMiddleware(redisClient), handlers.GetCurrentUser(db))
 	app.Put("/user", middleware.AuthMiddleware(redisClient), handlers.UpdateCurrentUser(db))
 	app.Get("/users/:id", middleware.AuthMiddleware(redisClient), handlers.GetUserByID(db))
-	app.Post("/getuser", middleware.GetUserInfo(false, redisClient))
-	app.Post("/getstudent", middleware.GetUserInfo(true, redisClient))
-	app.Post("/getstudentsbynumber", middleware.GetStudentInfoByParent(redisClient))
+	app.Post("/getuser", handlers.GetUserInfo(false, redisClient))
+	app.Post("/getstudent", handlers.GetUserInfo(true, redisClient))
+	app.Post("/getstudentsbynumber", handlers.GetStudentInfoByParent(redisClient))
 
 	// Start server
 	log.Fatal(app.Listen(":7777"))