72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package transport
|
|
|
|
import (
|
|
"database/sql"
|
|
"log/slog"
|
|
"sync"
|
|
"time"
|
|
|
|
"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/models/db"
|
|
"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/requests"
|
|
"gitea.local.lab/Lbenedar/foundry_helper_service/internal/foundry/types"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
type FoundryTransport struct {
|
|
WsConn *websocket.Conn
|
|
CurrWsId int
|
|
ChanMutex sync.Mutex
|
|
ReadChan types.ReadChannels
|
|
ExchangeChan types.ExchangeChannels
|
|
|
|
Http requests.FoundryHttpRequest
|
|
IsAuth bool
|
|
|
|
Models *db.Models
|
|
IsDbInit bool
|
|
|
|
Logger *slog.Logger
|
|
}
|
|
|
|
func NewFoundryTransport(dbConn *sql.DB, logger *slog.Logger, httpConfig *requests.FoundryHttpRequest) *FoundryTransport {
|
|
return &FoundryTransport{
|
|
CurrWsId: 0,
|
|
ExchangeChan: types.ExchangeChannels{
|
|
Msgs: make(map[int]chan []byte),
|
|
ProgressMsg: map[string]chan struct{}{},
|
|
},
|
|
|
|
Http: *httpConfig,
|
|
IsAuth: false,
|
|
|
|
Models: db.NewModels(dbConn),
|
|
IsDbInit: false,
|
|
|
|
Logger: logger,
|
|
}
|
|
}
|
|
|
|
func (t *FoundryTransport) CloseMsgChannel(id int, timeout time.Duration) bool {
|
|
time.Sleep(timeout)
|
|
|
|
t.ChanMutex.Lock()
|
|
defer t.ChanMutex.Unlock()
|
|
|
|
ok := true
|
|
if _, ok = t.ExchangeChan.Msgs[id]; !ok {
|
|
return false
|
|
}
|
|
select {
|
|
case _, ok = <-t.ExchangeChan.Msgs[id]:
|
|
if ok {
|
|
close(t.ExchangeChan.Msgs[id])
|
|
delete(t.ExchangeChan.Msgs, id)
|
|
}
|
|
default:
|
|
close(t.ExchangeChan.Msgs[id])
|
|
delete(t.ExchangeChan.Msgs, id)
|
|
}
|
|
|
|
return ok
|
|
}
|