Support broadcast deletion

This commit is contained in:
2024-09-19 22:23:32 +08:00
parent ee16094855
commit 184fc3ecba
6 changed files with 37 additions and 4 deletions

View File

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

View File

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