JAVA 生成Excel/CSV 并上传到FTP服务器
该代码是后台代码生成excel/CSV文件，并直接上传到服务器
1、上传ftp服务器工具类
public class FTPUtils {
private static FTPUtils ftpUtils;
private FTPClient ftpClient;
private String port; // 服务器端口
private String username; // 用户登录名
private String password; // 用户登录密码
private InputStream is; // 文件下载输入流
/**
* 私有构造方法
*/
private FTPUtils() {
initConfig();
if (null == ftpClient) {
ftpClient = new FTPClient();
}
}
/**
* 获取FTPUtils对象实例
*
* @return FTPUtils对象实例
*/
public synchronized static FTPUtils getInstance() {
if (null == ftpUtils) {
ftpUtils = new FTPUtils();
}
return ftpUtils;
}
/**
* 初始化FTP服务器连接属性
*/
public void initConfig() {
// 定义配置文件输入流
port = "21";
username = "test";
password = "test";
}
/**
* 连接（配置通用连接属性）至服务器
*
* @param serverName 服务器名称
* @param remotePath 当前访问目录
* @return true：连接成功
*
* false：连接失败
*/
public boolean connectToTheServer(String serverName, String remotePath) {
// 定义返回值
boolean result = false;
try {
// 连接至服务器，端口默认为21时，可直接通过URL连接
ftpClient.connect(serverName, Integer.parseInt(port));
// 登录服务器
ftpClient.login(username, password);
// 判断返回码是否合法
if (!FTPReply.isPositiveCompletion(ftpClient.getReplyCode())) {
// 不合法时断开连接
ftpClient.disconnect();
// 结束程序
return result;
}
// 设置文件操作目录
result = ftpClient.changeWorkingDirectory(remotePath);
// 设置文件类型，二进制
result = ftpClient.setFileType(FTPClient.BINARY_FILE_TYPE);
// 设置缓冲区大小
ftpClient.setBufferSize(3072);
// 设置字符编码
ftpClient.setControlEncoding("UTF-8");
} catch (IOException e) {
e.printStackTrace();
}
return result;
}
/**
* 上传文件至FTP服务器
*
* @param serverName 服务器名称
* @param storePath 上传文件存储路径
* @param fileName 上传文件存储名称
* @param is 上传文件输入流
* @return true：上传成功
*
* false：上传失败
*/
public boolean storeFile(String serverName, String storePath, String fileName, InputStream is) {
boolean result = false;
try {
// 连接至服务器
result = connectToTheServer(serverName, storePath);
// 判断服务器是否连接成功
if (result) {
// 上传文件
result = ftpClient.storeFile(fileName, is);
}
// 关闭输入流
is.close();
} catch (IOException e) {
e.printStackTrace();
} finally {
// 判断输入流是否存在
if (null != is) {
try {
// 关闭输入流
is.close();
} catch (IOException e) {
e.printStackTrace();
}
}
// 登出服务器并断开连接
ftpUtils.logout();
}
return result;
}
/**
* 下载FTP服务器文件至本地
* 操作完成后需调用logout方法与服务器断开连接
*
* @param serverName 服务器名称
* @param remotePath 下载文件存储路径
* @param fileName 下载文件存储名称
* @return InputStream：文件输入流
*/
public InputStream retrieveFile(String serverName, String remotePath, String fileName) {
try {
boolean result = false;
// 连接至服务器
result = connectToTheServer(serverName, remotePath);
// 判断服务器是否连接成功
if (result) {
// 获取文件输入流
is = ftpClient.retrieveFileStream(fileName);
}
} catch (IOException e) {
e.printStackTrace();
}
return is;
}
/**
* 删除FTP服务器文件
*
* @param serverName 服务器名称
* @param remotePath 当前访问目录
* @param fileName 文件存储名称
* @return true：删除成功
*
* false：删除失败
*/
public boolean deleteFile(String serverName, String remotePath, String fileName) {
boolean result = false;
// 连接至服务器
result = connectToTheServer(serverName, remotePath);
// 判断服务器是否连接成功
if (result) {
try {
// 删除文件
result = ftpClient.deleteFile(fileName);
} catch (IOException e) {
e.printStackTrace();
} finally {
// 登出服务器并断开连接
ftpUtils.logout();
}
}
return result;
}
/**
* 检测FTP服务器文件是否存在
*
* @param serverName 服务器名称
* @param remotePath 检测文件存储路径
* @param fileName 检测文件存储名称
* @return true：文件存在
*
* false：文件不存在
*/
public boolean checkFile(String serverName, String remotePath, String fileName) {
boolean result = false;
try {
// 连接至服务器
result = connectToTheServer(serverName, remotePath);
// 判断服务器是否连接成功
if (result) {
// 默认文件不存在
result = false;
// 获取文件操作目录下所有文件名称
String[] remoteNames = ftpClient.listNames();
// 循环比对文件名称，判断是否含有当前要下载的文件名
for (String remoteName : remoteNames) {
if (fileName.equals(remoteName)) {
result = true;
}
}
}
} catch (IOException e) {
e.printStackTrace();
} finally {
// 登出服务器并断开连接
ftpUtils.logout();
}
return result;
}
/**
* 登出服务器并断开连接
*
* @return true：操作成功
*
* false：操作失败
*/
public boolean logout() {
boolean result = false;
if (null != is) {
try {
// 关闭输入流
is.close();
} catch (IOException e) {
e.printStackTrace();
}
}
if (null != ftpClient) {
try {
// 登出服务器
result = ftpClient.logout();
} catch (IOException e) {
e.printStackTrace();
} finally {
// 判断连接是否存在
if (ftpClient.isConnected()) {
try {
// 断开连接
ftpClient.disconnect();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
return result;
}
}
2、工具类ExcelUtils
public class ExcelBuilder {
public Workbook buildExcelDocument(List data, String[] headers, List> funcs) throws Exception {
int rowIndex = 0;
Workbook workbook = new XSSFWorkbook();
Sheet sheet = createSheet(workbook, "sheet1");
CellStyle headStyle = createHeaderStyle(workbook);
CellStyle contentStyle = createContentStyle(workbook);
createHeaderCells(sheet, rowIndex++, headStyle, headers);
for (int i = 0; i < data.size(); i++) {
createContentCells(sheet, rowIndex++, contentStyle, data.get(i), funcs);
}
return workbook;
}
private Sheet createSheet(Workbook workbook, String sheetName) {
Sheet sheet = workbook.createSheet(sheetName);
sheet.setDefaultColumnWidth(13);
return sheet;
}
private CellStyle createHeaderStyle(Workbook workbook, String fontName, int fontSize, boolean bold) {
CellStyle headStyle = workbook.createCellStyle();
Font font = workbook.createFont();
font.setFontName(fontName);
font.setBold(bold);
font.setFontHeightInPoints((short) fontSize);
headStyle.setFont(font);
return headStyle;
}
private CellStyle createHeaderStyle(Workbook workbook) {
return createHeaderStyle(workbook, "黑体", 14, true);
}
private CellStyle createContentStyle(Workbook workbook) {
return createHeaderStyle(workbook, "黑体", 10, false);
}
private void createHeaderCells(Sheet sheet, Integer rowIndex, CellStyle style, String[] headers) {
Row totalSheetRow = sheet.createRow(rowIndex);
for (int i = 0; i < headers.length; i++) {
Cell cell = totalSheetRow.createCell(i);
cell.setCellStyle(style);
cell.setCellValue(headers[i]);
}
}
private void createContentCells(Sheet sheet, Integer rowIndex, CellStyle style, T content, List> funcs) {
Row totalSheetRow = sheet.createRow(rowIndex);
for (int i = 0; i < funcs.size(); i++) {
Cell cell = totalSheetRow.createCell(i);
cell.setCellStyle(style);
cell.setCellValue(funcs.get(i).apply(content));
}
}
}
3、测试接口【CSV文件不需要合成文件工具类】
@RequestMapping(value = "writeCsv", method = RequestMethod.GET)
@ResponseBody
public String writeCsv() throws Exception {
byte[] data = "表头1,表头2\r\n测试1,测试2".getBytes("UTF-8");
try (ByteArrayInputStream inputStream = new ByteArrayInputStream(data)) {
FTPUtils ftpUtils = FTPUtils.getInstance();
ftpUtils.storeFile("192.168.31.68", "MyDownloads", "test.csv", inputStream);
} catch (Exception ex) {
ex.printStackTrace();
}
return "OK";
}
@RequestMapping(value = "writeExcel", method = RequestMethod.GET)
@ResponseBody
public String exportExcel() throws Exception {
Workbook workbook = buildExcel();
try (ByteArrayOutputStream os = new ByteArrayOutputStream()) {
workbook.write(os);
byte[] data = os.toByteArray();
try (ByteArrayInputStream inputStream = new ByteArrayInputStream(data)) {
FTPUtils ftpUtils = FTPUtils.getInstance();
ftpUtils.storeFile("192.168.31.68", "MyDownloads", "test.xlsx", inputStream);
}
} catch (Exception ex) {
ex.printStackTrace();
}
return "OK";
}
public Workbook buildExcel() throws Exception {
return new ExcelBuilder().buildExcelDocument(Lists.newArrayList("1月", "2月"), new String[]{"年月"}, Lists.newArrayList(i -> {
return "2018-" + i;
}));
}