🎉 Initial Commit

pkg/internal/grpc/services.go

@@ -0,0 +1,70 @@
+package grpc
+
+import (
+	"context"
+	"fmt"
+	"git.solsynth.dev/hydrogen/dealer/pkg/internal/directory"
+	"git.solsynth.dev/hydrogen/dealer/pkg/proto"
+	"github.com/samber/lo"
+)
+
+func warpServiceInstanceToInfo(in *directory.ServiceInstance) *proto.ServiceInfo {
+	return &proto.ServiceInfo{
+		Id:       in.ID,
+		Type:     in.Type,
+		Label:    in.Label,
+		GrpcAddr: in.GrpcAddr,
+		HttpAddr: in.HttpAddr,
+	}
+}
+
+func (v *Server) GetService(ctx context.Context, request *proto.GetServiceRequest) (*proto.GetServiceResponse, error) {
+	if request.Id != nil {
+		out := directory.GetServiceInstance(request.GetId())
+		return &proto.GetServiceResponse{
+			Data: warpServiceInstanceToInfo(out),
+		}, nil
+	}
+	if request.Type != nil {
+		out := directory.GetServiceInstanceByType(request.GetType())
+		return &proto.GetServiceResponse{
+			Data: warpServiceInstanceToInfo(out),
+		}, nil
+	}
+	return nil, fmt.Errorf("no filter condition is provided")
+}
+
+func (v *Server) ListService(ctx context.Context, request *proto.ListServiceRequest) (*proto.ListServiceResponse, error) {
+	var out []*directory.ServiceInstance
+	if request.Type != nil {
+		out = directory.ListServiceInstanceByType(request.GetType())
+	} else {
+		out = directory.ListServiceInstance()
+	}
+	return &proto.ListServiceResponse{
+		Data: lo.Map(out, func(item *directory.ServiceInstance, index int) *proto.ServiceInfo {
+			return warpServiceInstanceToInfo(item)
+		}),
+	}, nil
+}
+
+func (v *Server) AddService(ctx context.Context, info *proto.ServiceInfo) (*proto.AddServiceResponse, error) {
+	in := &directory.ServiceInstance{
+		ID:       info.GetId(),
+		Type:     info.GetType(),
+		Label:    info.GetLabel(),
+		GrpcAddr: info.GetGrpcAddr(),
+		HttpAddr: info.HttpAddr,
+	}
+	directory.AddServiceInstance(in)
+	return &proto.AddServiceResponse{
+		IsSuccess: true,
+	}, nil
+}
+
+func (v *Server) RemoveService(ctx context.Context, request *proto.RemoveServiceRequest) (*proto.RemoveServiceResponse, error) {
+	directory.RemoveServiceInstance(request.GetId())
+	return &proto.RemoveServiceResponse{
+		IsSuccess: true,
+	}, nil
+}