From 23d14e4e4613de921e1f198077bdffcdf5dfea83 Mon Sep 17 00:00:00 2001 From: LittleSheep Date: Thu, 24 Oct 2024 23:46:09 +0800 Subject: [PATCH] :sparkles: MQ, KV and MQ allocator --- go.mod | 3 + go.sum | 6 + pkg/internal/grpc/allocator.go | 33 ++++ pkg/internal/grpc/server.go | 8 +- pkg/internal/mq/nats.go | 16 ++ pkg/main.go | 9 ++ pkg/nex/allocator.go | 34 ++++ pkg/proto/allocator.pb.go | 284 +++++++++++++++++++++++++++++++++ pkg/proto/allocator.proto | 26 +++ pkg/proto/allocator_grpc.pb.go | 159 ++++++++++++++++++ settings.toml | 3 + 11 files changed, 578 insertions(+), 3 deletions(-) create mode 100644 pkg/internal/grpc/allocator.go create mode 100644 pkg/internal/mq/nats.go create mode 100644 pkg/nex/allocator.go create mode 100644 pkg/proto/allocator.pb.go create mode 100644 pkg/proto/allocator.proto create mode 100644 pkg/proto/allocator_grpc.pb.go diff --git a/go.mod b/go.mod index ab97beb..2bc8923 100644 --- a/go.mod +++ b/go.mod @@ -69,6 +69,9 @@ require ( github.com/mitchellh/mapstructure v1.5.0 // indirect github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/nats-io/nats.go v1.37.0 // indirect + github.com/nats-io/nkeys v0.4.7 // indirect + github.com/nats-io/nuid v1.0.1 // indirect github.com/pelletier/go-toml/v2 v2.2.3 // indirect github.com/philhofer/fwd v1.1.3-0.20240916144458-20a13a1f6b7c // indirect github.com/pkg/errors v0.9.1 // indirect diff --git a/go.sum b/go.sum index b246939..85eae68 100644 --- a/go.sum +++ b/go.sum @@ -216,6 +216,12 @@ github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3Rllmb github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= github.com/mwitkow/go-conntrack v0.0.0-20161129095857-cc309e4a2223/go.mod h1:qRWi+5nqEBWmkhHvq77mSJWrCKwh8bxhgT7d/eI7P4U= +github.com/nats-io/nats.go v1.37.0 h1:07rauXbVnnJvv1gfIyghFEo6lUcYRY0WXc3x7x0vUxE= +github.com/nats-io/nats.go v1.37.0/go.mod h1:Ubdu4Nh9exXdSz0RVWRFBbRfrbSxOYd26oF0wkWclB8= +github.com/nats-io/nkeys v0.4.7 h1:RwNJbbIdYCoClSDNY7QVKZlyb/wfT6ugvFCiKy6vDvI= +github.com/nats-io/nkeys v0.4.7/go.mod h1:kqXRgRDPlGy7nGaEDMuYzmiJCIAAWDK0IMBtDmGD0nc= +github.com/nats-io/nuid v1.0.1 h1:5iA8DT8V7q8WK2EScv2padNa/rTESc1KdnPw4TC2paw= +github.com/nats-io/nuid v1.0.1/go.mod h1:19wcPz3Ph3q0Jbyiqsd0kePYG7A95tJPxeL+1OSON2c= github.com/pascaldekloe/goe v0.0.0-20180627143212-57f6aae5913c/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc= github.com/pascaldekloe/goe v0.1.0 h1:cBOtyMzM9HTpWjXfbbunk26uA6nG3a8n06Wieeh0MwY= github.com/pascaldekloe/goe v0.1.0/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc= diff --git a/pkg/internal/grpc/allocator.go b/pkg/internal/grpc/allocator.go new file mode 100644 index 0000000..142a2c7 --- /dev/null +++ b/pkg/internal/grpc/allocator.go @@ -0,0 +1,33 @@ +package grpc + +import ( + "context" + "git.solsynth.dev/hypernet/nexus/pkg/internal/kv" + "git.solsynth.dev/hypernet/nexus/pkg/internal/mq" + "git.solsynth.dev/hypernet/nexus/pkg/proto" + "github.com/spf13/viper" + "google.golang.org/grpc/codes" + "google.golang.org/grpc/status" +) + +func (v *Server) AllocMessageQueue(ctx context.Context, request *proto.AllocMqRequest) (*proto.AllocMqResponse, error) { + if mq.Kmq == nil { + return &proto.AllocMqResponse{IsSuccess: false}, status.Error(codes.Unavailable, "message queue wasn't configured") + } + + return &proto.AllocMqResponse{ + IsSuccess: true, + Addr: viper.GetString("mq.addr"), + }, nil +} + +func (v *Server) AllocKv(ctx context.Context, request *proto.AllocKvRequest) (*proto.AllocKvResponse, error) { + if kv.Kv == nil { + return &proto.AllocKvResponse{IsSuccess: false}, status.Error(codes.Unavailable, "kv wasn't configured") + } + + return &proto.AllocKvResponse{ + IsSuccess: true, + Endpoints: viper.GetStringSlice("kv.endpoints"), + }, nil +} diff --git a/pkg/internal/grpc/server.go b/pkg/internal/grpc/server.go index c6f2fa8..1db8a6e 100644 --- a/pkg/internal/grpc/server.go +++ b/pkg/internal/grpc/server.go @@ -1,7 +1,7 @@ package grpc import ( - directory2 "git.solsynth.dev/hypernet/nexus/pkg/internal/directory" + "git.solsynth.dev/hypernet/nexus/pkg/internal/directory" "net" "git.solsynth.dev/hypernet/nexus/pkg/proto" @@ -17,6 +17,7 @@ import ( type Server struct { proto.UnimplementedDatabaseServiceServer proto.UnimplementedStreamServiceServer + proto.UnimplementedAllocatorServiceServer health.UnimplementedHealthServer srv *grpc.Server @@ -27,10 +28,11 @@ func NewServer() *Server { srv: grpc.NewServer(), } - proto.RegisterDirectoryServiceServer(server.srv, &directory2.ServiceRpcServer{}) - proto.RegisterCommandProviderServer(server.srv, &directory2.CommandRpcServer{}) + proto.RegisterDirectoryServiceServer(server.srv, &directory.ServiceRpcServer{}) + proto.RegisterCommandProviderServer(server.srv, &directory.CommandRpcServer{}) proto.RegisterDatabaseServiceServer(server.srv, server) proto.RegisterStreamServiceServer(server.srv, server) + proto.RegisterAllocatorServiceServer(server.srv, server) health.RegisterHealthServer(server.srv, server) reflection.Register(server.srv) diff --git a/pkg/internal/mq/nats.go b/pkg/internal/mq/nats.go new file mode 100644 index 0000000..1f3e6f2 --- /dev/null +++ b/pkg/internal/mq/nats.go @@ -0,0 +1,16 @@ +package mq + +import "github.com/nats-io/nats.go" + +var Kmq *nats.Conn + +func ConnectNats(in string) error { + nc, err := nats.Connect(in) + if err != nil { + return err + } else { + Kmq = nc + } + + return nil +} diff --git a/pkg/main.go b/pkg/main.go index 7d0651e..2599f62 100644 --- a/pkg/main.go +++ b/pkg/main.go @@ -7,6 +7,7 @@ import ( "git.solsynth.dev/hypernet/nexus/pkg/internal/directory" "git.solsynth.dev/hypernet/nexus/pkg/internal/http" "git.solsynth.dev/hypernet/nexus/pkg/internal/kv" + "git.solsynth.dev/hypernet/nexus/pkg/internal/mq" "git.solsynth.dev/hypernet/nexus/pkg/nex/sec" "github.com/fatih/color" "os" @@ -58,6 +59,14 @@ func main() { log.Info().Msg("Connected to kv (etcd)!") } + // Connect to message queue (nats) + log.Info().Msg("Connecting to MQ (nats)...") + if err := mq.ConnectNats(viper.GetString("mq.addr")); err != nil { + log.Error().Err(err).Msg("An error occurred when connecting to MQ (nats). MQ related feature will be disabled.") + } else { + log.Info().Msg("Connected to MQ (nats)!") + } + // Connect to database log.Info().Msg("Connecting to database...") if db, err := database.Connect(viper.GetString("database.dsn")); err != nil { diff --git a/pkg/nex/allocator.go b/pkg/nex/allocator.go new file mode 100644 index 0000000..905b83e --- /dev/null +++ b/pkg/nex/allocator.go @@ -0,0 +1,34 @@ +package nex + +import ( + "context" + "git.solsynth.dev/hypernet/nexus/pkg/proto" +) + +type AllocatableResourceType = string + +const ( + AllocatableResourceMq = AllocatableResourceType("mq") + AllocatableResourceKv = AllocatableResourceType("kv") +) + +func (v *Conn) AllocResource(t AllocatableResourceType) any { + switch t { + case AllocatableResourceMq: + conn := v.GetNexusGrpcConn() + resp, err := proto.NewAllocatorServiceClient(conn).AllocMessageQueue(context.Background(), &proto.AllocMqRequest{}) + if err != nil || !resp.IsSuccess { + return nil + } + return resp.Addr + case AllocatableResourceKv: + conn := v.GetNexusGrpcConn() + resp, err := proto.NewAllocatorServiceClient(conn).AllocKv(context.Background(), &proto.AllocKvRequest{}) + if err != nil || !resp.IsSuccess { + return nil + } + return resp.Endpoints + default: + return nil + } +} diff --git a/pkg/proto/allocator.pb.go b/pkg/proto/allocator.pb.go new file mode 100644 index 0000000..5bd4d00 --- /dev/null +++ b/pkg/proto/allocator.pb.go @@ -0,0 +1,284 @@ +// Code generated by protoc-gen-go. DO NOT EDIT. +// versions: +// protoc-gen-go v1.35.1 +// protoc v5.28.2 +// source: allocator.proto + +package proto + +import ( + protoreflect "google.golang.org/protobuf/reflect/protoreflect" + protoimpl "google.golang.org/protobuf/runtime/protoimpl" + reflect "reflect" + sync "sync" +) + +const ( + // Verify that this generated code is sufficiently up-to-date. + _ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) + // Verify that runtime/protoimpl is sufficiently up-to-date. + _ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) +) + +type AllocMqRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields +} + +func (x *AllocMqRequest) Reset() { + *x = AllocMqRequest{} + mi := &file_allocator_proto_msgTypes[0] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) +} + +func (x *AllocMqRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*AllocMqRequest) ProtoMessage() {} + +func (x *AllocMqRequest) ProtoReflect() protoreflect.Message { + mi := &file_allocator_proto_msgTypes[0] + if x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use AllocMqRequest.ProtoReflect.Descriptor instead. +func (*AllocMqRequest) Descriptor() ([]byte, []int) { + return file_allocator_proto_rawDescGZIP(), []int{0} +} + +type AllocMqResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + IsSuccess bool `protobuf:"varint,1,opt,name=is_success,json=isSuccess,proto3" json:"is_success,omitempty"` + Addr string `protobuf:"bytes,2,opt,name=addr,proto3" json:"addr,omitempty"` +} + +func (x *AllocMqResponse) Reset() { + *x = AllocMqResponse{} + mi := &file_allocator_proto_msgTypes[1] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) +} + +func (x *AllocMqResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*AllocMqResponse) ProtoMessage() {} + +func (x *AllocMqResponse) ProtoReflect() protoreflect.Message { + mi := &file_allocator_proto_msgTypes[1] + if x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use AllocMqResponse.ProtoReflect.Descriptor instead. +func (*AllocMqResponse) Descriptor() ([]byte, []int) { + return file_allocator_proto_rawDescGZIP(), []int{1} +} + +func (x *AllocMqResponse) GetIsSuccess() bool { + if x != nil { + return x.IsSuccess + } + return false +} + +func (x *AllocMqResponse) GetAddr() string { + if x != nil { + return x.Addr + } + return "" +} + +type AllocKvRequest struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields +} + +func (x *AllocKvRequest) Reset() { + *x = AllocKvRequest{} + mi := &file_allocator_proto_msgTypes[2] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) +} + +func (x *AllocKvRequest) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*AllocKvRequest) ProtoMessage() {} + +func (x *AllocKvRequest) ProtoReflect() protoreflect.Message { + mi := &file_allocator_proto_msgTypes[2] + if x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use AllocKvRequest.ProtoReflect.Descriptor instead. +func (*AllocKvRequest) Descriptor() ([]byte, []int) { + return file_allocator_proto_rawDescGZIP(), []int{2} +} + +type AllocKvResponse struct { + state protoimpl.MessageState + sizeCache protoimpl.SizeCache + unknownFields protoimpl.UnknownFields + + IsSuccess bool `protobuf:"varint,1,opt,name=is_success,json=isSuccess,proto3" json:"is_success,omitempty"` + Endpoints []string `protobuf:"bytes,2,rep,name=endpoints,proto3" json:"endpoints,omitempty"` +} + +func (x *AllocKvResponse) Reset() { + *x = AllocKvResponse{} + mi := &file_allocator_proto_msgTypes[3] + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + ms.StoreMessageInfo(mi) +} + +func (x *AllocKvResponse) String() string { + return protoimpl.X.MessageStringOf(x) +} + +func (*AllocKvResponse) ProtoMessage() {} + +func (x *AllocKvResponse) ProtoReflect() protoreflect.Message { + mi := &file_allocator_proto_msgTypes[3] + if x != nil { + ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x)) + if ms.LoadMessageInfo() == nil { + ms.StoreMessageInfo(mi) + } + return ms + } + return mi.MessageOf(x) +} + +// Deprecated: Use AllocKvResponse.ProtoReflect.Descriptor instead. +func (*AllocKvResponse) Descriptor() ([]byte, []int) { + return file_allocator_proto_rawDescGZIP(), []int{3} +} + +func (x *AllocKvResponse) GetIsSuccess() bool { + if x != nil { + return x.IsSuccess + } + return false +} + +func (x *AllocKvResponse) GetEndpoints() []string { + if x != nil { + return x.Endpoints + } + return nil +} + +var File_allocator_proto protoreflect.FileDescriptor + +var file_allocator_proto_rawDesc = []byte{ + 0x0a, 0x0f, 0x61, 0x6c, 0x6c, 0x6f, 0x63, 0x61, 0x74, 0x6f, 0x72, 0x2e, 0x70, 0x72, 0x6f, 0x74, + 0x6f, 0x12, 0x05, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x10, 0x0a, 0x0e, 0x41, 0x6c, 0x6c, 0x6f, + 0x63, 0x4d, 0x71, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x22, 0x44, 0x0a, 0x0f, 0x41, 0x6c, + 0x6c, 0x6f, 0x63, 0x4d, 0x71, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x1d, 0x0a, + 0x0a, 0x69, 0x73, 0x5f, 0x73, 0x75, 0x63, 0x63, 0x65, 0x73, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, + 0x08, 0x52, 0x09, 0x69, 0x73, 0x53, 0x75, 0x63, 0x63, 0x65, 0x73, 0x73, 0x12, 0x12, 0x0a, 0x04, + 0x61, 0x64, 0x64, 0x72, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x61, 0x64, 0x64, 0x72, + 0x22, 0x10, 0x0a, 0x0e, 0x41, 0x6c, 0x6c, 0x6f, 0x63, 0x4b, 0x76, 0x52, 0x65, 0x71, 0x75, 0x65, + 0x73, 0x74, 0x22, 0x4e, 0x0a, 0x0f, 0x41, 0x6c, 0x6c, 0x6f, 0x63, 0x4b, 0x76, 0x52, 0x65, 0x73, + 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x1d, 0x0a, 0x0a, 0x69, 0x73, 0x5f, 0x73, 0x75, 0x63, 0x63, + 0x65, 0x73, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, 0x08, 0x52, 0x09, 0x69, 0x73, 0x53, 0x75, 0x63, + 0x63, 0x65, 0x73, 0x73, 0x12, 0x1c, 0x0a, 0x09, 0x65, 0x6e, 0x64, 0x70, 0x6f, 0x69, 0x6e, 0x74, + 0x73, 0x18, 0x02, 0x20, 0x03, 0x28, 0x09, 0x52, 0x09, 0x65, 0x6e, 0x64, 0x70, 0x6f, 0x69, 0x6e, + 0x74, 0x73, 0x32, 0x94, 0x01, 0x0a, 0x10, 0x41, 0x6c, 0x6c, 0x6f, 0x63, 0x61, 0x74, 0x6f, 0x72, + 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0x44, 0x0a, 0x11, 0x41, 0x6c, 0x6c, 0x6f, 0x63, + 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x51, 0x75, 0x65, 0x75, 0x65, 0x12, 0x15, 0x2e, 0x70, + 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x41, 0x6c, 0x6c, 0x6f, 0x63, 0x4d, 0x71, 0x52, 0x65, 0x71, 0x75, + 0x65, 0x73, 0x74, 0x1a, 0x16, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x41, 0x6c, 0x6c, 0x6f, + 0x63, 0x4d, 0x71, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x12, 0x3a, 0x0a, + 0x07, 0x41, 0x6c, 0x6c, 0x6f, 0x63, 0x4b, 0x76, 0x12, 0x15, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, + 0x2e, 0x41, 0x6c, 0x6c, 0x6f, 0x63, 0x4b, 0x76, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, + 0x16, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2e, 0x41, 0x6c, 0x6c, 0x6f, 0x63, 0x4b, 0x76, 0x52, + 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x42, 0x09, 0x5a, 0x07, 0x2e, 0x3b, 0x70, + 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, +} + +var ( + file_allocator_proto_rawDescOnce sync.Once + file_allocator_proto_rawDescData = file_allocator_proto_rawDesc +) + +func file_allocator_proto_rawDescGZIP() []byte { + file_allocator_proto_rawDescOnce.Do(func() { + file_allocator_proto_rawDescData = protoimpl.X.CompressGZIP(file_allocator_proto_rawDescData) + }) + return file_allocator_proto_rawDescData +} + +var file_allocator_proto_msgTypes = make([]protoimpl.MessageInfo, 4) +var file_allocator_proto_goTypes = []any{ + (*AllocMqRequest)(nil), // 0: proto.AllocMqRequest + (*AllocMqResponse)(nil), // 1: proto.AllocMqResponse + (*AllocKvRequest)(nil), // 2: proto.AllocKvRequest + (*AllocKvResponse)(nil), // 3: proto.AllocKvResponse +} +var file_allocator_proto_depIdxs = []int32{ + 0, // 0: proto.AllocatorService.AllocMessageQueue:input_type -> proto.AllocMqRequest + 2, // 1: proto.AllocatorService.AllocKv:input_type -> proto.AllocKvRequest + 1, // 2: proto.AllocatorService.AllocMessageQueue:output_type -> proto.AllocMqResponse + 3, // 3: proto.AllocatorService.AllocKv:output_type -> proto.AllocKvResponse + 2, // [2:4] is the sub-list for method output_type + 0, // [0:2] is the sub-list for method input_type + 0, // [0:0] is the sub-list for extension type_name + 0, // [0:0] is the sub-list for extension extendee + 0, // [0:0] is the sub-list for field type_name +} + +func init() { file_allocator_proto_init() } +func file_allocator_proto_init() { + if File_allocator_proto != nil { + return + } + type x struct{} + out := protoimpl.TypeBuilder{ + File: protoimpl.DescBuilder{ + GoPackagePath: reflect.TypeOf(x{}).PkgPath(), + RawDescriptor: file_allocator_proto_rawDesc, + NumEnums: 0, + NumMessages: 4, + NumExtensions: 0, + NumServices: 1, + }, + GoTypes: file_allocator_proto_goTypes, + DependencyIndexes: file_allocator_proto_depIdxs, + MessageInfos: file_allocator_proto_msgTypes, + }.Build() + File_allocator_proto = out.File + file_allocator_proto_rawDesc = nil + file_allocator_proto_goTypes = nil + file_allocator_proto_depIdxs = nil +} diff --git a/pkg/proto/allocator.proto b/pkg/proto/allocator.proto new file mode 100644 index 0000000..8f16504 --- /dev/null +++ b/pkg/proto/allocator.proto @@ -0,0 +1,26 @@ +syntax = "proto3"; + +option go_package = ".;proto"; + +package proto; + +service AllocatorService { + rpc AllocMessageQueue(AllocMqRequest) returns (AllocMqResponse) {} + rpc AllocKv(AllocKvRequest) returns (AllocKvResponse) {} +} + +message AllocMqRequest { +} + +message AllocMqResponse { + bool is_success = 1; + string addr = 2; +} + +message AllocKvRequest { +} + +message AllocKvResponse { + bool is_success = 1; + repeated string endpoints = 2; +} \ No newline at end of file diff --git a/pkg/proto/allocator_grpc.pb.go b/pkg/proto/allocator_grpc.pb.go new file mode 100644 index 0000000..5c73ab3 --- /dev/null +++ b/pkg/proto/allocator_grpc.pb.go @@ -0,0 +1,159 @@ +// Code generated by protoc-gen-go-grpc. DO NOT EDIT. +// versions: +// - protoc-gen-go-grpc v1.5.1 +// - protoc v5.28.2 +// source: allocator.proto + +package proto + +import ( + context "context" + grpc "google.golang.org/grpc" + codes "google.golang.org/grpc/codes" + status "google.golang.org/grpc/status" +) + +// This is a compile-time assertion to ensure that this generated file +// is compatible with the grpc package it is being compiled against. +// Requires gRPC-Go v1.64.0 or later. +const _ = grpc.SupportPackageIsVersion9 + +const ( + AllocatorService_AllocMessageQueue_FullMethodName = "/proto.AllocatorService/AllocMessageQueue" + AllocatorService_AllocKv_FullMethodName = "/proto.AllocatorService/AllocKv" +) + +// AllocatorServiceClient is the client API for AllocatorService service. +// +// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream. +type AllocatorServiceClient interface { + AllocMessageQueue(ctx context.Context, in *AllocMqRequest, opts ...grpc.CallOption) (*AllocMqResponse, error) + AllocKv(ctx context.Context, in *AllocKvRequest, opts ...grpc.CallOption) (*AllocKvResponse, error) +} + +type allocatorServiceClient struct { + cc grpc.ClientConnInterface +} + +func NewAllocatorServiceClient(cc grpc.ClientConnInterface) AllocatorServiceClient { + return &allocatorServiceClient{cc} +} + +func (c *allocatorServiceClient) AllocMessageQueue(ctx context.Context, in *AllocMqRequest, opts ...grpc.CallOption) (*AllocMqResponse, error) { + cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...) + out := new(AllocMqResponse) + err := c.cc.Invoke(ctx, AllocatorService_AllocMessageQueue_FullMethodName, in, out, cOpts...) + if err != nil { + return nil, err + } + return out, nil +} + +func (c *allocatorServiceClient) AllocKv(ctx context.Context, in *AllocKvRequest, opts ...grpc.CallOption) (*AllocKvResponse, error) { + cOpts := append([]grpc.CallOption{grpc.StaticMethod()}, opts...) + out := new(AllocKvResponse) + err := c.cc.Invoke(ctx, AllocatorService_AllocKv_FullMethodName, in, out, cOpts...) + if err != nil { + return nil, err + } + return out, nil +} + +// AllocatorServiceServer is the server API for AllocatorService service. +// All implementations must embed UnimplementedAllocatorServiceServer +// for forward compatibility. +type AllocatorServiceServer interface { + AllocMessageQueue(context.Context, *AllocMqRequest) (*AllocMqResponse, error) + AllocKv(context.Context, *AllocKvRequest) (*AllocKvResponse, error) + mustEmbedUnimplementedAllocatorServiceServer() +} + +// UnimplementedAllocatorServiceServer must be embedded to have +// forward compatible implementations. +// +// NOTE: this should be embedded by value instead of pointer to avoid a nil +// pointer dereference when methods are called. +type UnimplementedAllocatorServiceServer struct{} + +func (UnimplementedAllocatorServiceServer) AllocMessageQueue(context.Context, *AllocMqRequest) (*AllocMqResponse, error) { + return nil, status.Errorf(codes.Unimplemented, "method AllocMessageQueue not implemented") +} +func (UnimplementedAllocatorServiceServer) AllocKv(context.Context, *AllocKvRequest) (*AllocKvResponse, error) { + return nil, status.Errorf(codes.Unimplemented, "method AllocKv not implemented") +} +func (UnimplementedAllocatorServiceServer) mustEmbedUnimplementedAllocatorServiceServer() {} +func (UnimplementedAllocatorServiceServer) testEmbeddedByValue() {} + +// UnsafeAllocatorServiceServer may be embedded to opt out of forward compatibility for this service. +// Use of this interface is not recommended, as added methods to AllocatorServiceServer will +// result in compilation errors. +type UnsafeAllocatorServiceServer interface { + mustEmbedUnimplementedAllocatorServiceServer() +} + +func RegisterAllocatorServiceServer(s grpc.ServiceRegistrar, srv AllocatorServiceServer) { + // If the following call pancis, it indicates UnimplementedAllocatorServiceServer was + // embedded by pointer and is nil. This will cause panics if an + // unimplemented method is ever invoked, so we test this at initialization + // time to prevent it from happening at runtime later due to I/O. + if t, ok := srv.(interface{ testEmbeddedByValue() }); ok { + t.testEmbeddedByValue() + } + s.RegisterService(&AllocatorService_ServiceDesc, srv) +} + +func _AllocatorService_AllocMessageQueue_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(AllocMqRequest) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(AllocatorServiceServer).AllocMessageQueue(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: AllocatorService_AllocMessageQueue_FullMethodName, + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(AllocatorServiceServer).AllocMessageQueue(ctx, req.(*AllocMqRequest)) + } + return interceptor(ctx, in, info, handler) +} + +func _AllocatorService_AllocKv_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { + in := new(AllocKvRequest) + if err := dec(in); err != nil { + return nil, err + } + if interceptor == nil { + return srv.(AllocatorServiceServer).AllocKv(ctx, in) + } + info := &grpc.UnaryServerInfo{ + Server: srv, + FullMethod: AllocatorService_AllocKv_FullMethodName, + } + handler := func(ctx context.Context, req interface{}) (interface{}, error) { + return srv.(AllocatorServiceServer).AllocKv(ctx, req.(*AllocKvRequest)) + } + return interceptor(ctx, in, info, handler) +} + +// AllocatorService_ServiceDesc is the grpc.ServiceDesc for AllocatorService service. +// It's only intended for direct use with grpc.RegisterService, +// and not to be introspected or modified (even as a copy) +var AllocatorService_ServiceDesc = grpc.ServiceDesc{ + ServiceName: "proto.AllocatorService", + HandlerType: (*AllocatorServiceServer)(nil), + Methods: []grpc.MethodDesc{ + { + MethodName: "AllocMessageQueue", + Handler: _AllocatorService_AllocMessageQueue_Handler, + }, + { + MethodName: "AllocKv", + Handler: _AllocatorService_AllocKv_Handler, + }, + }, + Streams: []grpc.StreamDesc{}, + Metadata: "allocator.proto", +} diff --git a/settings.toml b/settings.toml index 52228e7..d1974f4 100644 --- a/settings.toml +++ b/settings.toml @@ -13,6 +13,9 @@ aliases = { id = "auth", uc = "files", co = "interactive", im = "messaging" } dsn = "host=localhost user=postgres dbname=postgres password=password port=5432 sslmode=disable" prefix = "sn_" +[mq] +addr = "localhost:4222" + [kv] endpoints = ["localhost:2379"]