✨ Basic queue for processing
This commit is contained in:
@ -1,76 +0,0 @@
|
||||
package grpc
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/database"
|
||||
"git.solsynth.dev/hydrogen/paperclip/pkg/proto"
|
||||
"google.golang.org/protobuf/types/known/emptypb"
|
||||
|
||||
"git.solsynth.dev/hydrogen/paperclip/pkg/internal/models"
|
||||
jsoniter "github.com/json-iterator/go"
|
||||
"github.com/samber/lo"
|
||||
)
|
||||
|
||||
func (v *Server) GetAttachment(ctx context.Context, request *proto.AttachmentLookupRequest) (*proto.Attachment, error) {
|
||||
var attachment models.Attachment
|
||||
|
||||
tx := database.C.Model(&models.Attachment{})
|
||||
if request.Id != nil {
|
||||
tx = tx.Where("id = ?", request.GetId())
|
||||
}
|
||||
if request.Uuid != nil {
|
||||
tx = tx.Where("uuid = ?", request.GetUuid())
|
||||
}
|
||||
if request.Usage != nil {
|
||||
tx = tx.Where("usage = ?", request.GetUsage())
|
||||
}
|
||||
|
||||
if err := tx.First(&attachment).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
rawMetadata, _ := jsoniter.Marshal(attachment.Metadata)
|
||||
|
||||
if attachment.AccountID == nil {
|
||||
attachment.AccountID = lo.ToPtr[uint](0)
|
||||
}
|
||||
|
||||
return &proto.Attachment{
|
||||
Id: uint64(attachment.ID),
|
||||
Uuid: attachment.Uuid,
|
||||
Size: attachment.Size,
|
||||
Name: attachment.Name,
|
||||
Alt: attachment.Alternative,
|
||||
Usage: attachment.Usage,
|
||||
Mimetype: attachment.MimeType,
|
||||
Hash: attachment.HashCode,
|
||||
Destination: attachment.Destination,
|
||||
Metadata: rawMetadata,
|
||||
IsMature: attachment.IsMature,
|
||||
AccountId: uint64(*attachment.AccountID),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (v *Server) CheckAttachmentExists(ctx context.Context, request *proto.AttachmentLookupRequest) (*emptypb.Empty, error) {
|
||||
tx := database.C.Model(&models.Attachment{})
|
||||
if request.Id != nil {
|
||||
tx = tx.Where("id = ?", request.GetId())
|
||||
}
|
||||
if request.Uuid != nil {
|
||||
tx = tx.Where("uuid = ?", request.GetUuid())
|
||||
}
|
||||
if request.Usage != nil {
|
||||
tx = tx.Where("usage = ?", request.GetUsage())
|
||||
}
|
||||
|
||||
var count int64
|
||||
if err := tx.Model(&models.Attachment{}).Count(&count).Error; err != nil {
|
||||
return nil, err
|
||||
} else if count == 0 {
|
||||
return nil, fmt.Errorf("record not found")
|
||||
}
|
||||
|
||||
return &emptypb.Empty{}, nil
|
||||
}
|
@ -1,16 +1,15 @@
|
||||
package grpc
|
||||
|
||||
import (
|
||||
"git.solsynth.dev/hydrogen/paperclip/pkg/proto"
|
||||
"net"
|
||||
|
||||
"github.com/spf13/viper"
|
||||
"google.golang.org/grpc"
|
||||
health "google.golang.org/grpc/health/grpc_health_v1"
|
||||
"google.golang.org/grpc/reflection"
|
||||
"net"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
proto.UnimplementedAttachmentsServer
|
||||
}
|
||||
|
||||
var S *grpc.Server
|
||||
@ -18,7 +17,6 @@ var S *grpc.Server
|
||||
func NewGRPC() {
|
||||
S = grpc.NewServer()
|
||||
|
||||
proto.RegisterAttachmentsServer(S, &Server{})
|
||||
health.RegisterHealthServer(S, &Server{})
|
||||
|
||||
reflection.Register(S)
|
||||
|
Reference in New Issue
Block a user