package com.zt.life.modules.mainPart.sys.service;
|
|
import com.zt.common.exception.RenException;
|
import com.zt.common.service.BaseService;
|
import com.zt.life.export.service.DownloadService;
|
import com.zt.life.modules.mainPart.sys.dao.SysMysqlDao;
|
import com.zt.life.modules.mainPart.sys.model.ColumnsTemp;
|
import com.zt.life.modules.mainPart.sys.model.SysMysql;
|
import org.apache.commons.io.input.ReversedLinesFileReader;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
|
import javax.servlet.ServletOutputStream;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.BufferedOutputStream;
|
import java.io.File;
|
import java.io.IOException;
|
import java.nio.file.Files;
|
import java.nio.file.Paths;
|
import java.nio.file.StandardOpenOption;
|
import java.util.ArrayList;
|
import java.util.Collections;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
|
@Service
|
public class SysMysqlService extends BaseService<SysMysqlDao, SysMysql> {
|
@Value("${zt.oss.log-path}")
|
private String logPath;
|
|
public SysMysql check(String sql) {
|
try {
|
SysMysql sysMysql = new SysMysql();
|
List<Map<String, Object>> data = baseDao.check(sql);
|
List<ColumnsTemp> tableColumns = new ArrayList<>();
|
|
for (Map<String, Object> map : data) {
|
ColumnsTemp obj = null;
|
for (Map.Entry<String, Object> entry : map.entrySet()) {
|
obj = new ColumnsTemp();
|
obj.setProp(entry.getKey());
|
obj.setLabel(entry.getKey());
|
tableColumns.add(obj);
|
}
|
break;
|
}
|
sysMysql.setTableColumns(tableColumns);
|
sysMysql.setTableData(data);
|
return sysMysql;
|
} catch (Exception e) {
|
e.printStackTrace();
|
throw new RenException(String.valueOf(e));
|
}
|
|
|
}
|
|
public void execute(String sql) {
|
try {
|
baseDao.execute(sql);
|
} catch (Exception e) {
|
e.printStackTrace();
|
throw new RenException(String.valueOf(e));
|
}
|
}
|
|
public void logDownload(HttpServletRequest request, HttpServletResponse response) {
|
int n = 300; // 要读取的行数
|
try {
|
File file = new File(logPath);
|
ReversedLinesFileReader reader = new ReversedLinesFileReader(file);
|
|
List<String> lines = new ArrayList<>();
|
String line;
|
|
while ((line = reader.readLine()) != null) {
|
lines.add(line);
|
if (lines.size() == n) {
|
break;
|
}
|
}
|
reader.close();
|
Collections.reverse(lines);
|
String result = lines.stream().collect(Collectors.joining(", "));
|
String formattedResult = result.replace(", ", "\n");
|
this.writeToTxt(request, response, formattedResult, "日志文件");
|
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
public void writeToTxt(HttpServletRequest request, HttpServletResponse response, String jsonString, String name) {//设置响应的字符集
|
//设置响应内容的类型
|
BufferedOutputStream buff = null;
|
ServletOutputStream outStr = null;
|
try {
|
response.setContentType("text/plain;charset=UTF-8");
|
String encodedFilename = DownloadService.getNameEncoder(request, name + "xml.txt");
|
response.addHeader("Content-Disposition", "attachment;filename=" + encodedFilename);
|
outStr = response.getOutputStream();
|
buff = new BufferedOutputStream(outStr);
|
buff.write(jsonString.getBytes("UTF-8"));
|
buff.flush();
|
buff.close();
|
} catch (Exception e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
buff.close();
|
outStr.close();
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
}
|
}
|
}
|
