- optimize

KinNeko-De · Dec 20, 2023 · 70e9085 · 70e9085
1 parent d84550d
commit 70e9085
Show file tree

Hide file tree

Showing 3 changed files with 530 additions and 19 deletions.
diff --git a/internal/app/file/file_service.go b/internal/app/file/file_service.go
@@ -72,11 +72,11 @@ func (s *FileServiceServer) StoreFile(stream apiRestaurantFile.FileService_Store
 		}
 	}
 
-	n := 512
-	if size < uint64(n) {
-		n = int(size)
+	sniffByteCount := size
+	if sniffByteCount > 512 {
+		sniffByteCount = 512
 	}
-	contentType := http.DetectContentType(sniff[:n])
+	contentType := http.DetectContentType(sniff[:sniffByteCount])
 
 	fileUuid, err := apiProtobuf.ToProtobuf(fileId)
 	if err != nil {

diff --git a/internal/app/file/file_service_test.go b/internal/app/file/file_service_test.go
@@ -40,7 +40,6 @@ func TestFile(t *testing.T) {
 	output, _ := os.Create("test.txt")
 	output.WriteString(variable)
 	output.Close()
-
 }
 
 func TestStoreFile_TextFile(t *testing.T) {