115 lines
2.6 KiB
Go
115 lines
2.6 KiB
Go
package converge
|
|
|
|
import (
|
|
"context"
|
|
"converge/pkg/models"
|
|
"converge/pkg/server/templates"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
type WebSessions struct {
|
|
mutex sync.Mutex
|
|
sessions map[*WebSession]bool
|
|
lastNotification *models.State
|
|
}
|
|
|
|
type WebSession struct {
|
|
notifications chan *models.State
|
|
conn net.Conn
|
|
}
|
|
|
|
func NewWebSessions(notifications chan *models.State) *WebSessions {
|
|
websessions := &WebSessions{
|
|
sessions: make(map[*WebSession]bool),
|
|
}
|
|
|
|
go func() {
|
|
for {
|
|
notification := <-notifications
|
|
websessions.notifyWebSessions(notification)
|
|
}
|
|
}()
|
|
return websessions
|
|
}
|
|
|
|
func (sessions *WebSessions) notifyWebSessions(notification *models.State) {
|
|
sessions.mutex.Lock()
|
|
defer sessions.mutex.Unlock()
|
|
sessions.lastNotification = notification
|
|
for session, _ := range sessions.sessions {
|
|
session.notifications <- notification
|
|
}
|
|
}
|
|
|
|
func (sessions *WebSessions) NewSession(wsConnection net.Conn) *WebSession {
|
|
sessions.mutex.Lock()
|
|
defer sessions.mutex.Unlock()
|
|
session := &WebSession{
|
|
notifications: make(chan *models.State, 10),
|
|
conn: wsConnection,
|
|
}
|
|
sessions.sessions[session] = true
|
|
sessions.logSessions()
|
|
session.notifications <- sessions.lastNotification
|
|
return session
|
|
}
|
|
|
|
func GetUserLocation(r *http.Request) (*time.Location, error) {
|
|
tzName := r.URL.Query().Get("timezone")
|
|
if tzName == "" {
|
|
tzName = r.Header.Get("X-Timezone")
|
|
}
|
|
//log.Println("GetUserLocation: timezone ", tzName)
|
|
if tzName == "" {
|
|
tzName = "UTC"
|
|
}
|
|
return time.LoadLocation(tzName)
|
|
}
|
|
|
|
func (session *WebSession) WriteNotifications(location *time.Location) {
|
|
timer := time.NewTicker(10 * time.Second)
|
|
defer timer.Stop()
|
|
for {
|
|
select {
|
|
case notification, ok := <-session.notifications:
|
|
if !ok {
|
|
log.Println("channel closed")
|
|
}
|
|
//log.Println("Got notification: ", notification)
|
|
err := templates.State(notification, location).Render(context.Background(), session.conn)
|
|
if err != nil {
|
|
log.Printf("WS connection closed: %v", err)
|
|
return
|
|
}
|
|
case <-timer.C:
|
|
// Read of 0 bytes seems to be more reliable than write
|
|
//_, err := session.conn.Write(make([]byte, 0, 0))
|
|
//if err != nil {
|
|
// log.Printf("WS connection closed: %v", err)
|
|
// return
|
|
//}
|
|
_, err := session.conn.Read(make([]byte, 0, 0))
|
|
if err != nil {
|
|
log.Printf("WS connection closed: %v", err)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (sessions *WebSessions) SessionClosed(session *WebSession) {
|
|
sessions.mutex.Lock()
|
|
defer sessions.mutex.Unlock()
|
|
|
|
delete(sessions.sessions, session)
|
|
sessions.logSessions()
|
|
}
|
|
|
|
func (sessions *WebSessions) logSessions() {
|
|
log.Printf("Web session count %d", len(sessions.sessions))
|
|
}
|