Compare commits

...

2 commits

Author SHA1 Message Date
bdfd5c3b84 feat: remove unused client mutex
- remove unused Mutex from `models.Client` struct
- remove lock from `HandleConnection()`
- replace websocket.Conn with models.Client in `handleClientCommands()`

🤖
2024-10-30 08:09:41 +01:00
94b6786c7c feat: implement asynchronous timer start
- add new StartAsync method to Timer for non-blocking execution
- use new method in RunPomodoro()

🤖
2024-10-30 07:46:29 +01:00
5 changed files with 17 additions and 16 deletions

View file

@ -42,7 +42,7 @@ func RunPomodoro() {
// Work // Work
shared.State.Mode = "Work" shared.State.Mode = "Work"
go timer.Start(pomodoroConfig.Work) timer.StartAsync(pomodoroConfig.Work)
if !waitForTimer(timer) { if !waitForTimer(timer) {
break break
} }
@ -50,13 +50,13 @@ func RunPomodoro() {
// Breaks // Breaks
if session < pomodoroConfig.Sessions { if session < pomodoroConfig.Sessions {
shared.State.Mode = "ShortBreak" shared.State.Mode = "ShortBreak"
go timer.Start(pomodoroConfig.ShortBreak) timer.StartAsync(pomodoroConfig.ShortBreak)
if !waitForTimer(timer) { if !waitForTimer(timer) {
break break
} }
} else { // last phase, prepare for finish } else { // last phase, prepare for finish
shared.State.Mode = "LongBreak" shared.State.Mode = "LongBreak"
go timer.Start(pomodoroConfig.LongBreak) timer.StartAsync(pomodoroConfig.LongBreak)
if !waitForTimer(timer) { if !waitForTimer(timer) {
break break
} }

View file

@ -24,7 +24,12 @@ func (t Timer) Init() Timer {
} }
} }
// Start the timer // Start the timer (goroutine)
func (t *Timer) StartAsync(duration int) {
go t.Start(duration)
}
// Start the timer (blocking)
func (t *Timer) Start(duration int) { func (t *Timer) Start(duration int) {
tick := time.NewTicker(time.Second) tick := time.NewTicker(time.Second)
for timeLeft := duration; timeLeft > 0; { for timeLeft := duration; timeLeft > 0; {

View file

@ -3,7 +3,6 @@ package websocket
import ( import (
"encoding/json" "encoding/json"
"github.com/charmbracelet/log" "github.com/charmbracelet/log"
"github.com/gorilla/websocket"
"git.smsvc.net/pomodoro/GoTomato/internal/pomodoro" "git.smsvc.net/pomodoro/GoTomato/internal/pomodoro"
"git.smsvc.net/pomodoro/GoTomato/internal/shared" "git.smsvc.net/pomodoro/GoTomato/internal/shared"
@ -11,7 +10,8 @@ import (
) )
// Listens for commands from a client and handles them // Listens for commands from a client and handles them
func handleClientCommands(ws *websocket.Conn) { func handleClientCommands(c models.Client) {
ws := c.Conn
for { for {
var clientCommand models.ClientCommand var clientCommand models.ClientCommand

View file

@ -31,12 +31,13 @@ func HandleConnection(w http.ResponseWriter, r *http.Request) {
log.Info("Client connected", "host", ws.NetConn().RemoteAddr(), "clients", len(Clients)+1) log.Info("Client connected", "host", ws.NetConn().RemoteAddr(), "clients", len(Clients)+1)
// Register the new client // Register the new client
mu.Lock() client := models.Client{
Clients[ws] = &models.Client{ Conn: ws,
Conn: ws, // Store the WebSocket connection
} }
mu.Lock()
Clients[ws] = &client
mu.Unlock() mu.Unlock()
// Listen for commands from the connected client // Listen for commands from the connected client
handleClientCommands(ws) handleClientCommands(client)
} }

View file

@ -3,7 +3,6 @@ package models
import ( import (
"github.com/charmbracelet/log" "github.com/charmbracelet/log"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
"sync"
) )
// Represents a command from the client (start/stop) // Represents a command from the client (start/stop)
@ -15,16 +14,12 @@ type ClientCommand struct {
// Represents a single client // Represents a single client
type Client struct { type Client struct {
Conn *websocket.Conn // Websocket connection of the client Conn *websocket.Conn
Mutex sync.Mutex // Mutex used to lock
} }
// Sends a message to the websocket. // Sends a message to the websocket.
// Automatically locks and unlocks the client mutex, to ensure that only one goroutine can write at a time. // Automatically locks and unlocks the client mutex, to ensure that only one goroutine can write at a time.
func (c *Client) SendMessage(messageType int, data []byte) error { func (c *Client) SendMessage(messageType int, data []byte) error {
c.Mutex.Lock()
defer c.Mutex.Unlock()
err := c.Conn.WriteMessage(messageType, data) err := c.Conn.WriteMessage(messageType, data)
if err != nil { if err != nil {
log.Error("Error writing to WebSocket:", "msg", err) log.Error("Error writing to WebSocket:", "msg", err)