开发不就这么点事吗，有个啥好bb的
controller
@RequestMapping("/DataDistributionController")
@RestController
public class DataDistributionController {
@Autowired
DataDistributionService dataDistributionService;
@PostMapping("/dataDistribution")
public JSONObject dataDistribution(DataDistributionVo dataDistributionVo){
return dataDistributionService.dataDistribution(dataDistributionVo);
}
}
serviceimpl
import cn.dreamit.dreamweb.util.UUIDGenerator;
import cn.dreamit.one.databus.constant.DatabusConstant;
import cn.dreamit.one.databus.service.DataDistributionService;
import cn.dreamit.one.databus.service.SubscriptionInfoService;
import cn.dreamit.one.databus.vo.DataDistributionVo;
import cn.dreamit.one.databus.vo.SendMqDataVo;
import cn.dreamit.one.entity.SubscriptionEntity;
import cn.dreamit.one.util.StringUtil;
import com.alibaba.fastjson.JSONObject;
import org.dom4j.Document;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.io.XMLWriter;
import org.springframework.amqp.core.AmqpTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;
import org.springframework.web.multipart.MultipartFile;
import xsf.data.DBManager;
import xsf.data.DbType;
import xsf.data.Parameter;
import xsf.data.Sql;
import java.io.*;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
@Service
public class DataDistributionServiceImpl implements DataDistributionService {
@Autowired
SubscriptionInfoService subscriptionInfoService;
@Autowired
private AmqpTemplate rabbitTemplate;
/**
* 数据发布接口需要执行：
* 1.判断发布范围（releaseScope=‘serviceCode,serviceCode,serviceCode’）不能超过订阅范围String[]
* 2.将文件校验(是否为空，是否超过300M)并转为XML文件存储(盘符:/App/Service/Date/xx.xml)
* xml格式：<list><service>serviceCode</service><range>releaseScope</range><data><![CDATA[文本内容]]></![CDATA[文本内容]]></data></list>
* 3.发送MQ：将订阅信息，发布范围，文件存储路径，token发送
* 4.添加数据发布记录
* 5.添加数据发布日志记录
* @param dataDistributionVo
* @return
*/
@Transactional
@Override
public JSONObject dataDistribution(DataDistributionVo dataDistributionVo) {
//错误信息收集
String errorLog = "";
StringBuffer stringBuffer = new StringBuffer(errorLog);
//发布范围不能超过订阅范围 获取订阅信息
JSONObject subscriptionInfo = subscriptionInfoService.getSubscriptionInfo(dataDistributionVo.getAppId(), dataDistributionVo.getAppCode(),
dataDistributionVo.getServiceName(), dataDistributionVo.getServiceCode(), null, null, null);
List<SubscriptionEntity> subscriptionList = (List<SubscriptionEntity>) subscriptionInfo.get("subscriptionList");
if(CollectionUtils.isEmpty(subscriptionList)){
errorLog = DatabusConstant.ERROR_LOG_SUBSCRIPTION;
stringBuffer.append(errorLog).append(";");
}else {
//获取订阅范围： 将所有服务标识获取出来存入字符数组，这个数组就是订阅范围
String[] strings = new String[subscriptionList.size()];
for (int i = 0; i < subscriptionList.size(); i++) {
strings[i] = subscriptionList.get(i).getSERVICE_CODE();
}
if(!StringUtil.isEmpty(dataDistributionVo.getReleaseScope())){
//遍历发布范围 OA,ERP,RS
String[] split = dataDistributionVo.getReleaseScope().split(",");
for (String dataDistributionServiceCode : split) {
//判断该发布范围是否包含在订阅范围内
boolean contains = Arrays.asList(strings).contains(dataDistributionServiceCode);
if(!contains){
//发布不包含在订阅范围内
errorLog = DatabusConstant.ERROR_LOG_RELEASESCOPE;
stringBuffer.append(errorLog).append(";");
}
}
}
}
String fileUrl = "";
//文件和文件名不能为空
if(dataDistributionVo.getFile().isEmpty() || "".equals(dataDistributionVo.getFile().getOriginalFilename())){
errorLog = DatabusConstant.ERROR_LOG_FILE_NOT_EMPTY;
stringBuffer.append(errorLog).append(";");
}else {
//文件大小不能超过300M,314572800是对应300M的字节
if(dataDistributionVo.getFile().getSize() > 314572800){
errorLog = DatabusConstant.ERROR_LOG_FILE_SIZE;
stringBuffer.append(errorLog).append(";");
}else {
String format = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
//拼接路径 盘符：D:/ + App/Service/Date/
fileUrl = DatabusConstant.DISK_CHARACTER + dataDistributionVo.getAppCode() + "/" + dataDistributionVo.getServiceCode() + "/" + format + "/";
try {
File saveFile = new File(fileUrl);
if(!saveFile.exists()){
//不存在时,创建文件夹
saveFile.mkdirs();
}
//将txt文本文件转成xml文件写入磁盘
txtToXml(dataDistributionVo,fileUrl,errorLog,stringBuffer);
// //直接保存的txt文件
// File data = new File(saveFile, dataDistributionVo.getFile().getName());
// //保存文件,使用transferTo()保存必须要绝对路径且文件夹必须已存在,否则报错
// dataDistributionVo.getFile().transferTo(data);
} catch (Exception e) {
errorLog = DatabusConstant.ERROT_LOG_FILE_UPLOAD + e.getMessage();
stringBuffer.append(errorLog).append(";");
System.out.println(e.getMessage());
e.printStackTrace();
}
}
}
//发送MQ
String sendKey = "serviceCode:" + dataDistributionVo.getServiceCode();
SendMqDataVo sendMqDataVo = new SendMqDataVo();
sendMqDataVo.setFileUrl(fileUrl + dataDistributionVo.getFile().getName());
sendMqDataVo.setReleaseRange(dataDistributionVo.getReleaseScope());
sendMqDataVo.setSubscriptionEntityList(subscriptionList);
sendMqDataVo.setToken(dataDistributionVo.getPubToken());
rabbitTemplate.convertAndSend(sendKey,sendMqDataVo);
//添加数据发布记录
String uuid = UUIDGenerator.getUUID();
Sql dataServiceSql = new Sql("INSERT INTO `g_app_data_service`(ID,APP_ID,APP_CODE,SERVICE_NAME,SERVICE_CODE,MEMO) VALUES(?,?,?,?,?,?)");
dataServiceSql.addParameter(new Parameter("ID", uuid , DbType.STRING));
dataServiceSql.addParameter(new Parameter("APP_ID", dataDistributionVo.getAppId(), DbType.STRING));
dataServiceSql.addParameter(new Parameter("APP_CODE", dataDistributionVo.getAppCode(), DbType.STRING));
dataServiceSql.addParameter(new Parameter("SERVICE_NAME", dataDistributionVo.getServiceName(), DbType.STRING));
dataServiceSql.addParameter(new Parameter("SERVICE_CODE", dataDistributionVo.getServiceCode(), DbType.STRING));
dataServiceSql.addParameter(new Parameter("MEMO", dataDistributionVo.getMemo(), DbType.STRING));
String flag = DBManager.execute(dataServiceSql) ? uuid : null;
if(StringUtil.isEmpty(flag)){
errorLog = DatabusConstant.ERROR_LOG_INSERT_DATADISTRIBUTION;
stringBuffer.append(errorLog).append(";");
}
//添加数据发布日志记录
Sql sqlDataServiceLog = new Sql("INSERT INTO `g_app_data_service_log`(ID,PUBAPP_CODE,PUBAPP_NAME,SERVICE_ID,SERVICE_NAME,PUB_RANGE,PUB_TOKEN,MQ_CHANNEL,PUB_TIME,ERROE_LOG) VALUES(?,?,?,?,?,?,?,?,?,?)");
sqlDataServiceLog.addParameter(new Parameter("ID", uuid , DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("PUBAPP_CODE", dataDistributionVo.getAppId() , DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("PUBAPP_NAME", null, DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("SERVICE_ID", null , DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("SERVICE_NAME", dataDistributionVo.getServiceName() , DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("PUB_RANGE", dataDistributionVo.getServiceCode() , DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("PUB_TOKEN", dataDistributionVo.getPubToken() , DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("MQ_CHANNEL", sendKey , DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("PUB_TIME", new Date(), DbType.STRING));
sqlDataServiceLog.addParameter(new Parameter("ERROE_LOG", stringBuffer.toString(), DbType.STRING));
String bool = DBManager.execute(dataServiceSql) ? uuid : null;
if(StringUtil.isEmpty(bool)){
errorLog = DatabusConstant.ERROR_LOG_INSERT_DATADISTRIBUTION_Log;
stringBuffer.append(errorLog).append(";");
}
//返回结果
JSONObject jsonObject = new JSONObject();
if(StringUtil.isEmpty(errorLog)){
jsonObject.put("success","执行成功");
}else {
jsonObject.put("faile",stringBuffer.toString());
}
return jsonObject;
}
/**
* txt类型转xml
* @param dataDistributionVo
* @param fileUrl
* @param errorLog
* @param stringBuffer
*/
private void txtToXml(DataDistributionVo dataDistributionVo,String fileUrl,String errorLog,StringBuffer stringBuffer){
//multipartFile类型转File类型
File multipartFileToFile = multipartFileToFile(dataDistributionVo.getFile(),fileUrl,errorLog,stringBuffer);
if(multipartFileToFile != null){
StringBuffer buffer = new StringBuffer();
try {
//解决中文乱码
String code = getFileEncode(multipartFileToFile.getAbsolutePath(),errorLog,stringBuffer);
if("asci".equals(code)){
// 这里采用GBK编码，而不用环境编码格式，因为环境默认编码不等于操作系统编码
code = "GBK";
}
BufferedReader br = new BufferedReader(new InputStreamReader(new FileInputStream(multipartFileToFile), code));
String line = null;
while((line = br.readLine()) != null){
buffer.append(line.trim());
}
String dataText = buffer.toString();
String txtToXml = "<![CDATA[" + dataText + "]]>";
//文本转xml
Document document = DocumentHelper.createDocument();
//需要一个根节点list,否则重新写入是会出现无法向此文档添加已存在的元素
Element list = document.addElement("list");
Element service = list.addElement("service");
Element range = list.addElement("range");
Element data = list.addElement("data");//<![CDATA[文本内容]]>
//设置节点数据
data.setText(txtToXml);
service.setText(dataDistributionVo.getServiceCode());
range.setText(dataDistributionVo.getReleaseScope());
//写入xml文件
String xmlUrl = fileUrl + dataDistributionVo.getFile().getOriginalFilename() + ".xml";
//如果存在则重新写入
Writer filewriter = new FileWriter(xmlUrl);
XMLWriter xmlWriter = new XMLWriter(filewriter);
xmlWriter.write(document);
xmlWriter.close();
} catch (IOException e) {
errorLog = DatabusConstant.ERROR_LOG_TXT_TO_XML;
stringBuffer.append(errorLog).append(";");
System.out.println(e.getMessage());
e.printStackTrace();
}
}else {
//如果file为空，则在multipartFile转File过程中发生异常，导致传出来是一个空的file
errorLog = DatabusConstant.ERROR_LOG_MULTIPARTFILE_TO_FILE;
stringBuffer.append(errorLog).append(";");
}
}
/**
* multipartFile转File
* @param file
* @param errorLog
* @param stringBuffer
* @return
*/
private File multipartFileToFile(MultipartFile file,String fileUrl,String errorLog,StringBuffer stringBuffer){
File toFile = null;
try {
InputStream ins = file.getInputStream();
toFile = new File(fileUrl,file.getOriginalFilename());
OutputStream os = new FileOutputStream(toFile);
//读取的内容的长度,用来装数据
int bytesRead = 0;
//缓冲区数组
byte[] buffer = new byte[8192];
while ((bytesRead = ins.read(buffer,0,8192)) != -1) {
os.write(buffer, 0, bytesRead);
}
os.close();
ins.close();
} catch (IOException e) {
errorLog = DatabusConstant.ERROR_LOG_MULTIPARTFILE_TO_FILE;
stringBuffer.append(errorLog).append(";");
System.out.println(e.getMessage());
e.printStackTrace();
}
return toFile;
}
/**
* 获取文件编码
* @param path 文件的路径
* @return
*/
private String getFileEncode(String path,String errorLog,StringBuffer stringBuffer) {
String charset ="asci";
byte[] first3Bytes = new byte[3];
BufferedInputStream bis = null;
try {
boolean checked = false;
bis = new BufferedInputStream(new FileInputStream(path));
bis.mark(0);
int read = bis.read(first3Bytes, 0, 3);
if (read == -1)
return charset;
if (first3Bytes[0] == (byte) 0xFF && first3Bytes[1] == (byte) 0xFE) {
charset = "Unicode";//UTF-16LE
checked = true;
} else if (first3Bytes[0] == (byte) 0xFE && first3Bytes[1] == (byte) 0xFF) {
charset = "Unicode";//UTF-16BE
checked = true;
} else if (first3Bytes[0] == (byte) 0xEF && first3Bytes[1] == (byte) 0xBB && first3Bytes[2] == (byte) 0xBF) {
charset = "UTF8";
checked = true;
}
bis.reset();
if (!checked) {
int loc = 0;
while ((read = bis.read()) != -1) {
loc++;
if (read >= 0xF0) {
break;
}
//单独出现BF以下的，也算是GBK
if (0x80 <= read && read <= 0xBF) {
break;
}
if (0xC0 <= read && read <= 0xDF) {
read = bis.read();
if (0x80 <= read && read <= 0xBF) {
//双字节 (0xC0 - 0xDF) (0x80 - 0xBF),也可能在GB编码内
continue;
}else{
break;
}
} else if (0xE0 <= read && read <= 0xEF) { //也有可能出错，但是几率较小
read = bis.read();
if (0x80 <= read && read <= 0xBF) {
read = bis.read();
if (0x80 <= read && read <= 0xBF) {
charset = "UTF-8";
break;
} else{
break;
}
} else{
break;
}
}
}
}
} catch (Exception e) {
errorLog = DatabusConstant.ERROR_LOG_FILE_ENCODE;
stringBuffer.append(errorLog).append(";");
System.out.println(e.getMessage());
e.printStackTrace();
} finally {
if (bis != null) {
try {
bis.close();
} catch (IOException ex) {
errorLog = DatabusConstant.ERROR_LOG_FILE_ENCODE;
stringBuffer.append(errorLog).append(";");
System.out.println(ex.getMessage());
}
}
}
return charset;
}
}
import org.springframework.web.multipart.MultipartFile;
public class DataDistributionVo {
MultipartFile file;//数据包
String releaseScope;//发布范围 例如：OA,ERP,RS
String appId;//应用id
String appCode;//应用标识
String serviceName;//服务名称
String serviceCode;//服务标识
String memo;//备注
String pubToken;//token
public MultipartFile getFile() {
return file;
}
public void setFile(MultipartFile file) {
this.file = file;
}
public String getReleaseScope() {
return releaseScope;
}
public void setReleaseScope(String releaseScope) {
this.releaseScope = releaseScope;
}
public String getAppId() {
return appId;
}
public void setAppId(String appId) {
this.appId = appId;
}
public String getAppCode() {
return appCode;
}
public void setAppCode(String appCode) {
this.appCode = appCode;
}
public String getServiceName() {
return serviceName;
}
public void setServiceName(String serviceName) {
this.serviceName = serviceName;
}
public String getServiceCode() {
return serviceCode;
}
public void setServiceCode(String serviceCode) {
this.serviceCode = serviceCode;
}
public String getMemo() {
return memo;
}
public void setMemo(String memo) {
this.memo = memo;
}
public String getPubToken() {
return pubToken;
}
public void setPubToken(String pubToken) {
this.pubToken = pubToken;
}
}