update

EduOJ · Jul 16, 2024 · a79c9e9 · a79c9e9
1 parent ccc9d0e
commit a79c9e9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/app/controller/problem.go b/app/controller/problem.go
@@ -401,8 +401,8 @@ func CreateTestCase(c echo.Context) error {
 		panic(errors.Wrap(err, "could not create test case"))
 	}
 	// upload to minio
-	utils.MustPutTestCase(*req.Sanitize, inputFile, c.Request().Context(), "problems", fmt.Sprintf("%d/input/%d.in", problem.ID, testCase.ID))
-	utils.MustPutTestCase(*req.Sanitize, outputFile, c.Request().Context(), "problems", fmt.Sprintf("%d/output/%d.out", problem.ID, testCase.ID))
+	utils.MustPutInputFile(*req.Sanitize, inputFile, c.Request().Context(), "problems", fmt.Sprintf("%d/input/%d.in", problem.ID, testCase.ID))
+	utils.MustPutObject(outputFile, c.Request().Context(), "problems", fmt.Sprintf("%d/output/%d.out", problem.ID, testCase.ID))
 
 	return c.JSON(http.StatusCreated, response.CreateTestCaseResponse{
 		Message: "SUCCESS",
@@ -487,11 +487,11 @@ func UpdateTestCase(c echo.Context) error {
 	}
 
 	if inputFile != nil {
-		utils.MustPutTestCase(*req.Sanitize, inputFile, c.Request().Context(), "problems", fmt.Sprintf("%d/input/%d.in", problem.ID, testCase.ID))
+		utils.MustPutInputFile(*req.Sanitize, inputFile, c.Request().Context(), "problems", fmt.Sprintf("%d/input/%d.in", problem.ID, testCase.ID))
 		testCase.InputFileName = inputFile.Filename
 	}
 	if outputFile != nil {
-		utils.MustPutTestCase(*req.Sanitize, outputFile, c.Request().Context(), "problems", fmt.Sprintf("%d/output/%d.out", problem.ID, testCase.ID))
+		utils.MustPutObject(outputFile, c.Request().Context(), "problems", fmt.Sprintf("%d/output/%d.out", problem.ID, testCase.ID))
 		testCase.OutputFileName = outputFile.Filename
 	}
 

diff --git a/base/utils/helpers.go b/base/utils/helpers.go
@@ -62,7 +62,7 @@ func MustPutObject(object *multipart.FileHeader, ctx context.Context, bucket str
 	}
 }
 
-func MustPutTestCase(sanitize bool, object *multipart.FileHeader, ctx context.Context, bucket string, path string) {
+func MustPutInputFile(sanitize bool, object *multipart.FileHeader, ctx context.Context, bucket string, path string) {
 	src, err := object.Open()
 	if err != nil {
 		panic(err)
@@ -80,10 +80,11 @@ func MustPutTestCase(sanitize bool, object *multipart.FileHeader, ctx context.Co
 		writer := bufio.NewWriter(tempFile)
 		for scanner.Scan() {
 			line := strings.ReplaceAll(scanner.Text(), "\r\n", "\n") // replace '\r\n' to '\n'
-			_, err := fmt.Fprint(writer, line)
+
 			if !strings.HasSuffix(line, "\n") {
 				line += "\n"
 			}
+			_, err := fmt.Fprint(writer, line)
 			if err != nil {
 				panic(err)
 			}