Merge pull request #1 from stydxm/main

feat: robot data support

Author: 1067088037

internal/analyze/schedule.go

@@ -14,7 +14,7 @@ import (
 
 // GetScheduleData 获取赛程数据
 func GetScheduleData() ([]byte, error) {
-	resp, err := http.Get(job.ScheduleUrl)
+	resp, err := http.Get(job.Params["schedule"].Url)
 	if err != nil {
 		return nil, err
 	}

internal/handler/group_rank_info.go

@@ -0,0 +1,61 @@
+package handler
+
+import (
+	"github.com/kataras/iris/v12"
+	"github.com/scutrobotlab/rm-schedule/internal/svc"
+)
+
+type RouteHandlerParam struct {
+	Name, CacheControl, OriginalUrl string
+	Static                          bool
+	SeasonMap                       map[string][]byte
+	Data                            []byte
+}
+
+func RouteHandlerFactory(param RouteHandlerParam) func(c iris.Context) {
+	return func(c iris.Context) {
+		if param.SeasonMap != nil {
+			season := c.URLParam("season")
+			if data, ok := param.SeasonMap[season]; ok {
+				c.Header("Cache-Control", "public, max-age=60")
+				c.ContentType("application/json")
+				_, err := c.Write(data)
+				if err != nil {
+					_ = c.JSON(iris.Map{"code": -1, "msg": "Failed to get " + param.Name})
+				}
+				return
+			}
+		}
+
+		if param.Static {
+			c.Header("Cache-Control", "public, max-age=60")
+			c.ContentType("application/json")
+			_, err := c.Write(param.Data)
+			if err != nil {
+				_ = c.JSON(iris.Map{"code": -1, "msg": "Failed to get " + param.Name})
+			}
+			return
+		}
+
+		// 是否存在 Tencent-Acceleration-Domain-Name
+		if c.GetHeader("Tencent-Acceleration-Domain-Name") != "" {
+			c.Header("Cache-Control", param.CacheControl)
+			c.Redirect(param.OriginalUrl, 301)
+			return
+		}
+
+		if cached, b := svc.Cache.Get(param.Name); b {
+			c.Header("Cache-Control", param.CacheControl)
+			c.ContentType("application/json")
+			_, err := c.Write(cached.([]byte))
+			if err != nil {
+				_ = c.JSON(iris.Map{"code": -1, "msg": "Failed to get " + param.Name})
+			}
+			return
+		}
+
+		c.Header("Cache-Control", param.CacheControl)
+		c.StatusCode(500)
+		_ = c.JSON(iris.Map{"code": -1, "msg": "Failed to get " + param.Name})
+	}
+}

internal/handler/handler_factory.go

@@ -6,21 +6,45 @@ import (
 	"github.com/robfig/cron/v3"
 )
 
+type CronJobParam struct {
+	Name, Url       string
+	ReplaceRMStatic bool
+}
+
+var Params = map[string]CronJobParam{
+	"group_rank_info": {
+		Name:            "group_rank_info",
+		Url:             "https://pro-robomasters-hz-n5i3.oss-cn-hangzhou.aliyuncs.com/live_json/group_rank_info.json",
+		ReplaceRMStatic: false,
+	},
+	"robot_data": {
+		Name:            "robot_data",
+		Url:             "https://pro-robomasters-hz-n5i3.oss-cn-hangzhou.aliyuncs.com/live_json/robot_data.json",
+		ReplaceRMStatic: false,
+	},
+	"schedule": {
+		Name:            "schedule",
+		Url:             "https://pro-robomasters-hz-n5i3.oss-cn-hangzhou.aliyuncs.com/live_json/schedule.json",
+		ReplaceRMStatic: true,
+	},
+}
+
 func InitCronJob() *cron.Cron {
+	var jobFuncArray []func()
+	for _, param := range Params {
+		jobFuncArray = append(jobFuncArray, CronJobFactory(param))
+	}
+
 	c := cron.New()
 
-	_, err := c.AddFunc("@every 5s", UpdateGroupRankInfo)
-	if err != nil {
-		log.Fatalf("cron add func failed: %v", err)
-	}
+	for _, jobFunc := range jobFuncArray {
+		_, err := c.AddFunc("@every 5s", jobFunc)
+		if err != nil {
+			log.Fatalf("cron add func failed: %v", err)
+		}
 
-	_, err = c.AddFunc("@every 5s", UpdateSchedule)
-	if err != nil {
-		log.Fatalf("cron add func failed: %v", err)
+		jobFunc()
 	}
 
-	UpdateGroupRankInfo()
-	UpdateSchedule()
-
 	return c
 }

internal/handler/schedule.go

@@ -0,0 +1,50 @@
+package job
+
+import (
+	"github.com/patrickmn/go-cache"
+	"github.com/scutrobotlab/rm-schedule/internal/svc"
+	"golang.org/x/text/cases"
+	"golang.org/x/text/language"
+	"io"
+	"log"
+	"net/http"
+	"strings"
+)
+
+func CronJobFactory(param CronJobParam) func() {
+	return func() {
+		resp, err := http.Get(param.Url)
+		if err != nil {
+			log.Printf("Failed to get %s: %v\n", param.Name, err)
+			return
+		}
+		defer resp.Body.Close()
+
+		if resp.StatusCode != http.StatusOK {
+			log.Printf("Failed to get %s: status code %d\n", param.Name, resp.StatusCode)
+			return
+		}
+
+		bytes, err := io.ReadAll(resp.Body)
+		if err != nil {
+			log.Printf("Failed to read %s: %v\n", param.Name, err)
+			return
+		}
+
+		if param.ReplaceRMStatic {
+			bytes = replaceRMStatic(bytes)
+		}
+
+		svc.Cache.Set(param.Name, bytes, cache.DefaultExpiration)
+
+		log.Printf("%s updated\n", strings.ReplaceAll(cases.Title(language.English).String(param.Name), "_", " "))
+	}
+}
+
+func replaceRMStatic(data []byte) []byte {
+	str := string(data)
+	str = strings.ReplaceAll(str, "https://rm-static.djicdn.com", "/api/static/rm-static_djicdn_com")
+	str = strings.ReplaceAll(str, "https://terra-cn-oss-cdn-public-pro.oss-cn-hangzhou.aliyuncs.com", "/api/static/terra-cn-oss-cdn-public-pro_oss-cn-hangzhou_aliyuncs_com")
+	str = strings.ReplaceAll(str, "https://pro-robomasters-hz-n5i3.oss-cn-hangzhou.aliyuncs.com", "/api/static/pro-robomasters-hz-n5i3_oss-cn-hangzhou_aliyuncs_com")
+	return []byte(str)
+}