实战：第十二章：txt文件转xml文件

开发不就这么点事吗，有个啥好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;
    }
}