package com.whyc.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.whyc.dto.Response;
|
import com.whyc.mapper.MacInitMapper;
|
import com.whyc.mapper.MacMapper;
|
import com.whyc.pojo.Mac;
|
import com.whyc.pojo.MacInit;
|
import com.whyc.util.ActionUtil;
|
import com.whyc.util.MacUtil;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.FileNotFoundException;
|
import java.io.IOException;
|
import java.io.OutputStream;
|
import java.io.UnsupportedEncodingException;
|
import java.net.URLEncoder;
|
import java.util.Date;
|
import java.util.List;
|
|
@Service
|
public class MacService {
|
@Autowired(required = false)
|
private MacMapper mapper;
|
|
@Autowired(required = false)
|
private MacInitMapper initMapper;
|
|
//查询mac历史记录
|
public Response searMacHis() {
|
QueryWrapper wrapper = Wrappers.query();
|
wrapper.orderByDesc("record_time");
|
List<Mac> list = mapper.selectList(wrapper);
|
return new Response().setII(1, list.size() > 0, list, "mac记录");
|
}
|
|
//获取mac
|
@Transactional
|
public void getMac(HttpServletRequest req, HttpServletResponse resp, int macNum) {
|
//获取初始位置
|
QueryWrapper wrapper = Wrappers.query();
|
wrapper.orderByDesc("record_time");
|
wrapper.last("limit 1");
|
Mac mac = mapper.selectOne(wrapper);
|
String startStr = "00:00:00";
|
String stopStr = "00:00:00";
|
if (mac != null) {
|
startStr = Integer.toHexString((mac.getStopNum() + 1));
|
} else {
|
startStr = Integer.toHexString(10001);
|
}
|
List<String> list = MacUtil.getMac(startStr, macNum);
|
//获取初始值
|
QueryWrapper initwrapper = Wrappers.query();
|
initwrapper.last("limit 1");
|
MacInit macInit = initMapper.selectOne(initwrapper);
|
//导出csv
|
Date date = new Date();
|
String filename = "mac_" + ActionUtil.sdf.format(date).replace(" ", "_").replace(":", "_") + ".csv";
|
try {
|
// 转码防止乱码
|
resp.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
|
resp.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "utf-8"));
|
OutputStream out = resp.getOutputStream();
|
if (list != null) {
|
for (int i = 0; i < list.size(); i++) {
|
if (i == 0) {
|
startStr = list.get(i);
|
}
|
if (i == list.size() - 1) {
|
stopStr = list.get(i);
|
}
|
byte[] buffer = (macInit.getMacStart() + ":" + list.get(i)).getBytes();
|
out.write(buffer, 0, buffer.length);
|
out.write("\r\n".getBytes());
|
}
|
}
|
//将结果存入数据库
|
Mac inMac = new Mac();
|
inMac.setNumber(macNum);
|
inMac.setRecordTime(date);
|
inMac.setStartNum(Integer.parseInt(startStr.replace(":", ""), 16));
|
inMac.setStartStr(startStr);
|
inMac.setStopNum(Integer.parseInt(stopStr.replace(":", ""), 16));
|
inMac.setStopStr(stopStr);
|
inMac.setUname(ActionUtil.getUser().getUName());
|
mapper.insert(inMac);
|
out.close();
|
} catch (FileNotFoundException | UnsupportedEncodingException e) {
|
e.printStackTrace();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
|
//从历史导出CSV
|
public void getMacHis(HttpServletRequest req, HttpServletResponse resp, List<String> list) {
|
//导出csv
|
Date date = new Date();
|
String filename = "mac_" + ActionUtil.sdf.format(date).replace(" ", "_").replace(":", "_") + ".csv";
|
try {
|
// 转码防止乱码
|
resp.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
|
resp.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "utf-8"));
|
OutputStream out = resp.getOutputStream();
|
if (list != null) {
|
for (int i = 0; i < list.size(); i++) {
|
byte[] buffer = list.get(i).getBytes();
|
out.write(buffer, 0, buffer.length);
|
out.write("\r\n".getBytes());
|
}
|
}
|
} catch (UnsupportedEncodingException e) {
|
e.printStackTrace();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
