From f6573cd9038a036d0bba826655c34ac347926ab0 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 07 四月 2023 13:56:52 +0800
Subject: [PATCH] 1

---
 src/main/java/com/terra/coal/controller/MainController.java |   35 +++++++++++++++++++++++++++++++++++
 1 files changed, 35 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/terra/coal/controller/MainController.java b/src/main/java/com/terra/coal/controller/MainController.java
index e58fffa..0c52b85 100644
--- a/src/main/java/com/terra/coal/controller/MainController.java
+++ b/src/main/java/com/terra/coal/controller/MainController.java
@@ -1,14 +1,21 @@
 package com.terra.coal.controller;
 
+import com.terra.coal.entity.ResponseMsg;
+import com.terra.coal.helper.StringHelper;
 import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiImplicitParam;
+import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.servlet.ModelAndView;
+import service.MainService;
 
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
+import java.io.File;
 
 /**
  * CoalController
@@ -18,6 +25,9 @@
 @RestController
 @RequestMapping("/main")
 public class MainController extends BaseController {
+    @Autowired
+    MainService mainService;
+
     @ApiOperation(value = "璺宠浆棣栭〉")
     @GetMapping({"/", "/toIndex"})
     public ModelAndView toIndex(ModelAndView mv, HttpServletRequest req) {
@@ -34,4 +44,29 @@
 
         return mv;
     }
+
+    @ApiOperation(value = "鍔犺浇54鏁版嵁鍏ュ簱")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "txtPath", value = "鍧椾綋鏁版嵁鏂囦欢璺緞", dataType = "String", paramType = "query", example = "D:\\鍧椾綋鏁版嵁.txt")
+    })
+    @GetMapping(value = "/load54Data")
+    public ResponseMsg<Integer> load54Data(String txtPath) {
+        try {
+            if (StringHelper.isEmpty(txtPath)) {
+                return fail("璇烽�夋嫨鈥樺潡浣撴暟鎹�.txt鈥�", null);
+            }
+
+            File f = new File(txtPath);
+            if (!f.exists() || f.isDirectory()) {
+                return fail("鏂囦欢涓嶅瓨鍦�", null);
+            }
+
+            Integer rows = mainService.load54Data(f);
+
+            return success("鎴愬姛", rows);
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+            return fail(ex.getMessage(), null);
+        }
+    }
 }

--
Gitblit v1.9.3