本文将介绍如何在 Java Structs 框架中，基于 ArrayList 的数据动态生成 CSV 文件，并将其上传到 FTP 服务器。我们将避免在磁盘上创建实际文件，而是利用 ByteArrayOutputStream 和 ByteArrayInputStream 在内存中处理文件数据，从而提高效率。最后，我们会提供一个完整的代码示例，并讨论相关的注意事项。
动态生成 CSV 文件
在传统的 Java 文件操作中，我们通常需要先创建一个 File 对象，然后通过 FileWriter 将数据写入到文件中。但是，在某些场景下，我们并不需要将文件持久化到磁盘上，而是直接将其上传到 FTP 服务器。这时，我们可以使用 ByteArrayOutputStream 和 ByteArrayInputStream 来在内存中生成 CSV 文件。
ByteArrayOutputStream 允许我们将数据写入到内存中的字节数组中，而 ByteArrayInputStream 则允许我们从内存中的字节数组中读取数据，从而实现文件数据的动态生成和读取。
以下是使用 ByteArrayOutputStream 和 ByteArrayInputStream 生成 CSV 文件的示例代码：
立即学习“Java免费学习笔记（深入）”；
import java.io.*;
public class CSVGenerator {
public static InputStream generateCSV(String[] business, String[] position) throws IOException {
ByteArrayOutputStream bos = new ByteArrayOutputStream();
try (Writer writer = new OutputStreamWriter(bos)) {
// 写入 CSV 文件的头部（如果需要）
writer.write("Business,Position\n");
// 写入 business 数据
for (String b : business) {
writer.write(b + ",");
}
// 写入 position 数据
for (String p : position) {
writer.write(p + ",");
}
writer.write("\n"); //换行符
}
return new ByteArrayInputStream(bos.toByteArray());
}
}代码解释：
- ByteArrayOutputStream bos = new ByteArrayOutputStream();：创建一个 ByteArrayOutputStream 对象，用于存储 CSV 文件的数据。
- try (Writer writer = new OutputStreamWriter(bos)) { ... }：使用 try-with-resources 语句创建 OutputStreamWriter 对象，确保资源在使用完毕后被正确关闭。
- writer.write("Business,Position\n");：写入 CSV 文件的头部，包括列名。根据实际需求修改。
- for (String b : business) { writer.write(b + ","); }：遍历 business 数组，将每个元素写入 CSV 文件，并以逗号分隔。
- for (String p : position) { writer.write(p + ","); }：遍历 position 数组，将每个元素写入 CSV 文件，并以逗号分隔。
- return new ByteArrayInputStream(bos.toByteArray());：将 ByteArrayOutputStream 中的数据转换为字节数组，并创建 ByteArrayInputStream 对象，返回该对象。
上传 CSV 文件到 FTP 服务器
有了 CSV 文件的 InputStream，我们就可以将其上传到 FTP 服务器了。以下是上传 CSV 文件到 FTP 服务器的示例代码（假设你已经有一个 SISSFTPManager 类来处理 FTP 上传）：
import java.io.InputStream;
import java.io.IOException;
public class FTPUploader {
public boolean uploadFile(InputStream isUploadFile, String dirName, String loggedInUser, String fileName) {
boolean storeRetVal = false;
String fileType = fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length());
storeRetVal = SISSFTPManager.getInstance().put(isUploadFile, dirName, fileName);
if (storeRetVal) {
try {
if (fileType.trim().equalsIgnoreCase("csv")) {
ICSAPI.getInstance().getSIMSOrderManager().createFileAudit(loggedInUser, fileName);
} else {
// Handle other file types if needed
}
} catch (Exception e) {
// TODO: Handle exceptions appropriately (logging, etc.)
e.printStackTrace();
}
System.out.println("BulkUploadAction:fileUpload SFTP Transfer file successfully!");
} else {
System.err.println("BulkUploadAction:fileUpload SFTP Transfer file FAILED!");
}
return storeRetVal;
}
}代码解释：
- SISSFTPManager.getInstance().put(isUploadFile, dirName, fileName);：调用 SISSFTPManager 类的 put 方法，将 InputStream 中的数据上传到 FTP 服务器。 dirName 指定上传的目录， fileName 指定上传的文件名。
- if (fileType.trim().equalsIgnoreCase("csv")) { ... }：判断文件类型是否为 CSV，如果是，则执行相应的操作（例如，创建文件审计记录）。
- 异常处理：添加适当的异常处理，例如日志记录，以确保应用程序的稳定性和可靠性。
在 Java Structs Action 中使用
现在，我们将上述代码集成到 Java Structs Action 中。以下是修改后的 generatePayroll 方法：
import org.apache.struts.action.Action;
import org.apache.struts.action.ActionForm;
import org.apache.struts.action.ActionForward;
import org.apache.struts.action.ActionMapping;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.InputStream;
import java.io.IOException;
public class PayrollAction extends Action {
@Override
public ActionForward generatePayroll(ActionMapping mapping, ActionForm form,
HttpServletRequest request, HttpServletResponse response) {
SessionInfoForm _form = (SessionInfoForm) form;
SisTransactionsSession _config = _form.getSisTransactionsSession();
String loggedInUser = _form.getLoggedinEmployeeDVO().getLoginId().toUpperCase();
String[] business = request.getParameterValues("selectedBusinessValues");
String[] position = request.getParameterValues("selectedPositionValues");
try {
// 1. Generate CSV InputStream
InputStream isFixedValue = CSVGenerator.generateCSV(business, position);
// 2. Upload to FTP
FTPUploader ftpUploader = new FTPUploader();
String fileName = "PAYROLL_PRM.csv"; // Define your filename
String dirName = "/payroll"; // Define your directory on the FTP server
boolean uploadSuccess = ftpUploader.uploadFile(isFixedValue, dirName, loggedInUser, fileName);
if (uploadSuccess) {
System.out.println("Payroll file uploaded successfully!");
} else {
System.err.println("Payroll file upload failed!");
}
} catch (IOException e) {
// TODO: Handle exceptions appropriately (logging, error page, etc.)
e.printStackTrace();
}
return mapping.findForward("success");
}
}代码解释：
- 获取请求参数 business 和 position，这些参数将用于生成 CSV 文件。
- 调用 CSVGenerator.generateCSV(business, position) 方法，生成 CSV 文件的 InputStream。
- 创建 FTPUploader 对象，并调用其 uploadFile 方法，将 CSV 文件上传到 FTP 服务器。
- 根据上传结果，打印相应的日志信息。
- 异常处理：添加适当的异常处理，例如日志记录，错误页面跳转等，以确保应用程序的稳定性和可靠性。
注意事项
- 异常处理： 在实际开发中，需要对可能出现的异常进行处理，例如 IOException，FtpException 等。
- FTP 连接： 确保 FTP 服务器的连接配置正确，包括主机名、端口号、用户名和密码。
- 字符编码： 确保 CSV 文件的字符编码与 FTP 服务器的字符编码一致，避免出现乱码问题。
- 安全性： 在生产环境中，需要对 FTP 连接进行加密，例如使用 SFTP 或 FTPS。
- 资源释放： 确保在使用完毕后，正确关闭 InputStream 和 OutputStream 等资源。
总结
本文介绍了如何在 Java Structs 框架中，基于 ArrayList 的数据动态生成 CSV 文件，并将其上传到 FTP 服务器。通过使用 ByteArrayOutputStream 和 ByteArrayInputStream，我们可以避免在磁盘上创建实际文件，从而提高效率。同时，我们也提供了一个完整的代码示例，并讨论了相关的注意事项。希望本文能够帮助你更好地理解和应用 Java 文件操作和 FTP 上传技术。
