MQ, KV and MQ allocator

This commit is contained in:
LittleSheep 2024-10-24 23:46:09 +08:00
parent aa67bd0b8f
commit 23d14e4e46
11 changed files with 578 additions and 3 deletions

3
go.mod
View File

@ -69,6 +69,9 @@ require (
github.com/mitchellh/mapstructure v1.5.0 // indirect github.com/mitchellh/mapstructure v1.5.0 // indirect
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
github.com/modern-go/reflect2 v1.0.2 // 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/pelletier/go-toml/v2 v2.2.3 // indirect
github.com/philhofer/fwd v1.1.3-0.20240916144458-20a13a1f6b7c // indirect github.com/philhofer/fwd v1.1.3-0.20240916144458-20a13a1f6b7c // indirect
github.com/pkg/errors v0.9.1 // indirect github.com/pkg/errors v0.9.1 // indirect

6
go.sum
View File

@ -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 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=
github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= 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/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.0.0-20180627143212-57f6aae5913c/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc=
github.com/pascaldekloe/goe v0.1.0 h1:cBOtyMzM9HTpWjXfbbunk26uA6nG3a8n06Wieeh0MwY= github.com/pascaldekloe/goe v0.1.0 h1:cBOtyMzM9HTpWjXfbbunk26uA6nG3a8n06Wieeh0MwY=
github.com/pascaldekloe/goe v0.1.0/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc= github.com/pascaldekloe/goe v0.1.0/go.mod h1:lzWF7FIEvWOWxwDKqyGYQf6ZUaNfKdP144TG7ZOy1lc=

View File

@ -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
}

View File

@ -1,7 +1,7 @@
package grpc package grpc
import ( import (
directory2 "git.solsynth.dev/hypernet/nexus/pkg/internal/directory" "git.solsynth.dev/hypernet/nexus/pkg/internal/directory"
"net" "net"
"git.solsynth.dev/hypernet/nexus/pkg/proto" "git.solsynth.dev/hypernet/nexus/pkg/proto"
@ -17,6 +17,7 @@ import (
type Server struct { type Server struct {
proto.UnimplementedDatabaseServiceServer proto.UnimplementedDatabaseServiceServer
proto.UnimplementedStreamServiceServer proto.UnimplementedStreamServiceServer
proto.UnimplementedAllocatorServiceServer
health.UnimplementedHealthServer health.UnimplementedHealthServer
srv *grpc.Server srv *grpc.Server
@ -27,10 +28,11 @@ func NewServer() *Server {
srv: grpc.NewServer(), srv: grpc.NewServer(),
} }
proto.RegisterDirectoryServiceServer(server.srv, &directory2.ServiceRpcServer{}) proto.RegisterDirectoryServiceServer(server.srv, &directory.ServiceRpcServer{})
proto.RegisterCommandProviderServer(server.srv, &directory2.CommandRpcServer{}) proto.RegisterCommandProviderServer(server.srv, &directory.CommandRpcServer{})
proto.RegisterDatabaseServiceServer(server.srv, server) proto.RegisterDatabaseServiceServer(server.srv, server)
proto.RegisterStreamServiceServer(server.srv, server) proto.RegisterStreamServiceServer(server.srv, server)
proto.RegisterAllocatorServiceServer(server.srv, server)
health.RegisterHealthServer(server.srv, server) health.RegisterHealthServer(server.srv, server)
reflection.Register(server.srv) reflection.Register(server.srv)

16
pkg/internal/mq/nats.go Normal file
View File

@ -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
}

View File

@ -7,6 +7,7 @@ import (
"git.solsynth.dev/hypernet/nexus/pkg/internal/directory" "git.solsynth.dev/hypernet/nexus/pkg/internal/directory"
"git.solsynth.dev/hypernet/nexus/pkg/internal/http" "git.solsynth.dev/hypernet/nexus/pkg/internal/http"
"git.solsynth.dev/hypernet/nexus/pkg/internal/kv" "git.solsynth.dev/hypernet/nexus/pkg/internal/kv"
"git.solsynth.dev/hypernet/nexus/pkg/internal/mq"
"git.solsynth.dev/hypernet/nexus/pkg/nex/sec" "git.solsynth.dev/hypernet/nexus/pkg/nex/sec"
"github.com/fatih/color" "github.com/fatih/color"
"os" "os"
@ -58,6 +59,14 @@ func main() {
log.Info().Msg("Connected to kv (etcd)!") 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 // Connect to database
log.Info().Msg("Connecting to database...") log.Info().Msg("Connecting to database...")
if db, err := database.Connect(viper.GetString("database.dsn")); err != nil { if db, err := database.Connect(viper.GetString("database.dsn")); err != nil {

34
pkg/nex/allocator.go Normal file
View File

@ -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
}
}

284
pkg/proto/allocator.pb.go Normal file
View File

@ -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
}

26
pkg/proto/allocator.proto Normal file
View File

@ -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;
}

View File

@ -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",
}

View File

@ -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" dsn = "host=localhost user=postgres dbname=postgres password=password port=5432 sslmode=disable"
prefix = "sn_" prefix = "sn_"
[mq]
addr = "localhost:4222"
[kv] [kv]
endpoints = ["localhost:2379"] endpoints = ["localhost:2379"]