🎉 Initial Commit

pkg/internal/grpc/health.go

@@ -0,0 +1,26 @@
+package grpc
+
+import (
+	"context"
+	health "google.golang.org/grpc/health/grpc_health_v1"
+	"time"
+)
+
+func (v *Server) Check(ctx context.Context, request *health.HealthCheckRequest) (*health.HealthCheckResponse, error) {
+	return &health.HealthCheckResponse{
+		Status: health.HealthCheckResponse_SERVING,
+	}, nil
+}
+
+func (v *Server) Watch(request *health.HealthCheckRequest, server health.Health_WatchServer) error {
+	for {
+		if server.Send(&health.HealthCheckResponse{
+			Status: health.HealthCheckResponse_SERVING,
+		}) != nil {
+			break
+		}
+		time.Sleep(1000 * time.Millisecond)
+	}
+
+	return nil
+}

pkg/internal/grpc/insight.go

@@ -0,0 +1,25 @@
+package grpc
+
+import (
+	"context"
+	"fmt"
+	"git.solsynth.dev/hypernet/insight/pkg/internal/services"
+	"git.solsynth.dev/hypernet/insight/pkg/proto"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+func (v *Server) GenerateInsight(ctx context.Context, request *proto.InsightRequest) (*proto.InsightResponse, error) {
+	input := request.GetSource()
+	if err := services.PlaceOrder(uint(request.GetUserId()), len(input)); err != nil {
+		return nil, status.Errorf(codes.ResourceExhausted, fmt.Sprintf("failed to place order: %v", err))
+	}
+
+	out, err := services.GenerateInsights(input)
+	if err != nil {
+		_ = services.MakeRefund(uint(request.GetUserId()), len(input))
+		return nil, status.Errorf(codes.Internal, fmt.Sprintf("failed to generate insight: %v", err))
+	}
+
+	return &proto.InsightResponse{Response: out}, nil
+}

pkg/internal/grpc/server.go

@@ -0,0 +1,43 @@
+package grpc
+
+import (
+	"net"
+
+	iproto "git.solsynth.dev/hypernet/insight/pkg/proto"
+	"git.solsynth.dev/hypernet/nexus/pkg/proto"
+	"github.com/spf13/viper"
+	"google.golang.org/grpc"
+	health "google.golang.org/grpc/health/grpc_health_v1"
+	"google.golang.org/grpc/reflection"
+)
+
+type Server struct {
+	proto.UnimplementedDirectoryServiceServer
+	iproto.UnimplementedInsightServiceServer
+	health.UnimplementedHealthServer
+
+	srv *grpc.Server
+}
+
+func NewGrpc() *Server {
+	server := &Server{
+		srv: grpc.NewServer(),
+	}
+
+	proto.RegisterDirectoryServiceServer(server.srv, server)
+	iproto.RegisterInsightServiceServer(server.srv, server)
+	health.RegisterHealthServer(server.srv, server)
+
+	reflection.Register(server.srv)
+
+	return server
+}
+
+func (v *Server) Listen() error {
+	listener, err := net.Listen("tcp", viper.GetString("grpc_bind"))
+	if err != nil {
+		return err
+	}
+
+	return v.srv.Serve(listener)
+}

pkg/internal/grpc/services.go

@@ -0,0 +1,42 @@
+package grpc
+
+import (
+	"context"
+	"git.solsynth.dev/hypernet/nexus/pkg/nex"
+	"strconv"
+
+	"git.solsynth.dev/hypernet/nexus/pkg/proto"
+	"git.solsynth.dev/hypernet/insight/pkg/internal/database"
+)
+
+func (v *Server) BroadcastEvent(ctx context.Context, in *proto.EventInfo) (*proto.EventResponse, error) {
+	switch in.GetEvent() {
+	case "deletion":
+		data := nex.DecodeMap(in.GetData())
+		resType, ok := data["type"].(string)
+		if !ok {
+			break
+		}
+		switch resType {
+		case "account":
+			id, ok := data["id"].(string)
+			if !ok {
+				break
+			}
+			numericId, err := strconv.Atoi(id)
+			if err != nil {
+				break
+			}
+			tx := database.C.Begin()
+			for _, model := range database.AutoMaintainRange {
+				switch model.(type) {
+				default:
+					tx.Delete(model, "account_id = ?", numericId)
+				}
+			}
+			tx.Commit()
+		}
+	}
+
+	return &proto.EventResponse{}, nil
+}