main finish

Author: ma5d

src/main/java/org/ma5d/javamock/controller/MockController.java

@@ -4,7 +4,6 @@
 import jakarta.servlet.http.HttpServletRequest;
 import org.ma5d.javamock.dto.SaveParam;
 import org.ma5d.javamock.service.JavaMockServiceImpl;
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Controller;
 import org.springframework.ui.Model;
 import org.springframework.web.bind.annotation.*;
@@ -15,10 +14,9 @@
 @Controller
 public class MockController {
     @Resource
-    JavaMockServiceImpl javaMockService;
-    @Autowired
-    private JavaMockServiceImpl javaMockServiceImpl;
+    private JavaMockServiceImpl javaMockService;
 
+    // region 查
     @GetMapping("/")
     public String index() {
         return "redirect:/index";
@@ -30,26 +28,47 @@ public String index(Model model) throws SQLException {
         model.addAttribute("saveParams", saveParams);
         return "index";
     }
+    // endregion
+
+    // region 增
+    @GetMapping("/new")
+    public String newItem(Model model) {
+        SaveParam saveParam = new SaveParam();
+        model.addAttribute("saveParam", saveParam);
+        return "detail";
+    }
 
     @PostMapping("/save")
     public String configure(@ModelAttribute("saveParam") SaveParam saveParam) throws SQLException {
         javaMockService.saveConfig(saveParam);
         return "redirect:/";
     }
+    // endregion
 
+    // region 删
+    @GetMapping("/delete/{id}")
+    public String delete(@PathVariable String id) throws SQLException {
+        javaMockService.deleteByTimeStamp(id);
+        return "redirect:/";
+    }
+    // endregion
+
+    // region 改
     @GetMapping("/detail/{timeStamp}")
     public String detail(@PathVariable String timeStamp, Model model) throws SQLException {
         SaveParam saveParam = javaMockService.queryLine(timeStamp);
         model.addAttribute("saveParam", saveParam);
         return "detail";
     }
+    // endregion
 
+    // region 主要逻辑
     @ResponseBody
     @GetMapping("/p/**")
     public String proxyGet(HttpServletRequest request) throws SQLException {
         String requestURI = request.getRequestURI();
         String pathWithParam = "/" + requestURI.split("/p/")[1];
-        SaveParam saveParam = javaMockServiceImpl.getSaveParamByPathWithParam(pathWithParam);
+        SaveParam saveParam = javaMockService.getSaveParamByPathWithParam(pathWithParam);
         return saveParam.getResponse();
     }
 
@@ -58,8 +77,8 @@ public String proxyGet(HttpServletRequest request) throws SQLException {
     public String proxyPost(HttpServletRequest request) throws SQLException {
         String requestURI = request.getRequestURI();
         String pathWithParam = "/" + requestURI.split("/p/")[1];
-        SaveParam saveParam = javaMockServiceImpl.getSaveParamByPathWithParam(pathWithParam);
+        SaveParam saveParam = javaMockService.getSaveParamByPathWithParam(pathWithParam);
         return saveParam.getResponse();
     }
-
+    // endregion
 }

src/main/java/org/ma5d/javamock/service/JavaMockServiceImpl.java

@@ -13,21 +13,25 @@
 import java.util.List;
 
 @Service
+@SuppressWarnings("UnusedReturnValue")
 public class JavaMockServiceImpl {
 
     @Resource
     HikariDataSource dataSource;
 
+
     public boolean saveConfig(SaveParam saveParam) throws SQLException {
         Connection connection = dataSource.getConnection();
         // language=SQLite
-        String query = "insert into java_mock(time_stamp, adomain, path_param, response) values(?, ?, ?, ?)";
+        String query = "replace into java_mock(time_stamp, adomain, path_param, response) values(?, ?, ?, ?)";
         PreparedStatement stmt = connection.prepareStatement(query);
         stmt.setString(1, saveParam.getTimeStamp());
         stmt.setString(2, saveParam.getDomainCom());
         stmt.setString(3, saveParam.getPathWithParam());
         stmt.setString(4, saveParam.getResponse());
-        return stmt.execute();
+        boolean execute = stmt.execute();
+        connection.close();
+        return execute;
     }
 
     public List<SaveParam> queryAllLines() throws SQLException {
@@ -73,4 +77,15 @@ public SaveParam getSaveParamByPathWithParam(String pathWithParam) throws SQLExc
         connection.close();
         return saveParam;
     }
+
+    public boolean deleteByTimeStamp(String id) throws SQLException {
+        Connection connection = dataSource.getConnection();
+        // language=SQLite
+        String query = "delete from java_mock where time_stamp = ?";
+        PreparedStatement preparedStatement = connection.prepareStatement(query);
+        preparedStatement.setString(1, id);
+        boolean execute = preparedStatement.execute();
+        connection.close();
+        return execute;
+    }
 }

src/main/resources/templates/index.html

@@ -21,6 +21,7 @@
         <th>域名</th>
         <th>路径及参数</th>
         <th>返回数据</th>
+        <th>删除</th>
     </tr>
     </thead>
     <tbody>
@@ -41,6 +42,12 @@
 
         <td th:text="${#strings.length(saveParam.response) > 100 ? saveParam.response.substring(0, 100) + '...' : saveParam.response}"></td>
 
+        <td>
+            <a th:href="@{http://{domainCom}/delete/{timeStamp}(domainCom=${saveParam.domainCom}, timeStamp=${saveParam.timeStamp})}"
+               th:text="${saveParam.timeStamp}">
+            </a>
+        </td>
+
     </tr>
     </tbody>
 </table>