MaMe82
122 lines
2.7 KiB
Go
122 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
pb "github.com/mame82/P4wnP1_go/proto/gopherjs"
|
|
"context"
|
|
"sync"
|
|
"github.com/johanbrandhorst/protobuf/grpcweb"
|
|
"time"
|
|
)
|
|
|
|
type Rpc struct {
|
|
*sync.Mutex
|
|
Client pb.P4WNP1Client
|
|
eventListeningOn bool
|
|
eventListeningCtx *context.Context
|
|
eventListeningCancel *context.CancelFunc
|
|
}
|
|
|
|
/*
|
|
func (rpc *Rpc) StartListenEvents(evtType int64) (err error) {
|
|
rpc.Lock()
|
|
if rpc.eventListeningOn {
|
|
rpc.Unlock()
|
|
return errors.New("Already listening to events")
|
|
}
|
|
// shouldn't happen
|
|
if rpc.eventListeningCancel != nil {
|
|
//Cancel old eventListeners
|
|
cancel := *rpc.eventListeningCancel
|
|
cancel()
|
|
}
|
|
|
|
ctx,cancel := context.WithCancel(context.Background())
|
|
rpc.eventListeningCtx = &ctx
|
|
rpc.eventListeningCancel = &cancel
|
|
rpc.eventListeningOn = true
|
|
rpc.Unlock()
|
|
|
|
|
|
evStream, err := rpc.Client.EventListen(ctx, &pb.EventRequest{ListenType: evtType})
|
|
if err != nil { return err }
|
|
|
|
go func() {
|
|
defer rpc.StopEventListening()
|
|
for {
|
|
event, err := evStream.Recv()
|
|
if err == io.EOF { break }
|
|
if err != nil { return }
|
|
|
|
println("Event: ", event)
|
|
}
|
|
return
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (rpc *Rpc) StopEventListening() {
|
|
rpc.Lock()
|
|
if rpc.eventListeningCancel != nil {
|
|
(*rpc.eventListeningCancel)()
|
|
}
|
|
rpc.eventListeningCancel = nil
|
|
rpc.eventListeningCtx= nil
|
|
rpc.eventListeningOn = false
|
|
rpc.Unlock()
|
|
}
|
|
*/
|
|
|
|
func NewRpcClient(addr string) Rpc {
|
|
rcl := Rpc{}
|
|
rcl.Mutex = &sync.Mutex{}
|
|
cl := pb.NewP4WNP1Client(addr, grpcweb.WithDefaultCallOptions(grpcweb.ForceWebsocketTransport()))
|
|
rcl.Client = cl
|
|
return rcl
|
|
}
|
|
|
|
func RpcGetDeployedGadgetSettings(timeout time.Duration) (*pb.GadgetSettings, error) {
|
|
//gs := vue.GetVM(c).Get("gadgetSettings")
|
|
println("RpcGetDeployedGadgetSettings called")
|
|
|
|
ctx,cancel := context.WithTimeout(context.Background(), timeout)
|
|
defer cancel()
|
|
|
|
|
|
return Client.Client.GetDeployedGadgetSetting(ctx, &pb.Empty{})
|
|
|
|
}
|
|
|
|
|
|
func RpcSetRemoteGadgetSettings(targetGS *pb.GadgetSettings, timeout time.Duration) (err error) {
|
|
//gs := vue.GetVM(c).Get("gadgetSettings")
|
|
println("RpcSetRemoteGadgetSettings called")
|
|
|
|
ctx,cancel := context.WithTimeout(context.Background(), timeout)
|
|
defer cancel()
|
|
|
|
|
|
//Set gadget settings
|
|
_, err = Client.Client.SetGadgetSettings(ctx, targetGS)
|
|
if err != nil {
|
|
//js.Global.Call("alert", "Error setting given gadget settings: " + status.Convert(err).Message())
|
|
//println(err)
|
|
//c.UpdateFromDeployedGadgetSettings(vm)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func RpcDeployRemoteGadgetSettings(timeout time.Duration) (*pb.GadgetSettings, error) {
|
|
//gs := vue.GetVM(c).Get("gadgetSettings")
|
|
println("RpcDeployRemoteGadgetSettings called")
|
|
|
|
ctx,cancel := context.WithTimeout(context.Background(), timeout)
|
|
defer cancel()
|
|
|
|
|
|
return Client.Client.DeployGadgetSetting(ctx, &pb.Empty{})
|
|
|
|
}
|