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"
|
2023-04-21 20:28:23 +08:00
|
|
|
"strconv"
|
2023-03-20 18:19:02 +08:00
|
|
|
|
|
|
|
"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
|
|
|
)
|
|
|
|
|
2023-04-21 20:28:23 +08:00
|
|
|
func ws(c echo.Context) error {
|
|
|
|
websocket.Handler(server).ServeHTTP(c.Response(), c.Request())
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
func add(c echo.Context) error {
|
|
|
|
tip_number++
|
|
|
|
return c.String(http.StatusOK, strconv.Itoa(tip_number))
|
|
|
|
}
|
|
|
|
|
|
|
|
func server(ws *websocket.Conn) {
|
|
|
|
defer ws.Close()
|
|
|
|
fmt.Printf("new connection\n")
|
|
|
|
for {
|
|
|
|
if tip_number == 0 {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
var strr string = strconv.Itoa(tip_number)
|
|
|
|
err := websocket.Message.Send(ws, strr)
|
|
|
|
if err != nil {
|
|
|
|
fmt.Print(err)
|
|
|
|
}
|
|
|
|
tip_number = 0
|
|
|
|
}
|
|
|
|
|
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()
|
2023-04-21 20:28:23 +08:00
|
|
|
e.Use(middleware.Logger())
|
|
|
|
e.Use(middleware.Recover())
|
|
|
|
e.Static("/", "./public")
|
2023-04-20 23:43:26 +08:00
|
|
|
e.GET("/ws", ws)
|
2023-04-21 20:28:23 +08:00
|
|
|
e.GET("/add", add)
|
2023-03-20 18:19:02 +08:00
|
|
|
e.Logger.Fatal(e.Start(":1323"))
|
|
|
|
}
|