fix controller

yaklang · Jul 19, 2024 · 5f15e6f · 5f15e6f
1 parent fa7df65
commit 5f15e6f
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/...sample/src/main/java/com/example/demo/controller/fastjsondemo/FastJSONDemoController.java b/...sample/src/main/java/com/example/demo/controller/fastjsondemo/FastJSONDemoController.java
@@ -2,19 +2,17 @@
 
 import com.alibaba.fastjson.JSON;
 import org.apache.ibatis.annotations.Param;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
 
 @RestController
 @RequestMapping("/fastjson")
 public class FastJSONDemoController {
     @GetMapping("/fromId")
-    public String loadFromParam(@RequestParam(name = "id") String id) {
+    public ResponseEntity<Object> loadFromParam(@RequestParam(name = "id") String id) {
         // This is a FASTJSON Vuln typically.
         Object anyJSON = JSON.parse(id);
-        return JSON.toJSONString(anyJSON);
+        return ResponseEntity.ok(anyJSON);
     }
 }