Enha: stop timer if turn ended
This commit is contained in:
@ -1,13 +1,19 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"gralias/models"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type roomTimer struct {
|
||||
ticker *time.Ticker
|
||||
done chan bool
|
||||
}
|
||||
|
||||
var (
|
||||
timers = make(map[string]*time.Ticker)
|
||||
timers = make(map[string]*roomTimer)
|
||||
mu sync.Mutex
|
||||
)
|
||||
|
||||
@ -18,33 +24,41 @@ func StartTurnTimer(roomID string, duration time.Duration) {
|
||||
if _, exists := timers[roomID]; exists {
|
||||
return // Timer already running
|
||||
}
|
||||
|
||||
ticker := time.NewTicker(1 * time.Second)
|
||||
timers[roomID] = ticker
|
||||
done := make(chan bool)
|
||||
timers[roomID] = &roomTimer{ticker: ticker, done: done}
|
||||
|
||||
go func() {
|
||||
for range ticker.C {
|
||||
room, err := getRoomByID(roomID)
|
||||
if err != nil {
|
||||
log.Error("failed to get room by id", "error", err)
|
||||
StopTurnTimer(roomID)
|
||||
for {
|
||||
select {
|
||||
case <-done:
|
||||
return
|
||||
}
|
||||
if room.Settings.TurnSecondsLeft == 0 {
|
||||
log.Info("turn time is over", "room_id", roomID)
|
||||
room.ChangeTurn()
|
||||
room.MimeDone = false
|
||||
case <-ticker.C:
|
||||
room, err := getRoomByID(roomID)
|
||||
if err != nil {
|
||||
log.Error("failed to get room by id", "error", err)
|
||||
StopTurnTimer(roomID)
|
||||
return
|
||||
}
|
||||
if room.Settings.TurnSecondsLeft <= 0 {
|
||||
log.Info("turn time is over", "room_id", roomID)
|
||||
room.ChangeTurn()
|
||||
room.MimeDone = false
|
||||
if err := saveRoom(room); err != nil {
|
||||
log.Error("failed to save room", "error", err)
|
||||
}
|
||||
notify(models.NotifyTurnTimerPrefix+room.ID, fmt.Sprintf("%d", room.Settings.TurnSecondsLeft))
|
||||
notifyBotIfNeeded(room)
|
||||
StopTurnTimer(roomID)
|
||||
return
|
||||
}
|
||||
room.Settings.TurnSecondsLeft--
|
||||
if err := saveRoom(room); err != nil {
|
||||
log.Error("failed to save room", "error", err)
|
||||
}
|
||||
notify(models.NotifyRoomUpdatePrefix+room.ID, "")
|
||||
notifyBotIfNeeded(room)
|
||||
StopTurnTimer(roomID)
|
||||
return
|
||||
}
|
||||
room.Settings.TurnSecondsLeft--
|
||||
if err := saveRoom(room); err != nil {
|
||||
log.Error("failed to save room", "error", err)
|
||||
}
|
||||
notify(models.NotifyRoomUpdatePrefix+room.ID, "")
|
||||
}
|
||||
}()
|
||||
}
|
||||
@ -52,9 +66,10 @@ func StartTurnTimer(roomID string, duration time.Duration) {
|
||||
func StopTurnTimer(roomID string) {
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
|
||||
if ticker, exists := timers[roomID]; exists {
|
||||
ticker.Stop()
|
||||
if timer, exists := timers[roomID]; exists {
|
||||
timer.ticker.Stop()
|
||||
close(timer.done)
|
||||
delete(timers, roomID)
|
||||
}
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user