lee/backrest
1
0
Fork 0
mirror of https://github.com/garethgeorge/backrest.git synced 2025-12-14 17:45:36 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add webui

Browse Source
This commit is contained in:
garethgeorge
2023-11-11 00:39:24 -08:00
parent f365913430
commit b57ff969dd
26 changed files with 4553 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
internal/api/server.go
View File

@@ -3,6 +3,8 @@ package api
import (
"context"
"fmt"
"sync"
"sync/atomic"
"time"
v1 "github.com/garethgeorge/resticui/gen/go/v1"
@@ -11,17 +13,44 @@ import (
"google.golang.org/protobuf/types/known/emptypb"
)
type server struct {
type Server struct {
*v1.UnimplementedResticUIServer
reqId atomic.Uint64
eventChannelsMu sync.Mutex
eventChannels map[uint64]chan *v1.Event
}
var _ v1.ResticUIServer = &server{}
var _ v1.ResticUIServer = &Server{}
func (s *server) GetConfig(ctx context.Context, empty *emptypb.Empty) (*v1.Config, error) {
func NewServer() *Server {
s := &Server{
eventChannels: make(map[uint64]chan *v1.Event),
}
go func() {
for {
time.Sleep(3 * time.Second)
s.PublishEvent(&v1.Event{
Event: &v1.Event_Log{
Log: &v1.LogEvent{
Message: fmt.Sprintf("event push test, it is %v", time.Now().Format(time.RFC3339)),
},
},
})
}
}()
return s
}
// GetConfig implements GET /v1/config
func (s *Server) GetConfig(ctx context.Context, empty *emptypb.Empty) (*v1.Config, error) {
return config.Default.Get()
}
func (s *server) SetConfig(ctx context.Context, c *v1.Config) (*v1.Config, error) {
// SetConfig implements PUT /v1/config
func (s *Server) SetConfig(ctx context.Context, c *v1.Config) (*v1.Config, error) {
err := config.Default.Update(c)
if err != nil {
return nil, fmt.Errorf("failed to update config: %w", err)
@@ -29,27 +58,33 @@ func (s *server) SetConfig(ctx context.Context, c *v1.Config) (*v1.Config, error
return config.Default.Get()
}
func (s *server) GetEvents(_ *emptypb.Empty, stream v1.ResticUI_GetEventsServer) error {
// GetEvents implements GET /v1/events
func (s *Server) GetEvents(_ *emptypb.Empty, stream v1.ResticUI_GetEventsServer) error {
reqId := s.reqId.Add(1)
// Register a channel to receive events for this invocation
s.eventChannelsMu.Lock()
eventChan := make(chan *v1.Event, 3)
s.eventChannels[reqId] = eventChan
s.eventChannelsMu.Unlock()
defer delete(s.eventChannels, reqId)
for {
zap.S().Info("Sending event")
stream.Send(&v1.Event{
Timestamp: 0,
Event: &v1.Event_BackupStatusChange{
BackupStatusChange: &v1.BackupStatusEvent{
Status: v1.Status_IN_PROGRESS,
Percent: 0,
Plan: "myplan",
},
},
})
timer := time.NewTimer(time.Second * 1)
select {
case <-stream.Context().Done():
zap.S().Info("Get events hangup")
return nil
case <-timer.C:
case event := <-eventChan:
stream.Send(event)
}
}
}
// PublishEvent publishes an event to all GetEvents streams. It is effectively a multicast.
func (s *Server) PublishEvent(event *v1.Event) {
zap.S().Debug("Publishing event", zap.Any("event", event))
s.eventChannelsMu.Lock()
defer s.eventChannelsMu.Unlock()
for _, ch := range s.eventChannels {
ch <- event
}
}