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