add speed to server

XORbit01 · Apr 12, 2023 · b695cd9 · b695cd9
1 parent 820039a
commit b695cd9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/dispatcher/camp.go b/dispatcher/camp.go
@@ -11,6 +11,7 @@ type Soldier struct {
 	Name        string    `json:"name"`
 	Ip          string    `json:"ip"`
 	LastRequest time.Time `json:"last_request"`
+	Speed       int       `json:"speed"`
 }
 
 const (

diff --git a/dispatcher/server.go b/dispatcher/server.go
@@ -6,6 +6,7 @@ import (
 	"net"
 	"net/http"
 	"os"
+	"strconv"
 	"time"
 )
 
@@ -37,6 +38,14 @@ func HandleCamp(writer http.ResponseWriter, request *http.Request, d *Dispatcher
 		if soldier.Name != "" {
 			soldier.LastRequest = time.Now()
 		}
+		speed := request.URL.Query().Get("speed")
+		if speed != "" {
+			soldier.Speed, err = strconv.Atoi(speed)
+			if err != nil {
+				http.Error(writer, "error getting speed", http.StatusBadRequest)
+				return
+			}
+		}
 	}
 	if request.Method == "POST" {
 		//add soldier to camp