Streamer272/keys
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

🔨 Changes go net websocket to gorilla websocket

Browse Source
This commit is contained in:
Daniel Svitan
2025-03-22 12:51:21 +01:00
parent 0264c51970
commit a86eed2f72
3 changed files with 64 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

115
server/main.go
View File

@@ -8,11 +8,11 @@ import (
"syscall"
"time"
"github.com/gorilla/websocket"
"github.com/joho/godotenv"
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
"github.com/labstack/gommon/log"
"golang.org/x/net/websocket"
)
const TimeFormat = "2006-01-02 15:04:05"
@@ -27,6 +27,7 @@ var id = 0
var clients = []*Client{}
var keyDir = "."
var token = ""
var upgrader = websocket.Upgrader{}
type ReqData struct {
ID int `json:"id"`
@@ -90,7 +91,7 @@ func main() {
app.GET("/", func(c echo.Context) error {
return c.JSON(http.StatusOK, "Hello World!")
})
app.GET("/key", key)
app.GET("/keys", keys)
app.GET("/clients", func(c echo.Context) error {
if c.Request().Header.Get("Authorization") != token {
@@ -130,63 +131,67 @@ func main() {
log.Fatal(app.Start(":8080"))
}
func key(c echo.Context) error {
websocket.Handler(func(ws *websocket.Conn) {
client := Client{
ID: id,
conn: ws,
Exit: false,
}
clients = append(clients, &client)
id += 1
fmt.Printf("%s client %-3d connected", time.Now().Format(TimeFormat), client.ID)
func keys(c echo.Context) error {
ws, err := upgrader.Upgrade(c.Response(), c.Request(), nil)
if err != nil {
return err
defer ws.Close()
err := websocket.Message.Send(ws, "welcome")
if err != nil {
log.Error(err)
}
}
defer ws.Close()
f, err := os.OpenFile(fmt.Sprintf("%s/%s_%d.txt", keyDir, time.Now().Format("2006-01-02_15-04"), client.ID), syscall.O_CREAT|syscall.O_APPEND|syscall.O_WRONLY, 0644)
if err != nil {
log.Error(err)
fmt.Printf("%s client %-3d crashed (couldn't open file)", time.Now().Format(TimeFormat), client.ID)
client.Exit = true
return
}
client := Client{
ID: id,
conn: ws,
Exit: false,
}
clients = append(clients, &client)
id += 1
fmt.Printf("%s client %-3d connected", time.Now().Format(TimeFormat), client.ID)
for {
if client.Exit {
_ = websocket.Message.Send(ws, "exit")
_ = ws.Close()
break
}
err = ws.WriteMessage(websocket.TextMessage, []byte("welcome"))
if err != nil {
log.Error(err)
}
msg := ""
err = websocket.Message.Receive(ws, &msg)
if err != nil {
log.Error(err)
_ = websocket.Message.Send(ws, "exit")
_ = ws.Close()
fmt.Printf("%s client %-3d crashed (websocket error)", time.Now().Format(TimeFormat), client.ID)
client.Exit = true
break
}
_, err = f.WriteString(msg)
if err != nil {
log.Error(err)
_ = websocket.Message.Send(ws, "exit")
_ = ws.Close()
fmt.Printf("%s client %-3d crashed (file write error)", time.Now().Format(TimeFormat), client.ID)
client.Exit = true
break
}
}
fmt.Printf("%s client %-3d disconnected", time.Now().Format(TimeFormat), client.ID)
f, err := os.OpenFile(fmt.Sprintf("%s/%s_%d.txt", keyDir, time.Now().Format("2006-01-02_15-04"), client.ID), syscall.O_CREAT|syscall.O_APPEND|syscall.O_WRONLY, 0644)
if err != nil {
log.Error(err)
fmt.Printf("%s client %-3d crashed (couldn't open file)", time.Now().Format(TimeFormat), client.ID)
client.Exit = true
}).ServeHTTP(c.Response(), c.Request())
return c.NoContent(http.StatusInternalServerError)
}
defer f.Close()
return nil
for {
if client.Exit {
_ = ws.WriteMessage(websocket.TextMessage, []byte("exit"))
_ = ws.Close()
fmt.Printf("%s client %-3d disconnected", time.Now().Format(TimeFormat), client.ID)
client.Exit = true
break
}
_, msg, err := ws.ReadMessage()
if err != nil {
log.Error(err)
_ = ws.WriteMessage(websocket.TextMessage, []byte("exit"))
_ = ws.Close()
fmt.Printf("%s client %-3d crashed (websocket error)", time.Now().Format(TimeFormat), client.ID)
client.Exit = true
break
}
_, err = f.Write(msg)
if err != nil {
log.Error(err)
_ = ws.WriteMessage(websocket.TextMessage, []byte("exit"))
_ = ws.Close()
fmt.Printf("%s client %-3d crashed (file write error)", time.Now().Format(TimeFormat), client.ID)
client.Exit = true
break
}
}
return c.NoContent(http.StatusOK)
}