Support broadcast deletion

This commit is contained in:
2024-09-19 22:25:53 +08:00
parent e147937cbb
commit 560c8a9143
4 changed files with 35 additions and 1 deletions

View File

@ -3,6 +3,7 @@ package grpc
import (
"net"
"git.solsynth.dev/hydrogen/dealer/pkg/proto"
"github.com/spf13/viper"
"google.golang.org/grpc"
health "google.golang.org/grpc/health/grpc_health_v1"
@ -10,6 +11,7 @@ import (
)
type Server struct {
proto.UnimplementedServiceDirectoryServer
}
var S *grpc.Server
@ -18,6 +20,7 @@ func NewGRPC() {
S = grpc.NewServer()
health.RegisterHealthServer(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/paperclip/pkg/internal/database"
"git.solsynth.dev/hydrogen/paperclip/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
}