✨ Support broadcast deletion

pkg/internal/database/migrator.go

@@ -5,7 +5,7 @@ import (
 	"gorm.io/gorm"
 )
 
-var DatabaseAutoActionRange = []any{
+var AutoMaintainRange = []any{
 	&models.Account{},
 	&models.Realm{},
 	&models.Channel{},
@@ -15,7 +15,7 @@ var DatabaseAutoActionRange = []any{
 }
 
 func RunMigration(source *gorm.DB) error {
-	if err := source.AutoMigrate(DatabaseAutoActionRange...); err != nil {
+	if err := source.AutoMigrate(AutoMaintainRange...); err != nil {
 		return err
 	}
 

pkg/internal/grpc/server.go

@@ -12,6 +12,7 @@ import (
 
 type Server struct {
 	proto.UnimplementedStreamControllerServer
+	proto.UnimplementedServiceDirectoryServer
 }
 
 var S *grpc.Server
@@ -21,6 +22,7 @@ func NewGRPC() {
 
 	health.RegisterHealthServer(S, &Server{})
 	proto.RegisterStreamControllerServer(S, &Server{})
+	proto.RegisterServiceDirectoryServer(S, &Server{})
 
 	reflection.Register(S)
 }

pkg/internal/grpc/services.go

@@ -0,0 +1,29 @@
+package grpc
+
+import (
+	"context"
+	"strconv"
+
+	"git.solsynth.dev/hydrogen/dealer/pkg/proto"
+	"git.solsynth.dev/hydrogen/messaging/pkg/internal/database"
+	"git.solsynth.dev/hydrogen/messaging/pkg/internal/models"
+)
+
+func (v *Server) BroadcastDeletion(ctx context.Context, request *proto.DeletionRequest) (*proto.DeletionResponse, error) {
+	switch request.GetResourceType() {
+	case "account":
+		numericId, err := strconv.Atoi(request.GetResourceId())
+		if err != nil {
+			break
+		}
+		for _, model := range database.AutoMaintainRange {
+			switch model.(type) {
+			default:
+				database.C.Delete(model, "account_id = ?", numericId)
+			}
+		}
+		database.C.Delete(&models.Account{}, "id = ?", numericId)
+	}
+
+	return &proto.DeletionResponse{}, nil
+}

pkg/internal/services/cleaner.go

@@ -13,7 +13,7 @@ func DoAutoDatabaseCleanup() {
 
 	// Deal soft-deletion
 	var count int64
-	for _, model := range database.DatabaseAutoActionRange {
+	for _, model := range database.AutoMaintainRange {
 		tx := database.C.Unscoped().Delete(model, "deleted_at >= ?", deadline)
 		if tx.Error != nil {
 			log.Error().Err(tx.Error).Msg("An error occurred when running database cleanup...")