Mingcai SHEN
2 years ago
7 changed files with 520 additions and 438 deletions
Split View
Diff Options
-
+7 -1go.mod
-
+20 -0go.sum
-
+35 -28helper.go
-
+30 -30http.go
-
+37 -33main.go
-
+244 -232recog.go
-
+147 -114scene.go
+ 7
- 1
go.mod
View File
+ 20
- 0
go.sum
View File
+ 35
- 28
helper.go
View File
| @ -1,47 +1,54 @@ | |||
| package main | |||
| import ( | |||
| "github.com/gofiber/fiber/v2" | |||
| "log" | |||
| "time" | |||
| "github.com/gofiber/fiber/v2" | |||
| "log" | |||
| "time" | |||
| ) | |||
| type StatusResult struct { | |||
| Code string `json:"code"` | |||
| Message string `json:"msg"` | |||
| Timestamp int64 `json:"timestamp"` | |||
| Code string `json:"code"` | |||
| Message string `json:"msg"` | |||
| Timestamp int64 `json:"timestamp"` | |||
| } | |||
| type StatusResultEx struct { | |||
| StatusResult | |||
| Scene string `json:"scene"` | |||
| SequenceId int64 `json:"sequenceId"` | |||
| StatusResult | |||
| Scene string `json:"scene"` | |||
| SequenceId int64 `json:"sequenceId"` | |||
| } | |||
| func failureResponse(ctx *fiber.Ctx, code string, msg string, stat int) error { | |||
| log.Println("failureResponse:> ", ctx.Request().URI(), msg) | |||
| ctx.Response().SetStatusCode(stat) | |||
| var ret = make(map[string]interface{}) | |||
| ret["status"] = StatusResult{Code: code, Message: msg, Timestamp: time.Now().Unix()} | |||
| return ctx.JSON(ret) | |||
| log.Println("failureResponse:> ", ctx.Request().URI(), msg) | |||
| ctx.Response().SetStatusCode(stat) | |||
| var ret = make(map[string]interface{}) | |||
| ret["status"] = StatusResult{Code: code, Message: msg, Timestamp: time.Now().Unix()} | |||
| return ctx.JSON(ret) | |||
| } | |||
| func successResponse(ctx *fiber.Ctx, msg string) error { | |||
| var ret = make(map[string]interface{}) | |||
| ret["status"] = StatusResult{Code: "0", Message: msg, Timestamp: time.Now().Unix()} | |||
| return ctx.JSON(ret) | |||
| var ret = make(map[string]interface{}) | |||
| ret["status"] = StatusResult{Code: "0", Message: msg, Timestamp: time.Now().Unix()} | |||
| return ctx.JSON(ret) | |||
| } | |||
| func successResponseData(ctx *fiber.Ctx, msg string, data interface{}) error { | |||
| var ret = make(map[string]interface{}) | |||
| ret["status"] = StatusResult{Code: "0", Message: msg, Timestamp: time.Now().Unix()} | |||
| ret["data"] = data | |||
| return ctx.JSON(ret) | |||
| } | |||
| func successResponseEx(ctx *fiber.Ctx, msg string, scene string, seq int64) error { | |||
| var ret = make(map[string]interface{}) | |||
| ret["status"] = StatusResultEx{ | |||
| StatusResult: StatusResult{ | |||
| Code: "0", | |||
| Message: msg, | |||
| Timestamp: time.Now().Unix(), | |||
| }, | |||
| SequenceId: seq, | |||
| Scene: scene, | |||
| } | |||
| return ctx.JSON(ret) | |||
| var ret = make(map[string]interface{}) | |||
| ret["status"] = StatusResultEx{ | |||
| StatusResult: StatusResult{ | |||
| Code: "0", | |||
| Message: msg, | |||
| Timestamp: time.Now().Unix(), | |||
| }, | |||
| SequenceId: seq, | |||
| Scene: scene, | |||
| } | |||
| return ctx.JSON(ret) | |||
| } | |||
+ 30
- 30
http.go
View File
| @ -1,49 +1,49 @@ | |||
| package main | |||
| import ( | |||
| "bytes" | |||
| "encoding/json" | |||
| "errors" | |||
| "log" | |||
| "net/http" | |||
| "time" | |||
| "bytes" | |||
| "encoding/json" | |||
| "errors" | |||
| "log" | |||
| "net/http" | |||
| "time" | |||
| ) | |||
| var httpClient *http.Client | |||
| var fakeClient = false | |||
| func httpPost(url string, data interface{}) error { | |||
| log.Println("httpPost:url>", url) | |||
| if fakeClient { | |||
| log.Println("httpPost:data>", data) | |||
| return nil | |||
| } | |||
| if bs, e := json.Marshal(data); nil != e { | |||
| return e | |||
| } else if resp, e := httpClient.Post(url, "application/json", bytes.NewBuffer(bs)); nil != e { | |||
| return e | |||
| } else if resp.StatusCode != http.StatusOK { | |||
| return errors.New("invalid resp status: " + resp.Status) | |||
| } else { | |||
| return nil | |||
| } | |||
| log.Println("httpPost:url>", url) | |||
| if fakeClient { | |||
| log.Println("httpPost:data> fake client ") | |||
| return nil | |||
| } | |||
| if bs, e := json.Marshal(data); nil != e { | |||
| return e | |||
| } else if resp, e := httpClient.Post(url, "application/json", bytes.NewBuffer(bs)); nil != e { | |||
| return e | |||
| } else if resp.StatusCode != http.StatusOK { | |||
| return errors.New("invalid resp status: " + resp.Status) | |||
| } else { | |||
| return nil | |||
| } | |||
| } | |||
| func httpGet(url string, resp interface{}) error { | |||
| return nil | |||
| return nil | |||
| } | |||
| func httpPostEx(url string, data interface{}, retries int) { | |||
| for i := 0; i < retries; i++ { | |||
| if e := httpPost(url, data); nil != e { | |||
| log.Println(">>>", e) | |||
| time.Sleep(3 * time.Second) | |||
| } else { | |||
| return | |||
| } | |||
| } | |||
| for i := 0; i < retries; i++ { | |||
| if e := httpPost(url, data); nil != e { | |||
| log.Println(">>>", e) | |||
| time.Sleep(3 * time.Second) | |||
| } else { | |||
| return | |||
| } | |||
| } | |||
| } | |||
| func init() { | |||
| httpClient = http.DefaultClient | |||
| httpClient = http.DefaultClient | |||
| } | |||
+ 37
- 33
main.go
View File
+ 244
- 232
recog.go
View File
+ 147
- 114
scene.go
View File
| @ -1,148 +1,181 @@ | |||
| package main | |||
| import ( | |||
| "errors" | |||
| "github.com/gofiber/fiber/v2" | |||
| "log" | |||
| "net/url" | |||
| "time" | |||
| "errors" | |||
| "github.com/gofiber/fiber/v2" | |||
| "log" | |||
| "net/url" | |||
| "time" | |||
| ) | |||
| type SceneCommand struct { | |||
| Scene string `json:"scene"` | |||
| PushUrl string `json:"pushUrl"` | |||
| CameraUrls []string `json:"cameraURls"` | |||
| proc func(int64) error | |||
| Scene string `json:"scene"` | |||
| PushUrl string `json:"pushUrl"` | |||
| CameraUrls []string `json:"cameraURls"` | |||
| proc func(int64) error | |||
| areaConfigs []AreaConfig `json:"-"` | |||
| } | |||
| type SceneData struct { | |||
| State string `json:"state"` | |||
| LiveUrl []string `json:"liveUrl"` | |||
| State string `json:"state"` | |||
| LiveUrl []string `json:"liveUrl"` | |||
| } | |||
| type Pointer struct { | |||
| Id string `json:"id"` | |||
| PercentX float32 `json:"percentX"` | |||
| PercentY float32 `json:"percentY"` | |||
| } | |||
| type AreaConfig struct { | |||
| AreaId string `json:"areaId"` | |||
| PointersNum int `json:"pointersNum"` | |||
| Pointers []Pointer `json:"pointers"` | |||
| } | |||
| type SceneReadyResponse struct { | |||
| Scene string `json:"scene"` | |||
| Timestamp int64 `json:"timestamp"` | |||
| SceneData SceneData `json:"sceneData"` | |||
| Scene string `json:"scene"` | |||
| Timestamp int64 `json:"timestamp"` | |||
| SceneData SceneData `json:"sceneData"` | |||
| } | |||
| var currentScene *SceneCommand | |||
| var sceneStopChannel chan bool | |||
| func validateSceneCmd(cmd SceneCommand) error { | |||
| switch cmd.Scene { | |||
| case "pullUp", "standJump", "sitUps": | |||
| if len(cmd.CameraUrls) < 1 { | |||
| return errors.New("cameraUrls can not less than 1") | |||
| } | |||
| case "race": | |||
| if len(cmd.CameraUrls) < 2 { | |||
| return errors.New("cameraUrls can not less than 2") | |||
| } | |||
| default: | |||
| return errors.New("unknown scene type:" + cmd.Scene) | |||
| } | |||
| if u, e := url.Parse(cmd.PushUrl); nil != e { | |||
| return errors.New("invalid pushUrl:" + e.Error()) | |||
| } else if u.Scheme != "http" && u.Scheme != "https" { | |||
| return errors.New("can not support pushUrl with " + u.Scheme) | |||
| } | |||
| return nil | |||
| switch cmd.Scene { | |||
| case "pullUp", "standJump", "sitUps": | |||
| if len(cmd.CameraUrls) < 1 { | |||
| return errors.New("cameraUrls can not less than 1") | |||
| } | |||
| case "race": | |||
| if len(cmd.CameraUrls) < 2 { | |||
| return errors.New("cameraUrls can not less than 2") | |||
| } | |||
| default: | |||
| return errors.New("unknown scene type:" + cmd.Scene) | |||
| } | |||
| if u, e := url.Parse(cmd.PushUrl); nil != e { | |||
| return errors.New("invalid pushUrl:" + e.Error()) | |||
| } else if u.Scheme != "http" && u.Scheme != "https" { | |||
| return errors.New("can not support pushUrl with " + u.Scheme) | |||
| } | |||
| return nil | |||
| } | |||
| func setScene(c *fiber.Ctx) error { | |||
| var sceneCmd SceneCommand | |||
| if err := c.BodyParser(&sceneCmd); nil != err { | |||
| log.Println("ERROR:", c.Body()) | |||
| return failureResponse(c, "-1", err.Error(), 200) | |||
| } else if nil != currentScene { | |||
| return failureResponse(c, "1", "A working scene is started", 200) | |||
| } else if err = validateSceneCmd(sceneCmd); nil != err { | |||
| return failureResponse(c, "2", err.Error(), 200) | |||
| } else { | |||
| // currentScene = &sceneCmd | |||
| log.Println("Starting scene ...") | |||
| log.Println(sceneCmd) | |||
| if e := sceneTaskProc(&sceneCmd); nil != e { | |||
| return failureResponse(c, "3", err.Error(), 200) | |||
| } else { | |||
| return successResponse(c, "Updated successfully.") | |||
| } | |||
| } | |||
| var sceneCmd SceneCommand | |||
| if err := c.BodyParser(&sceneCmd); nil != err { | |||
| log.Println("ERROR:", c.Body()) | |||
| return failureResponse(c, "-1", err.Error(), 200) | |||
| } else if nil != currentScene { | |||
| return failureResponse(c, "1", "A working scene is started", 200) | |||
| } else if err = validateSceneCmd(sceneCmd); nil != err { | |||
| return failureResponse(c, "2", err.Error(), 200) | |||
| } else { | |||
| // currentScene = &sceneCmd | |||
| log.Println("Starting scene ...") | |||
| log.Println(sceneCmd) | |||
| if e := sceneTaskProc(&sceneCmd); nil != e { | |||
| return failureResponse(c, "3", err.Error(), 200) | |||
| } else { | |||
| return successResponse(c, "Updated successfully.") | |||
| } | |||
| } | |||
| } | |||
| func setAreaConfig(c *fiber.Ctx) error { | |||
| var cfg []AreaConfig | |||
| if e := c.BodyParser(&cfg); nil != e { | |||
| log.Println("ERROR:", c.Body()) | |||
| return failureResponse(c, "-1", e.Error(), 200) | |||
| } else if nil == currentScene { | |||
| return failureResponse(c, "1", "No working scene is started", 200) | |||
| } | |||
| currentScene.areaConfigs = cfg | |||
| return successResponse(c, "Updated successfully.") | |||
| } | |||
| func getAreaConfig(c *fiber.Ctx) error { | |||
| if nil == currentScene { | |||
| return failureResponse(c, "1", "No working scene is started", 200) | |||
| } else { | |||
| return successResponseData(c, "success", currentScene.areaConfigs) | |||
| } | |||
| } | |||
| func stopScene(c *fiber.Ctx) error { | |||
| if nil != testStopChannel { | |||
| testStopChannel <- true | |||
| } | |||
| if nil != sceneStopChannel { | |||
| sceneStopChannel <- true | |||
| } | |||
| return successResponse(c, "Updated successfully.") | |||
| if nil != testStopChannel { | |||
| testStopChannel <- true | |||
| } | |||
| if nil != sceneStopChannel { | |||
| sceneStopChannel <- true | |||
| } | |||
| return successResponse(c, "Updated successfully.") | |||
| } | |||
| func getStatus(c *fiber.Ctx) error { | |||
| var scene = "" | |||
| var seq int64 = 0 | |||
| if nil != currentScene { | |||
| scene = currentScene.Scene | |||
| } | |||
| if nil != currentTest { | |||
| seq = currentTest.SequenceId | |||
| } | |||
| return successResponseEx(c, "success", scene, seq) | |||
| var scene = "" | |||
| var seq int64 = 0 | |||
| if nil != currentScene { | |||
| scene = currentScene.Scene | |||
| } | |||
| if nil != currentTest { | |||
| seq = currentTest.SequenceId | |||
| } | |||
| return successResponseEx(c, "success", scene, seq) | |||
| } | |||
| func sceneTaskProc(cmd *SceneCommand) error { | |||
| var f func(int64) error = nil | |||
| switch cmd.Scene { | |||
| case "pullUp": | |||
| f = pullUpTask | |||
| case "standJump": | |||
| f = standJumpTask | |||
| case "sitUps": | |||
| f = sitUpsTask | |||
| case "race": | |||
| f = raceTask | |||
| default: | |||
| return errors.New("unknown scene type:" + cmd.Scene) | |||
| } | |||
| sceneStopChannel = make(chan bool) | |||
| var tk = time.NewTicker(time.Second * 3) | |||
| go func() { | |||
| defer func() { | |||
| tk.Stop() | |||
| close(sceneStopChannel) | |||
| sceneStopChannel = nil | |||
| }() | |||
| currentScene = cmd | |||
| currentScene.proc = f | |||
| time.Sleep(time.Second) | |||
| var st = SceneReadyResponse{ | |||
| Scene: cmd.Scene, | |||
| Timestamp: time.Now().Unix(), | |||
| SceneData: SceneData{ | |||
| State: "SCENE READY", | |||
| LiveUrl: cmd.CameraUrls, | |||
| }, | |||
| } | |||
| go httpPostEx(cmd.PushUrl, st, 3) | |||
| for { | |||
| select { | |||
| case <-tk.C: | |||
| // log.Println(" ...") | |||
| case <-sceneStopChannel: | |||
| log.Println("Scene Stopping...") | |||
| go pushEvent(cmd.PushUrl, cmd.Scene, 0, "SCENE END", false, false) | |||
| currentScene = nil | |||
| return | |||
| } | |||
| } | |||
| }() | |||
| return nil | |||
| var f func(int64) error = nil | |||
| switch cmd.Scene { | |||
| case "pullUp": | |||
| f = pullUpTask | |||
| case "standJump": | |||
| f = standJumpTask | |||
| case "sitUps": | |||
| f = sitUpsTask | |||
| case "race": | |||
| f = raceTask | |||
| default: | |||
| return errors.New("unknown scene type:" + cmd.Scene) | |||
| } | |||
| sceneStopChannel = make(chan bool) | |||
| var tk = time.NewTicker(time.Second * 3) | |||
| go func() { | |||
| defer func() { | |||
| tk.Stop() | |||
| close(sceneStopChannel) | |||
| sceneStopChannel = nil | |||
| }() | |||
| currentScene = cmd | |||
| currentScene.proc = f | |||
| time.Sleep(time.Second) | |||
| var st = SceneReadyResponse{ | |||
| Scene: cmd.Scene, | |||
| Timestamp: time.Now().Unix(), | |||
| SceneData: SceneData{ | |||
| State: "SCENE READY", | |||
| LiveUrl: cmd.CameraUrls, | |||
| }, | |||
| } | |||
| go httpPostEx(cmd.PushUrl, st, 3) | |||
| for { | |||
| select { | |||
| case <-tk.C: | |||
| // log.Println(" ...") | |||
| case <-sceneStopChannel: | |||
| log.Println("Scene Stopping...") | |||
| go pushEvent(cmd.PushUrl, cmd.Scene, 0, "SCENE END", false) | |||
| currentScene = nil | |||
| return | |||
| } | |||
| } | |||
| }() | |||
| return nil | |||
| } | |||
| func init() { | |||
| currentScene = nil | |||
| sceneStopChannel = nil | |||
| currentScene = nil | |||
| sceneStopChannel = nil | |||
| } | |||