From 37c47f983957c1defccbcf1b34b8c11bed7987c8 Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Tue, 20 Aug 2024 19:17:43 +0800
Subject: [PATCH] :sparkles: Make it more implementable

---
 .idea/workspace.xml                         | 19 ++++++-------------
 pkg/internal/models/attachments.go          |  1 -
 pkg/internal/server/api/up_multipart_api.go |  7 ++++---
 pkg/internal/services/analyzer.go           | 14 ++++++++------
 pkg/internal/services/merger.go             |  2 ++
 settings.toml                               |  2 +-
 6 files changed, 21 insertions(+), 24 deletions(-)

diff --git a/.idea/workspace.xml b/.idea/workspace.xml
index 1a3ab33..41a8899 100644
--- a/.idea/workspace.xml
+++ b/.idea/workspace.xml
@@ -4,19 +4,12 @@
     <option name="autoReloadType" value="ALL" />
   </component>
   <component name="ChangeListManager">
-    <list default="true" id="18dd0d68-b4b8-40db-9734-9119b5c848bd" name="更改" comment=":sparkles: Un-public indexable &amp; select by pools">
-      <change afterPath="$PROJECT_DIR$/pkg/internal/server/api/up_direct_api.go" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/pkg/internal/server/api/up_multipart_api.go" afterDir="false" />
-      <change afterPath="$PROJECT_DIR$/pkg/internal/services/merger.go" afterDir="false" />
+    <list default="true" id="18dd0d68-b4b8-40db-9734-9119b5c848bd" name="更改" comment=":sparkles: Multipart file upload">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/pkg/internal/models/attachments.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/models/attachments.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/internal/server/api/attachments_api.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/server/api/attachments_api.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/internal/server/api/index.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/server/api/index.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/internal/services/attachments.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/attachments.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/internal/services/random_id.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/random_id.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/internal/services/recycler.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/recycler.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/internal/services/uploader.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/uploader.go" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/pkg/main.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/main.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/server/api/up_multipart_api.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/server/api/up_multipart_api.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/services/analyzer.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/analyzer.go" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/pkg/internal/services/merger.go" beforeDir="false" afterPath="$PROJECT_DIR$/pkg/internal/services/merger.go" afterDir="false" />
       <change beforePath="$PROJECT_DIR$/settings.toml" beforeDir="false" afterPath="$PROJECT_DIR$/settings.toml" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
@@ -128,7 +121,6 @@
     </option>
   </component>
   <component name="VcsManagerConfiguration">
-    <MESSAGE value=":truck: Update url mapping" />
     <MESSAGE value=":bug: Fix uuid duplicate when link exists" />
     <MESSAGE value="&#10;:sparkles: Add health check" />
     <MESSAGE value=":arrow_up: Upgrade Passport and use Hyper SDK" />
@@ -153,7 +145,8 @@
     <MESSAGE value=":boom: Use attachment rid instead of primary key when create" />
     <MESSAGE value=":bug: Fix crash on maintain cache" />
     <MESSAGE value=":sparkles: Un-public indexable &amp; select by pools" />
-    <option name="LAST_COMMIT_MESSAGE" value=":sparkles: Un-public indexable &amp; select by pools" />
+    <MESSAGE value=":sparkles: Multipart file upload" />
+    <option name="LAST_COMMIT_MESSAGE" value=":sparkles: Multipart file upload" />
   </component>
   <component name="VgoProject">
     <settings-migrated>true</settings-migrated>
diff --git a/pkg/internal/models/attachments.go b/pkg/internal/models/attachments.go
index d62eee5..ca5c960 100644
--- a/pkg/internal/models/attachments.go
+++ b/pkg/internal/models/attachments.go
@@ -25,7 +25,6 @@ type Attachment struct {
 	Alternative string        `json:"alt"`
 	MimeType    string        `json:"mimetype"`
 	HashCode    string        `json:"hash"`
-	UserHash    *string       `json:"user_hash"`
 	Destination AttachmentDst `json:"destination"`
 	RefCount    int           `json:"ref_count"`
 
diff --git a/pkg/internal/server/api/up_multipart_api.go b/pkg/internal/server/api/up_multipart_api.go
index 991b7bb..8b17573 100644
--- a/pkg/internal/server/api/up_multipart_api.go
+++ b/pkg/internal/server/api/up_multipart_api.go
@@ -19,7 +19,6 @@ func createAttachmentMultipartPlaceholder(c *fiber.Ctx) error {
 	var data struct {
 		Pool        string         `json:"pool" validate:"required"`
 		Size        int64          `json:"size" validate:"required"`
-		Hash        string         `json:"hash" validate:"required"`
 		Alternative string         `json:"alt"`
 		MimeType    string         `json:"mimetype"`
 		Metadata    map[string]any `json:"metadata"`
@@ -43,7 +42,6 @@ func createAttachmentMultipartPlaceholder(c *fiber.Ctx) error {
 	}
 
 	metadata, err := services.NewAttachmentPlaceholder(database.C, user, models.Attachment{
-		UserHash:    &data.Hash,
 		Alternative: data.Alternative,
 		MimeType:    data.MimeType,
 		Metadata:    data.Metadata,
@@ -57,7 +55,10 @@ func createAttachmentMultipartPlaceholder(c *fiber.Ctx) error {
 		return fiber.NewError(fiber.StatusBadRequest, err.Error())
 	}
 
-	return c.JSON(metadata)
+	return c.JSON(fiber.Map{
+		"chunk_size": viper.GetInt64("performance.file_chunk_size"),
+		"meta":       metadata,
+	})
 }
 
 func uploadAttachmentMultipart(c *fiber.Ctx) error {
diff --git a/pkg/internal/services/analyzer.go b/pkg/internal/services/analyzer.go
index caa28ca..adebacd 100644
--- a/pkg/internal/services/analyzer.go
+++ b/pkg/internal/services/analyzer.go
@@ -99,6 +99,14 @@ func AnalyzeAttachment(file models.Attachment) error {
 
 	var start time.Time
 
+	if len(file.HashCode) == 0 {
+		if hash, err := HashAttachment(file); err != nil {
+			return err
+		} else {
+			file.HashCode = hash
+		}
+	}
+
 	// Do analyze jobs
 	if !file.IsAnalyzed || len(file.HashCode) == 0 {
 		destMap := viper.GetStringMap("destinations.temporary")
@@ -158,12 +166,6 @@ func AnalyzeAttachment(file models.Attachment) error {
 				"color_space": stream.ColorSpace,
 			}
 		}
-
-		if hash, err := HashAttachment(file); err != nil {
-			return err
-		} else {
-			file.HashCode = hash
-		}
 	}
 
 	tx := database.C.Begin()
diff --git a/pkg/internal/services/merger.go b/pkg/internal/services/merger.go
index 24eefc2..a6821b3 100644
--- a/pkg/internal/services/merger.go
+++ b/pkg/internal/services/merger.go
@@ -44,5 +44,7 @@ func MergeFileChunks(meta models.Attachment, arrange []string) (models.Attachmen
 	meta.IsUploaded = true
 	database.C.Save(&meta)
 
+	PublishAnalyzeTask(meta)
+
 	return meta, nil
 }
diff --git a/settings.toml b/settings.toml
index 3c82f1f..35ebb20 100644
--- a/settings.toml
+++ b/settings.toml
@@ -30,7 +30,7 @@ access_token_duration = 300
 refresh_token_duration = 2592000
 
 [performance]
-file_chunk_size = 5242880
+file_chunk_size = 26214400
 
 [database]
 dsn = "host=localhost user=postgres password=password dbname=hy_paperclip port=5432 sslmode=disable"