50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
package platform
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/Gleipnir-Technology/nidus-sync/db"
|
|
"github.com/Gleipnir-Technology/nidus-sync/db/enums"
|
|
"github.com/Gleipnir-Technology/nidus-sync/db/models"
|
|
"github.com/Gleipnir-Technology/nidus-sync/userfile"
|
|
"github.com/aarondl/opt/omit"
|
|
"github.com/aarondl/opt/omitnull"
|
|
)
|
|
|
|
type PoolUpload struct {
|
|
ID int32
|
|
}
|
|
|
|
func NewPoolUpload(ctx context.Context, u *models.User, upload userfile.FileUpload) (PoolUpload, error) {
|
|
txn, err := db.PGInstance.BobDB.BeginTx(ctx, nil)
|
|
if err != nil {
|
|
return PoolUpload{}, fmt.Errorf("Failed to begin transaction: %w", err)
|
|
}
|
|
|
|
file, err := models.FileuploadFiles.Insert(&models.FileuploadFileSetter{
|
|
ContentType: omit.From(upload.ContentType),
|
|
Created: omit.From(time.Now()),
|
|
CreatorID: omit.From(u.ID),
|
|
Deleted: omitnull.FromPtr[time.Time](nil),
|
|
Name: omit.From(upload.Name),
|
|
Status: omit.From(enums.FileuploadFilestatustypeUploaded),
|
|
SizeBytes: omit.From(int32(upload.SizeBytes)),
|
|
FileUUID: omit.From(upload.UUID),
|
|
}).One(ctx, txn)
|
|
if err != nil {
|
|
return PoolUpload{}, fmt.Errorf("Failed to create file upload: %w", err)
|
|
}
|
|
_, err = models.FileuploadCSVS.Insert(&models.FileuploadCSVSetter{
|
|
FileID: omit.From(file.ID),
|
|
Type: omit.From(enums.FileuploadCsvtypePoollist),
|
|
}).One(ctx, txn)
|
|
if err != nil {
|
|
return PoolUpload{}, fmt.Errorf("Failed to create csv: %w", err)
|
|
}
|
|
txn.Commit(ctx)
|
|
return PoolUpload{
|
|
ID: file.ID,
|
|
}, nil
|
|
}
|