90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
package directory
|
|
|
|
import (
|
|
"context"
|
|
"git.solsynth.dev/hypernet/nexus/pkg/proto"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
"io"
|
|
"time"
|
|
)
|
|
|
|
type CommandRpcServer struct {
|
|
proto.UnimplementedCommandControllerServer
|
|
}
|
|
|
|
func (c CommandRpcServer) AddCommand(ctx context.Context, info *proto.CommandInfo) (*proto.AddCommandResponse, error) {
|
|
clientId, err := GetClientId(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
service := GetServiceInstanceByType(clientId)
|
|
if service == nil {
|
|
return nil, status.Errorf(codes.NotFound, "service not found")
|
|
}
|
|
|
|
AddCommand(info.GetId(), info.GetMethod(), info.GetTags(), service)
|
|
return &proto.AddCommandResponse{
|
|
IsSuccess: true,
|
|
}, nil
|
|
}
|
|
|
|
func (c CommandRpcServer) RemoveCommand(ctx context.Context, request *proto.CommandLookupRequest) (*proto.RemoveCommandResponse, error) {
|
|
RemoveCommand(request.GetId(), request.GetMethod())
|
|
return &proto.RemoveCommandResponse{
|
|
IsSuccess: true,
|
|
}, nil
|
|
}
|
|
|
|
func (c CommandRpcServer) SendCommand(ctx context.Context, argument *proto.CommandArgument) (*proto.CommandReturn, error) {
|
|
id := argument.GetCommand()
|
|
method := argument.GetMethod()
|
|
|
|
handler := GetCommandHandler(id, method)
|
|
if handler == nil {
|
|
return nil, status.Errorf(codes.NotFound, "command not found")
|
|
}
|
|
|
|
conn, err := handler.GetGrpcConn()
|
|
if err != nil {
|
|
return nil, status.Errorf(codes.Unavailable, "service unavailable")
|
|
}
|
|
|
|
contx, cancel := context.WithTimeout(context.Background(), time.Second*10)
|
|
defer cancel()
|
|
|
|
return proto.NewCommandControllerClient(conn).SendCommand(contx, argument)
|
|
}
|
|
|
|
func (c CommandRpcServer) SendStreamCommand(g grpc.BidiStreamingServer[proto.CommandArgument, proto.CommandReturn]) error {
|
|
for {
|
|
pck, err := g.Recv()
|
|
if err == io.EOF {
|
|
return nil
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
id := pck.GetCommand()
|
|
method := pck.GetMethod()
|
|
|
|
handler := GetCommandHandler(id, method)
|
|
if handler == nil {
|
|
return status.Errorf(codes.NotFound, "command not found")
|
|
}
|
|
|
|
conn, err := handler.GetGrpcConn()
|
|
|
|
contx, cancel := context.WithTimeout(context.Background(), time.Second*10)
|
|
result, _ := proto.NewCommandControllerClient(conn).SendCommand(contx, pck)
|
|
cancel()
|
|
|
|
_ = g.Send(&proto.CommandReturn{
|
|
Status: result.Status,
|
|
Payload: result.Payload,
|
|
})
|
|
}
|
|
}
|