115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package mockserver
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
"os"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/ESilva15/TelemetryMockserver/constants"
|
|
bngsdk "github.com/ESilva15/gobngsdk"
|
|
)
|
|
|
|
type recorderViewData struct {
|
|
TotalBytes int64
|
|
Og bngsdk.Outgauge
|
|
}
|
|
|
|
type Recorder struct {
|
|
SDK *bngsdk.BeamNGSDK
|
|
// Views
|
|
mut sync.RWMutex
|
|
viewDataMut sync.RWMutex
|
|
viewData recorderViewData
|
|
viewCh chan *recorderViewData
|
|
}
|
|
|
|
func NewRecorder(fp string, address string, port int) (*Recorder, error) {
|
|
var recorder Recorder
|
|
var err error
|
|
|
|
recorder.SDK, err = bngsdk.NewBngSDK(bngsdk.Options{
|
|
Logger: slog.Default().With("SDK", "BeamNG"),
|
|
SourceType: bngsdk.UDPData,
|
|
ImportUDPAddress: address,
|
|
ImportUDPPort: port,
|
|
ExportData: true,
|
|
ExportDataType: bngsdk.BinaryFile,
|
|
ExportDataPath: fp,
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
recorder.viewData = recorderViewData{}
|
|
recorder.viewCh = make(chan *recorderViewData, 1)
|
|
|
|
return &recorder, nil
|
|
}
|
|
|
|
func (r *Recorder) Close() {
|
|
r.SDK.Close()
|
|
}
|
|
|
|
func (r *Recorder) view(ctx context.Context) {
|
|
var buf bytes.Buffer
|
|
|
|
buf.Grow(2048)
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case viewData := <-r.viewCh:
|
|
buf.Reset()
|
|
buf.WriteString("\x1b[2J\x1b[H")
|
|
fmt.Fprintf(&buf, "\x1b]0;%s - Recording ", constants.ProgramName)
|
|
stringifyRecordingProgress(&buf, viewData.TotalBytes)
|
|
fmt.Fprintf(&buf, "\x07")
|
|
|
|
stringifyRecordingProgress(&buf, viewData.TotalBytes)
|
|
fmt.Fprintf(&buf, "\n\n")
|
|
|
|
r.viewDataMut.RLock()
|
|
stringifyOutgaugeData(&buf, &viewData.Og)
|
|
r.viewDataMut.RUnlock()
|
|
|
|
_, _ = buf.WriteTo(os.Stdout)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Record records data from the UDP connection created by address and port
|
|
func (r *Recorder) Record(ctx context.Context) error {
|
|
ticker := time.NewTicker(time.Second / 60)
|
|
defer ticker.Stop()
|
|
|
|
go r.view(ctx)
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return nil
|
|
case <-ticker.C:
|
|
og, err := r.SDK.Update()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
r.viewData.TotalBytes = r.SDK.GetTotalWritten()
|
|
r.viewData.Og = *og
|
|
|
|
// Send the data to the view
|
|
select {
|
|
case r.viewCh <- &r.viewData:
|
|
// Sent the data
|
|
default:
|
|
// Dropped the frame!
|
|
}
|
|
}
|
|
}
|
|
}
|