revise upload
This commit is contained in:
vanilia
No files matched your search
@@ -0,0 +1,90 @@
|
||||
package upload
|
||||
|
||||
import (
|
||||
"errors"
|
||||
pl "simrs-vx/pkg/logger"
|
||||
pu "simrs-vx/pkg/use-case-helper"
|
||||
|
||||
dg "github.com/karincake/apem/db-gorm-pg"
|
||||
d "github.com/karincake/dodol"
|
||||
"gorm.io/gorm"
|
||||
|
||||
eru "simrs-vx/internal/domain/references/upload"
|
||||
|
||||
e "simrs-vx/internal/domain/main-entities/upload"
|
||||
)
|
||||
|
||||
const source = "upload"
|
||||
|
||||
func Upload(input e.CreateDto) (*d.Data, error) {
|
||||
event := pl.Event{
|
||||
Feature: "Upload",
|
||||
Source: source,
|
||||
}
|
||||
|
||||
// Start log
|
||||
pl.SetLogInfo(&event, input, "started", "create")
|
||||
|
||||
var data interface{}
|
||||
err := dg.I.Transaction(func(tx *gorm.DB) error {
|
||||
mwRunner := newMiddlewareRunner(&event, tx)
|
||||
mwRunner.setMwType(pu.MWTPre)
|
||||
|
||||
// Run pre-middleware
|
||||
err := mwRunner.RunCreateMiddleware(createPreMw, &input, &data)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// validate entityType_Code and Type_Code
|
||||
valid, msg := eru.IsValidUploadCode(input.EntityType_Code, input.Type_Code)
|
||||
if !valid {
|
||||
event.Status = "failed"
|
||||
event.ErrInfo = pl.ErrorInfo{
|
||||
Code: "invalid-code",
|
||||
Detail: msg,
|
||||
Raw: errors.New(msg),
|
||||
}
|
||||
return pl.SetLogError(&event, input)
|
||||
}
|
||||
|
||||
// upload file
|
||||
input.FilePath, err = uploadAndGenerateFileUrl(input, &event)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if input.EntityType_Code == eru.ETCEncounter {
|
||||
data, err = setEncounterDocument(input, &event, tx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
} else if input.EntityType_Code == eru.ETCPerson {
|
||||
data, err = setPersonAttachment(input, &event, tx)
|
||||
}
|
||||
|
||||
mwRunner.setMwType(pu.MWTPost)
|
||||
// Run post-middleware
|
||||
if err := mwRunner.RunCreateMiddleware(createPostMw, &input, &data); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
pl.SetLogInfo(&event, nil, "complete")
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &d.Data{
|
||||
Meta: d.II{
|
||||
"source": source,
|
||||
"structure": "single-data",
|
||||
"status": "created",
|
||||
},
|
||||
Data: input.ToResponse(),
|
||||
}, nil
|
||||
}
|
||||
@@ -0,0 +1,184 @@
|
||||
/*
|
||||
DESCRIPTION:
|
||||
Any functions that are used internally by the use-case
|
||||
*/
|
||||
package upload
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"path/filepath"
|
||||
ecore "simrs-vx/internal/domain/base-entities/core"
|
||||
eru "simrs-vx/internal/domain/references/upload"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
pl "simrs-vx/pkg/logger"
|
||||
pmh "simrs-vx/pkg/minio-helper"
|
||||
puh "simrs-vx/pkg/upload-helper"
|
||||
|
||||
dg "github.com/karincake/apem/db-gorm-pg"
|
||||
"gorm.io/gorm"
|
||||
|
||||
ee "simrs-vx/internal/domain/main-entities/encounter-document"
|
||||
ep "simrs-vx/internal/domain/main-entities/person"
|
||||
e "simrs-vx/internal/domain/main-entities/upload"
|
||||
|
||||
ue "simrs-vx/internal/use-case/main-use-case/encounter-document"
|
||||
up "simrs-vx/internal/use-case/main-use-case/person"
|
||||
)
|
||||
|
||||
func uploadAndGenerateFileUrl(input e.CreateDto, event *pl.Event) (string, error) {
|
||||
pl.SetLogInfo(event, input, "started", "uploadAndGenerateFileUrl")
|
||||
bucket := string(input.EntityType_Code)
|
||||
ext := strings.ToLower(filepath.Ext(input.Filename))
|
||||
|
||||
if !puh.IsValidFileType(ext, bucket) {
|
||||
return "", fmt.Errorf("invalid file type: %s", input.Filename)
|
||||
}
|
||||
objectName := fmt.Sprintf("%v/%s-%d%s", *input.Ref_Id, input.Type_Code, time.Now().UnixNano(), ext)
|
||||
|
||||
uploadInput := pmh.UploadReaderInput{
|
||||
BucketName: bucket,
|
||||
Name: objectName,
|
||||
File: input.File,
|
||||
Size: input.Size,
|
||||
ContentType: input.MimeType,
|
||||
}
|
||||
|
||||
_, err := pmh.I.PutObject(uploadInput)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// Build URL for access
|
||||
publicUrl := pmh.I.GenerateUrl(bucket, objectName)
|
||||
|
||||
pl.SetLogInfo(event, nil, "complete")
|
||||
return publicUrl, nil
|
||||
}
|
||||
|
||||
func removeUploadedFile(bucket, fileUrl string, refId uint, event *pl.Event) error {
|
||||
pl.SetLogInfo(event, nil, "started", "removeUploadedFile")
|
||||
filename, err := pmh.GetFilename(fileUrl)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
filePath := fmt.Sprintf("%v/%s", refId, filename)
|
||||
err = pmh.I.RemoveObject(bucket, filePath)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
pl.SetLogInfo(event, nil, "complete")
|
||||
return nil
|
||||
}
|
||||
|
||||
func updatePersonAttachment(dataPerson *ep.Person, event *pl.Event, dbx ...*gorm.DB) error {
|
||||
pl.SetLogInfo(event, dataPerson, "started", "DBUpdate")
|
||||
|
||||
var tx *gorm.DB
|
||||
if len(dbx) > 0 {
|
||||
tx = dbx[0]
|
||||
} else {
|
||||
tx = dg.I
|
||||
}
|
||||
|
||||
if err := tx.Save(&dataPerson).Error; err != nil {
|
||||
event.Status = "failed"
|
||||
event.ErrInfo = pl.ErrorInfo{
|
||||
Code: "data-update-fail",
|
||||
Detail: "Database update failed",
|
||||
Raw: err,
|
||||
}
|
||||
return pl.SetLogError(event, dataPerson)
|
||||
}
|
||||
|
||||
pl.SetLogInfo(event, nil, "complete")
|
||||
return nil
|
||||
}
|
||||
|
||||
func setPersonAttachment(input e.CreateDto, event *pl.Event, tx *gorm.DB) (*ep.Person, error) {
|
||||
// get person
|
||||
dataPerson, err := up.ReadDetailData(ep.ReadDetailDto{Id: *input.Ref_Id}, event, tx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var removeUrl string
|
||||
switch input.Type_Code {
|
||||
case eru.UCPRN:
|
||||
removeUrl = *dataPerson.ResidentIdentityFileUrl
|
||||
dataPerson.ResidentIdentityFileUrl = &input.FilePath
|
||||
case eru.UCPDL:
|
||||
removeUrl = *dataPerson.DrivingLicenseFileUrl
|
||||
dataPerson.DrivingLicenseFileUrl = &input.FilePath
|
||||
case eru.UCPP:
|
||||
removeUrl = *dataPerson.PassportFileUrl
|
||||
dataPerson.PassportFileUrl = &input.FilePath
|
||||
case eru.UCPFC:
|
||||
removeUrl = *dataPerson.FamilyIdentityFileUrl
|
||||
dataPerson.FamilyIdentityFileUrl = &input.FilePath
|
||||
}
|
||||
|
||||
if removeUrl != "" {
|
||||
if err := removeUploadedFile(string(input.EntityType_Code), removeUrl, *input.Ref_Id, event); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
if err = updatePersonAttachment(dataPerson, event, tx); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return dataPerson, nil
|
||||
}
|
||||
|
||||
func setEncounterDocument(input e.CreateDto, event *pl.Event, tx *gorm.DB) (*ee.EncounterDocument, error) {
|
||||
data := ee.EncounterDocument{}
|
||||
|
||||
// get EncounterDocument
|
||||
dataUpload, _, err := ue.ReadListData(ee.ReadListDto{
|
||||
FilterDto: ee.FilterDto{
|
||||
Encounter_Id: input.Ref_Id,
|
||||
Type_Code: input.Type_Code},
|
||||
Pagination: ecore.Pagination{PageSize: 1}}, event, tx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// set data
|
||||
createDto := ee.CreateDto{
|
||||
Encounter_Id: input.Ref_Id,
|
||||
Type_Code: input.Type_Code,
|
||||
Name: input.Name,
|
||||
FilePath: input.FilePath,
|
||||
Filename: input.Filename,
|
||||
Upload_Employee_Id: input.Upload_Employee_Id,
|
||||
}
|
||||
|
||||
if len(dataUpload) > 0 {
|
||||
if err := removeUploadedFile(string(input.EntityType_Code), *dataUpload[0].FilePath, *dataUpload[0].Encounter_Id, event); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// update data
|
||||
err = ue.UpdateData(ee.UpdateDto{
|
||||
Id: uint16(dataUpload[0].Id),
|
||||
CreateDto: createDto}, &dataUpload[0], event, tx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
data = dataUpload[0]
|
||||
data.FilePath = &input.FilePath
|
||||
} else {
|
||||
// insert data
|
||||
resData, err := ue.CreateData(createDto, event, tx)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
data = *resData
|
||||
}
|
||||
|
||||
return &data, nil
|
||||
}
|
||||
@@ -0,0 +1,104 @@
|
||||
package upload
|
||||
|
||||
import (
|
||||
pl "simrs-vx/pkg/logger"
|
||||
pu "simrs-vx/pkg/use-case-helper"
|
||||
|
||||
"gorm.io/gorm"
|
||||
|
||||
e "simrs-vx/internal/domain/main-entities/upload"
|
||||
)
|
||||
|
||||
type middlewareRunner struct {
|
||||
Event *pl.Event
|
||||
Tx *gorm.DB
|
||||
MwType pu.MWType
|
||||
}
|
||||
|
||||
// NewMiddlewareExecutor creates a new middleware executor
|
||||
func newMiddlewareRunner(event *pl.Event, tx *gorm.DB) *middlewareRunner {
|
||||
return &middlewareRunner{
|
||||
Event: event,
|
||||
Tx: tx,
|
||||
}
|
||||
}
|
||||
|
||||
// ExecuteCreateMiddleware executes create middleware
|
||||
func (me *middlewareRunner) RunCreateMiddleware(middlewares []createMw, input *e.CreateDto, data interface{}) error {
|
||||
for _, middleware := range middlewares {
|
||||
logData := pu.GetLogData(input, data)
|
||||
|
||||
pl.SetLogInfo(me.Event, logData, "started", middleware.Name)
|
||||
|
||||
if err := middleware.Func(input, data, me.Tx); err != nil {
|
||||
return pu.HandleMiddlewareError(me.Event, string(me.MwType), middleware.Name, logData, err)
|
||||
}
|
||||
|
||||
pl.SetLogInfo(me.Event, nil, "complete")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (me *middlewareRunner) RunReadListMiddleware(middlewares []readListMw, input *e.ReadListDto, data interface{}) error {
|
||||
for _, middleware := range middlewares {
|
||||
logData := pu.GetLogData(input, data)
|
||||
|
||||
pl.SetLogInfo(me.Event, logData, "started", middleware.Name)
|
||||
|
||||
if err := middleware.Func(input, data, me.Tx); err != nil {
|
||||
return pu.HandleMiddlewareError(me.Event, string(me.MwType), middleware.Name, logData, err)
|
||||
}
|
||||
|
||||
pl.SetLogInfo(me.Event, nil, "complete")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (me *middlewareRunner) RunReadDetailMiddleware(middlewares []readDetailMw, input *e.ReadDetailDto, data interface{}) error {
|
||||
for _, middleware := range middlewares {
|
||||
logData := pu.GetLogData(input, data)
|
||||
|
||||
pl.SetLogInfo(me.Event, logData, "started", middleware.Name)
|
||||
|
||||
if err := middleware.Func(input, data, me.Tx); err != nil {
|
||||
return pu.HandleMiddlewareError(me.Event, string(me.MwType), middleware.Name, logData, err)
|
||||
}
|
||||
|
||||
pl.SetLogInfo(me.Event, nil, "complete")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (me *middlewareRunner) RunUpdateMiddleware(middlewares []readDetailMw, input *e.ReadDetailDto, data interface{}) error {
|
||||
for _, middleware := range middlewares {
|
||||
logData := pu.GetLogData(input, data)
|
||||
|
||||
pl.SetLogInfo(me.Event, logData, "started", middleware.Name)
|
||||
|
||||
if err := middleware.Func(input, data, me.Tx); err != nil {
|
||||
return pu.HandleMiddlewareError(me.Event, string(me.MwType), middleware.Name, logData, err)
|
||||
}
|
||||
|
||||
pl.SetLogInfo(me.Event, nil, "complete")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (me *middlewareRunner) RunDeleteMiddleware(middlewares []readDetailMw, input *e.ReadDetailDto, data interface{}) error {
|
||||
for _, middleware := range middlewares {
|
||||
logData := pu.GetLogData(input, data)
|
||||
|
||||
pl.SetLogInfo(me.Event, logData, "started", middleware.Name)
|
||||
|
||||
if err := middleware.Func(input, data, me.Tx); err != nil {
|
||||
return pu.HandleMiddlewareError(me.Event, string(me.MwType), middleware.Name, logData, err)
|
||||
}
|
||||
|
||||
pl.SetLogInfo(me.Event, nil, "complete")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (me *middlewareRunner) setMwType(mwType pu.MWType) {
|
||||
me.MwType = mwType
|
||||
}
|
||||
@@ -0,0 +1,9 @@
|
||||
package upload
|
||||
|
||||
// example of middleware
|
||||
// func init() {
|
||||
// createPreMw = append(createPreMw,
|
||||
// CreateMw{Name: "modif-input", Func: pm.ModifInput},
|
||||
// CreateMw{Name: "check-data", Func: pm.CheckData},
|
||||
// )
|
||||
// }
|
||||
@@ -0,0 +1,44 @@
|
||||
/*
|
||||
DESCRIPTION:
|
||||
A sample, part of the package that contains type, constants, and/or variables.
|
||||
|
||||
In this sample it also provides type and variable regarding the needs of the
|
||||
middleware to separate from main use-case which has the basic CRUD
|
||||
functionality. The purpose of this is to make the code more maintainable.
|
||||
*/
|
||||
package upload
|
||||
|
||||
import (
|
||||
"gorm.io/gorm"
|
||||
|
||||
e "simrs-vx/internal/domain/main-entities/upload"
|
||||
)
|
||||
|
||||
type createMw struct {
|
||||
Name string
|
||||
Func func(input *e.CreateDto, data interface{}, tx *gorm.DB) error
|
||||
}
|
||||
|
||||
type readListMw struct {
|
||||
Name string
|
||||
Func func(input *e.ReadListDto, data interface{}, tx *gorm.DB) error
|
||||
}
|
||||
|
||||
type readDetailMw struct {
|
||||
Name string
|
||||
Func func(input *e.ReadDetailDto, data interface{}, tx *gorm.DB) error
|
||||
}
|
||||
|
||||
type UpdateMw = readDetailMw
|
||||
type DeleteMw = readDetailMw
|
||||
|
||||
var createPreMw []createMw // preprocess middleware
|
||||
var createPostMw []createMw // postprocess middleware
|
||||
var readListPreMw []readListMw // ..
|
||||
var readListPostMw []readListMw // ..
|
||||
var readDetailPreMw []readDetailMw
|
||||
var readDetailPostMw []readDetailMw
|
||||
var updatePreMw []readDetailMw
|
||||
var updatePostMw []readDetailMw
|
||||
var deletePreMw []readDetailMw
|
||||
var deletePostMw []readDetailMw
|
||||
Reference in New Issue
Block a user