sqlite

src/main.go

@@ -3,14 +3,10 @@ package main
 import (
 	"html/template"
 	"io"
-	"strconv"
-	"strings"
-	"sync"
 
 	"github.com/labstack/echo/v4"
 	"github.com/labstack/echo/v4/middleware"
 	"github.com/labstack/gommon/log"
-	"golang.org/x/net/websocket"
 )
 
 type Templates struct {
@@ -38,113 +34,27 @@ func newPage(boxes map[int]bool) Page {
 }
 
 var (
-	wsConnections = make(map[*websocket.Conn]bool)
-	wsMutex       sync.Mutex
-	boxes         = make(map[int]bool, 1000)
-	e             = echo.New()
+	e = echo.New()
 )
 
-func broadcastMessage(message string) {
-	wsMutex.Lock()
-	defer wsMutex.Unlock()
-	for conn := range wsConnections {
-		if err := websocket.Message.Send(conn, message); err != nil {
-			conn.Close()
-			delete(wsConnections, conn)
-		}
-	}
-}
-
 func main() {
 	e.Renderer = NewTemplates()
 
 	e.Logger.SetLevel(log.DEBUG)
 	e.Use(middleware.Logger())
 
-	for i := 0; i < 1000; i++ {
-		boxes[i] = false
-	}
-
-	page := newPage(boxes)
+	boxes := getBoxes()
 
 	e.Static("/images", "images")
 	e.Static("/css", "css")
 
 	e.GET("/", func(c echo.Context) error {
-		return c.Render(200, "index", page)
+		return c.Render(200, "index", newPage(boxes))
 	})
 
 	e.GET("/ws", initWs)
 
 	e.Logger.Fatal(e.Start(":54321"))
-}
-
-func sendUpdate(index int, checked bool) {
-	message := formatMessage(index, checked)
-	broadcastMessage(message)
-}
-
-func formatMessage(index int, checked bool) string {
-	message := "u:" + strconv.Itoa(index)
-	if checked {
-		message += ":+"
-	} else {
-		message += ":-"
-	}
-	return message
-}
-
-func handleMessage(msg string, ws *websocket.Conn) error {
-	if strings.Contains(msg, "u:") {
-		parts := strings.Split(msg, ":")
-		index, err := strconv.Atoi(parts[1])
-		if err != nil {
-			return err
-		}
-		checked := parts[2] == "+"
-
-		boxes[index] = checked
-
-		sendUpdate(index, checked)
-	}
-	if strings.Contains(msg, "ping") {
-		len := len(wsConnections)
-		websocket.Message.Send(ws, "pong-"+strconv.Itoa(len))
-	}
-	return nil
-}
-
-func initWs(c echo.Context) error {
-	websocket.Handler(func(ws *websocket.Conn) {
-		wsMutex.Lock()
-		wsConnections[ws] = true
-		wsMutex.Unlock()
-
-		initialState := "i:"
-		for index, checked := range boxes {
-			if checked {
-				initialState += strconv.Itoa(index) + ";"
-			}
-		}
-
-		if err := websocket.Message.Send(ws, initialState); err != nil {
-			log.Errorf("Failed to send initial data: %v", err)
-			return
-		}
-
-		for {
-			var msg string
-			if err := websocket.Message.Receive(ws, &msg); err != nil {
-				wsMutex.Lock()
-				delete(wsConnections, ws)
-				wsMutex.Unlock()
-				ws.Close()
-				break
-			}
-			handleMessage(msg, ws)
-			log.Infof("Received message: %s", msg)
-		}
-	}).ServeHTTP(c.Response(), c.Request())
-
-	return nil
+
+	defer db.Close()
 }