go_echo/main.go

56 lines
1.1 KiB
Go
Raw Normal View History

2023-03-20 18:19:02 +08:00
package main
import (
2023-04-20 23:43:26 +08:00
"fmt"
2023-03-20 18:19:02 +08:00
"net/http"
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
2023-04-20 23:43:26 +08:00
"golang.org/x/net/websocket"
2023-03-20 18:19:02 +08:00
)
type result struct {
// 通过json标签指定json字段名
2023-04-20 23:43:26 +08:00
Sataus int `json:"sataus"`
Code int `json:"code"`
Tip_number int `json:"tip_number"`
2023-03-20 18:19:02 +08:00
}
2023-04-20 23:43:26 +08:00
var tip_number int = 0
2023-03-20 18:19:02 +08:00
func main() {
e := echo.New()
e.Use(middleware.CORSWithConfig(middleware.CORSConfig{AllowOrigins: []string{"*"}, AllowMethods: []string{echo.GET, echo.HEAD, echo.PUT, echo.PATCH, echo.POST, echo.DELETE}}))
2023-04-20 23:43:26 +08:00
e.GET("/add", add)
e.GET("/ws", ws)
2023-03-20 18:19:02 +08:00
e.Logger.Fatal(e.Start(":1323"))
}
2023-04-20 23:43:26 +08:00
func add(c echo.Context) error {
tip_number++
return c.JSON(http.StatusOK, "ok")
2023-03-20 18:19:02 +08:00
}
2023-04-20 23:43:26 +08:00
func ws(c echo.Context) error {
websocket.Handler(func(ws *websocket.Conn) {
defer ws.Close()
for {
// Write
err := websocket.Message.Send(ws, tip_number)
if err != nil {
c.Logger().Error(err)
}
// Read
msg := ""
err = websocket.Message.Receive(ws, &msg)
if err != nil {
c.Logger().Error(err)
}
fmt.Printf("%s\n", msg)
2023-03-20 18:19:02 +08:00
}
2023-04-20 23:43:26 +08:00
}).ServeHTTP(c.Response(), c.Request())
return nil
2023-03-20 18:19:02 +08:00
}