/**
|
* Copyright (c) 2018-2028, Chill Zhuang 庄骞 (smallchill@163.com).
|
* <p>
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
* you may not use this file except in compliance with the License.
|
* You may obtain a copy of the License at
|
* <p>
|
* http://www.apache.org/licenses/LICENSE-2.0
|
* <p>
|
* Unless required by applicable law or agreed to in writing, software
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
* See the License for the specific language governing permissions and
|
* limitations under the License.
|
*/
|
package org.springblade.modules.mountainrain.controller;
|
|
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.AllArgsConstructor;
|
import org.apache.poi.hssf.usermodel.HSSFSheet;
|
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
|
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
|
import org.apache.poi.ss.usermodel.CellType;
|
import org.springblade.core.boot.ctrl.BladeController;
|
import org.springblade.core.excel.util.ExcelUtil;
|
import org.springblade.core.log.annotation.ApiLog;
|
import org.springblade.core.mp.support.Condition;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.modules.mountainrain.entity.Yucbig;
|
import org.springblade.modules.mountainrain.entity.Yucsma;
|
import org.springblade.modules.mountainrain.excel.MgrExcel;
|
import org.springblade.modules.mountainrain.excel.MgrImporter;
|
import org.springblade.modules.mountainrain.service.IBigriverService;
|
import org.springblade.modules.mountainrain.service.ISmariverService;
|
import org.springblade.modules.mountainrain.wrapper.SmariverWrapper;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.validation.Valid;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.text.ParseException;
|
import java.text.SimpleDateFormat;
|
import java.util.*;
|
|
/**
|
* 控制器
|
*
|
* @author Blade
|
* @since 2019-11-07
|
*/
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("/smariver")
|
@Api(value = "", tags = "接口")
|
public class SmariverController extends BladeController {
|
|
private ISmariverService iSmariverService;
|
private IBigriverService iBigriverService;
|
/**
|
* 详情
|
*/
|
@ApiLog("详情")
|
@GetMapping("/detail")
|
@ApiOperationSupport(order = 1)
|
@ApiOperation(value = "详情", notes = "site")
|
public R<Yucsma> detail(Yucsma yucsma) {
|
Yucsma detail = iSmariverService.getOne(Condition.getQueryWrapper(yucsma));
|
return R.data(SmariverWrapper.build().entityVO(detail));
|
}
|
|
|
/**
|
* 新增
|
*/
|
@ApiLog("新增")
|
@PostMapping("/save")
|
@ApiOperationSupport(order = 4)
|
@ApiOperation(value = "新增", notes = "传入site")
|
public R save(@Valid @RequestBody Yucsma yucsma) {
|
return R.status(iSmariverService.save(yucsma));
|
}
|
|
/**
|
* 修改
|
*/
|
@ApiLog("修改")
|
@PostMapping("/update")
|
@ApiOperationSupport(order = 5)
|
@ApiOperation(value = "修改", notes = "传入site")
|
public R update(@Valid @RequestBody Yucsma yucsma) {
|
return R.status(iSmariverService.updateById(yucsma));
|
}
|
|
// /**
|
// * 中小河流预报数据导入
|
// */
|
// @PostMapping("import-mgr")
|
// public R importmgr(MultipartFile file) {
|
// MgrImporter mgrImporter = new MgrImporter(iSmariverService, false);
|
// ExcelUtil.save(file, mgrImporter, MgrExcel.class);
|
// return R.success("操作成功");
|
// }
|
|
/**
|
* 中小河流预警
|
*
|
* @return
|
*/
|
@GetMapping("/selectsmriver")
|
public R selectsmriver() {
|
List<Map<String, Object>> selctsmriver = iSmariverService.selctsmriver();
|
//预警数据
|
List listyj = new ArrayList();
|
//无预警数据
|
List listzc = new ArrayList();
|
for (int i = 0; i < selctsmriver.size(); i++) {
|
//警戒水位
|
String yjsw = selctsmriver.get(i).get("yjsw").toString();
|
Double yjz = Double.parseDouble(yjsw);
|
//水位
|
String zs = selctsmriver.get(i).get("Z").toString();
|
Double z = Double.parseDouble(zs);
|
if (z > yjz) {
|
listyj.add(selctsmriver.get(i));
|
} else {
|
listzc.add(selctsmriver.get(i));
|
}
|
}
|
Map map = new HashMap();
|
map.put("yj", listyj);
|
map.put("zc", listzc);
|
List list = new ArrayList();
|
list.add(map);
|
return R.data(list);
|
}
|
|
|
|
@PostMapping("import-mgr")
|
public void importmgr(MultipartFile file) throws IOException, InvalidFormatException {
|
InputStream is = file.getInputStream();
|
HSSFWorkbook workbook = new HSSFWorkbook(is);
|
//确定版本
|
boolean isExcel2003 = file.getOriginalFilename().endsWith("xls") ? true : false;
|
List<Map<String, Object>> returnMap;//返回execlClass类见下
|
if (isExcel2003) { //判断版本
|
HSSFSheet sheet = workbook.getSheetAt(1);
|
sheet.getRow(4).getCell(12).setCellType(CellType.STRING);
|
sheet.getRow(4).getCell(13).setCellType(CellType.STRING);
|
sheet.getRow(4).getCell(14).setCellType(CellType.STRING);
|
sheet.getRow(4).getCell(15).setCellType(CellType.STRING);
|
sheet.getRow(4).getCell(8).setCellType(CellType.STRING);
|
//发布时间
|
String ftimes = sheet.getRow(1).getCell(11).getStringCellValue();
|
String substring = ftimes.substring(5, ftimes.length());
|
String year = substring.substring(0, substring.lastIndexOf("年"));
|
String yues = substring.substring(substring.indexOf("年")+1, substring.lastIndexOf("月"));
|
String yue;
|
if (yues.length()==1){
|
yue=0+yues;
|
}
|
else {
|
yue=yues;
|
}
|
String r = substring.substring(substring.indexOf("月")+1, substring.lastIndexOf("日"));
|
String ms = substring.substring(substring.indexOf("日")+1, substring.lastIndexOf("时"));
|
String ftime=year+"-"+yue+"-"+r+" "+ms+":00:00";
|
//河名
|
String river = sheet.getRow(4).getCell(0).getStringCellValue();
|
//站点名称
|
String stnames = sheet.getRow(4).getCell(1).getStringCellValue();
|
String stcd = iBigriverService.selectINfo(stnames);
|
//预报时间
|
String ytimes = sheet.getRow(4).getCell(12).getStringCellValue();
|
String[] split = ytimes.split(",");
|
String a;
|
if (split[0].length()==1){
|
a=0+yues;
|
}
|
else {
|
a=yues;
|
}
|
String ytime=year+"-"+a+"-"+split[1]+" "+split[2]+":00:00";
|
//预测水位
|
String yuz = sheet.getRow(4).getCell(13).getStringCellValue();
|
//预警流量
|
String yq = sheet.getRow(4).getCell(14).getStringCellValue();
|
//预警水位
|
String yjsw = sheet.getRow(4).getCell(15).getStringCellValue();
|
String qzsw = sheet.getRow(4).getCell(8).getStringCellValue();
|
Yucsma yucsma=new Yucsma();
|
yucsma.setStcd(stcd);
|
yucsma.setStname(stnames);
|
yucsma.setRiver(river);
|
yucsma.setFtime(ftime);
|
yucsma.setYjsw(yjsw);
|
yucsma.setYtime(ytime);
|
yucsma.setYuz(yuz);
|
yucsma.setYq(yq);
|
yucsma.setQzsw(qzsw);
|
iSmariverService.save(yucsma);
|
} else {
|
//有多少个sheet
|
int sheets = workbook.getNumberOfSheets();
|
System.out.println("其他:" + sheets);
|
}
|
}
|
}
|
