数据库信息生成csv并下载

最新推荐文章于 2025-04-30 14:13:28 发布

半夏尤笙

最新推荐文章于 2025-04-30 14:13:28 发布

阅读量335

点赞数

本文链接：https://blog.youkuaiyun.com/weixin_43812065/article/details/95310441

版权

这段代码主要实现了AI模型参数的下载功能。首先根据传入的uuid获取AI模型参数，若参数为空则抛出异常。接着将参数处理成特定格式的列表，最后使用CsvUtil类生成并下载CSV文件。CsvUtil类负责创建文件、写入头部和内容等操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

public Result downModelParams(String uuid, HttpServletResponse response) {
if (StringUtils.isBlank(uuid)) {
throw new IllegalArgumentException(“参数不能为空”);
}
AiModel aiModel = aiModelMapper.selectModelByUuid(uuid);
if (aiModel == null || aiModel.getParams() == null) {
throw new IllegalArgumentException(“模型参数为空”);
}
Map<String, String> map = new LinkedHashMap<>();
map.put(“1”, “ID”);
map.put(“2”, “标题”);
map.put(“3”, “数据类型”);
List list = new ArrayList<>();
String params = aiModel.getParams();
//保持参数顺序一致
LinkedHashMap<String, Object> json = JSON.parseObject(params, LinkedHashMap.class, Feature.OrderedField);
JSONObject paramsJson = new JSONObject(true);
paramsJson.putAll(json);
for (String key : paramsJson.keySet()) {
Map<String, String> paramsMap = new LinkedHashMap<>();
paramsMap.put(“1”, key);
paramsMap.put(“2”, key);
if (paramsJson.getString(key).equals(“int”)) {
paramsMap.put(“3”, “Integer”);
} else if (paramsJson.getString(key).equals(“double”)) {
paramsMap.put(“3”, “Double”);
} else if (paramsJson.getString(key).equals(“date”)) {
paramsMap.put(“3”, “Date”);
} else if (paramsJson.getString(key).equals(“long”)) {
paramsMap.put(“3”, “Long”);
} else if (paramsJson.getString(key).equals(“boolean”)) {
paramsMap.put(“3”, “Boolean”);
} else {
paramsMap.put(“3”, “String”);
}
list.add(paramsMap);
}
try {
CsvUtil.exportDataFile(response, list, map, modelParamsPath, aiModel.getModelName());
} catch (IOException e) {
logger.error(e.getLocalizedMessage(),e);
throw new IllegalArgumentException(“下载失败”);
}
return Result.newSuccess();
}

package com.beagledata.gaea.workbench.common;
import org.apache.commons.lang3.StringUtils;

import javax.servlet.http.HttpServletResponse;
import java.io.;
import java.net.URLEncoder;
import java.util.;

/**

Created by mahongfei on 2019/3/21.
/
public class CsvUtil {
/*
- @Author: mahongfei
- @description: 生成并下载csv文件
  */
  public static void exportDataFile(HttpServletResponse response,
  List exportData,
  Map<String, String> map,
  String outPutPath,
  String fileName) throws IOException{
  File csvFile = null;
  BufferedWriter csvFileOutputStream = null;
  try {
  File file = new File(outPutPath);
  if (!file.exists()) {
  file.mkdirs();
  }
  //定义文件名格式并创建
  csvFile = new File(outPutPath + fileName+".csv");
  if(csvFile.exists()){
  csvFile.delete();
  }
  csvFile.createNewFile();
  // UTF-8使正确读取分隔符","
  //如果生产文件乱码，windows下用gbk，linux用UTF-8
  csvFileOutputStream = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(csvFile), “UTF-8”), 1024);
  //写入前段字节流，防止乱码
  csvFileOutputStream.write(getBOM());
  // 写入文件头部
  for (Iterator propertyIterator = map.entrySet().iterator(); propertyIterator.hasNext()? {
  java.util.Map.Entry propertyEntry = (java.util.Map.Entry) propertyIterator.next();
  csvFileOutputStream.write((String) propertyEntry.getValue() != null ? (String) propertyEntry.getValue() : “” );
  if (propertyIterator.hasNext()) {
  csvFileOutputStream.write(",");
  }
  }
  csvFileOutputStream.newLine();
  // 写入文件内容
  for (Iterator iterator = exportData.iterator(); iterator.hasNext()? {
  Object row = (Object) iterator.next();
  for (Iterator propertyIterator = map.entrySet().iterator(); propertyIterator.hasNext()? {
  java.util.Map.Entry propertyEntry = (java.util.Map.Entry) propertyIterator.next();
  String str=row!=null?((String)((Map)row).get( propertyEntry.getKey())):"";
  if(StringUtils.isEmpty(str)){
  str="";
  }else{
  str=str.replaceAll(""","""");
  if(str.indexOf(",")>=0){
  str="""+str+""";
  }
  }
  csvFileOutputStream.write(str);
  if (propertyIterator.hasNext()) {
  csvFileOutputStream.write(",");
  }
  }
  if (iterator.hasNext()) {
  csvFileOutputStream.newLine();
  }
  }
  csvFileOutputStream.flush();
  } catch (Exception e) {
  e.printStackTrace();
  } finally {
  try {
  csvFileOutputStream.close();
  } catch (IOException e) {
  e.printStackTrace();
  }
  }
  InputStream in = null;
  try {
  in = new FileInputStream(outPutPath+fileName+".csv");
  int len = 0;
  byte[] buffer = new byte[1024];
```
 OutputStream out = response.getOutputStream();
 response.reset();

 response.setContentType("application/csv;charset=UTF-8");
 response.setHeader("Content-Disposition","attachment; filename=" + URLEncoder.encode(fileName+".csv", "UTF-8"));
 response.setCharacterEncoding("UTF-8");
 while ((len = in.read(buffer)) > 0) {
     out.write(new byte[] { (byte) 0xEF, (byte) 0xBB, (byte) 0xBF });
     out.write(buffer, 0, len);
 }
   out.close();
```
  } catch (FileNotFoundException e) {
  
  } finally {
  if (in != null) {
  try {
  in.close();
  } catch (Exception e) {
  throw new RuntimeException(e);
  }
  }
  }