rnsrk/wisski-cloud-distillery
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

websocket actions: Refactor registration

Browse source
This commit is contained in:
Tom Wiesing 2023-11-09 14:58:19 +01:00
parent 08ab7b4383
commit 7b5f8a9882
No known key found for this signature in database
4 changed files with 102 additions and 87 deletions
Download patch file Download diff file Expand all files Collapse all files

86
internal/dis/component/server/admin/socket/actions.go
View file

@ -3,42 +3,80 @@ package socket
import (
"context"
"io"
"github.com/FAU-CDI/wisski-distillery/internal/dis/component/server/admin/socket/actions"
"github.com/FAU-CDI/wisski-distillery/internal/dis/component/server/admin/socket/proto"
"github.com/rs/zerolog"
)
func (sockets *Sockets) Actions() ActionMap {
actions := make(ActionMap, len(sockets.dependencies.Actions)+len(sockets.dependencies.IActions))
func (sockets *Sockets) Actions(ctx context.Context) proto.ActionMap {
logger := zerolog.Ctx(ctx)
actions := make(proto.ActionMap, len(sockets.dependencies.Actions)+len(sockets.dependencies.IActions))
// setup basic actions
for _, a := range sockets.dependencies.Actions {
a := a
meta := a.Action()
actions[meta.Name] = Action{
NumParams: meta.NumParams,
Scope: meta.Scope,
ScopeParam: meta.ScopeParam,
Handle: a.Act,
name, action := sockets.regularAction(a)
if _, ok := actions[name]; ok {
logger.Warn().Str("name", name).Str("type", "regular").Msg("duplicate websocket action")
}
actions[name] = action
logger.Info().
Str("name", name).
Str("type", "regular").
Int("params", action.NumParams).
Str("scope", string(action.Scope)).
Str("scopeParam", action.ScopeParam).
Msg("registering websocket action")
}
// setup instance actions
for _, a := range sockets.dependencies.IActions {
a := a
meta := a.Action()
actions[meta.Name] = Action{
NumParams: meta.NumParams + 1,
Scope: meta.Scope,
ScopeParam: meta.ScopeParam,
Handle: func(ctx context.Context, in io.Reader, out io.Writer, params ...string) error {
instance, err := sockets.dependencies.Instances.WissKI(ctx, params[0])
if err != nil {
return err
}
return a.Act(ctx, instance, in, out, params[1:]...)
},
name, action := sockets.instanceAction(a)
if _, ok := actions[name]; ok {
zerolog.Ctx(ctx).Warn().Str("name", name).Str("type", "instance").Msg("duplicate websocket action")
}
actions[name] = action
logger.Info().
Str("name", name).
Str("type", "instance").
Int("params", action.NumParams-1).
Str("scope", string(action.Scope)).
Str("scopeParam", action.ScopeParam).
Msg("registering websocket action")
}
return actions
}
func (sockets *Sockets) regularAction(a actions.WebsocketAction) (name string, action proto.Action) {
meta := a.Action()
return meta.Name, proto.Action{
NumParams: meta.NumParams + 1,
Scope: meta.Scope,
ScopeParam: meta.ScopeParam,
Handle: a.Act,
}
}
func (sockets *Sockets) instanceAction(a actions.WebsocketInstanceAction) (name string, action proto.Action) {
meta := a.Action()
return meta.Name, proto.Action{
NumParams: meta.NumParams + 1,
Scope: meta.Scope,
ScopeParam: meta.ScopeParam,
Handle: func(ctx context.Context, in io.Reader, out io.Writer, params ...string) error {
instance, err := sockets.dependencies.Instances.WissKI(ctx, params[0])
if err != nil {
return err
}
return a.Act(ctx, instance, in, out, params[1:]...)
},
}
}