nquest/pkg/service/file.go

package service

import (
	"context"
	"mime/multipart"

	"github.com/google/uuid"
	"gitrepo.ru/neonxp/nquest/pkg/models"
	"gorm.io/gorm"
)

type File struct {
	DB *gorm.DB
}

func NewFile(db *gorm.DB) *File {
	return &File{
		DB: db,
	}
}

func (u *File) Upload(
	ctx context.Context,
	filename string,
	contentType string,
	size int,
	r multipart.File,
) (uuid.UUID, error) {
	buf := make([]byte, size)
	if _, err := r.Read(buf); err != nil {
		return uuid.UUID{}, err
	}

	file := &models.File{
		ID:          uuid.New(),
		Filename:    filename,
		ContentType: contentType,
		Size:        size,
		Body:        buf,
	}

	return file.ID, u.DB.WithContext(ctx).Create(file).Error
}

func (u *File) GetFile(ctx context.Context, uid uuid.UUID) (*models.File, error) {
	f := new(models.File)

	return f, u.DB.WithContext(ctx).First(f, uid).Error
}