CurlCmd

package com.rsgis.utils;

import com.rsgis.bean.CurlResponse;
import lombok.AllArgsConstructor;
import lombok.NoArgsConstructor;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

@NoArgsConstructor
@AllArgsConstructor
public class CurlCmd {
    private String url;
    private int con_timeout;
    private int ope_timeout;
    private String osName;

    public CurlResponse doCurlCmd() throws InterruptedException {
        try {
            String curlCmd = null;
            //linux中在处理和输出字符串时会自动加双引号！！！
            if (this.osName.startsWith("Windows")) {
                curlCmd = "curl -L -k --connect-timeout " + con_timeout + " -m " + ope_timeout + " -w \"\\n\"%{time_namelookup}::%{time_connect}::%{time_starttransfer}::%{time_total}::%{speed_download}::%{http_code}::%{size_download}::%{content_type}\"\\n\" \""
                        + url + "\"";
            } else if ((this.osName.contains("Linux")) || (this.osName.contains("CentOS"))) {
                curlCmd = "curl -L -k --connect-timeout " + con_timeout + " -m " + ope_timeout + " -w \"\\n\"%{time_namelookup}::%{time_connect}::%{time_starttransfer}::%{time_total}::%{speed_download}::%{http_code}::%{size_download}::%{content_type}\"\\n\" "
                        + url;
            }

            Process process = Runtime.getRuntime().exec(curlCmd);
            final InputStreamReader ir1 = new InputStreamReader(process
                    .getInputStream(), "GBK");
            final InputStreamReader ir2 = new InputStreamReader(process
                    .getErrorStream(), "GBK");
            BufferedReader br1 = new BufferedReader(ir1);

            String line = null;
            final List<String> response = new ArrayList<String>();
            final List<String> errs = new ArrayList<String>();

            //这里，如果无法正常curl，就会不断产生errorStream,因为输出缓冲区的size有限，如果不及时处理的话就会卡住，所以这里要同时处理errorStream;
            //但是如果能正常curl的话，如果不及时处理inputStream也会卡住；要及时读取那个不断生成的流
            Thread thread = new Thread(() -> {
                BufferedReader br2 = new BufferedReader(ir2);
                try {
                    String line2 = null;
                    while ((line2 = br2.readLine()) != null) {
                        if (!"".equals(line2)) {
                            errs.add(line2);
                        }
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                } finally {
                    try {
                        ir2.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            });
            thread.start();

            while ((line = br1.readLine()) != null) {
                if (!"".equals(line)) {
                    response.add(line);
                }
            }

            thread.join();
            int exitValue = process.waitFor();
            process.destroy();

            ir1.close();
            return parseMessage(response, exitValue);

        } catch (IOException e) {
            System.out.println("IOException   " + e.getMessage());
            return null;
        }
    }

    private CurlResponse parseMessage(List<String> response, int exitValue) {
        int size1 = response.size();
        CurlResponse cr = new CurlResponse();
        if (size1 == 0) {
            cr.setTime_connect(0);
            cr.setTime_server(0);
            cr.setTime_transfer(0);
            cr.setTime_total(0);
            cr.setData_size(0);
            cr.setDownload_speed(0);
            cr.setStatusCode(700);
            cr.setContentType("");
            return cr;
        } else {
            String mess = null;
            if (this.osName.startsWith("Windows")) {
                mess = response.get(size1 - 1);
            } else if ((this.osName.contains("Linux")) || (osName.contains("CentOS"))) {
                String messT = response.get(size1 - 2);
                mess = messT.substring(1, messT.length() - 1);
            }
            assert mess != null;
            String[] message = mess.split("::");
            String type = null;
            if (message.length == 8) {
                type = message[7].toLowerCase();
            } else {
                type = "";
            }
            boolean valid = false;
            if (exitValue == 0) {
                valid = ContentValid(response,type);//判断返回内容是否有效
            }

            float time0 = Float.parseFloat(message[0]);
            float time1 = Float.parseFloat(message[1]);
            float time2 = Float.parseFloat(message[2]);
            float time3 = Float.parseFloat(message[3]);
            int code = Integer.parseInt(message[5]);
            String data_type = type;

            float connect, server, tran, total, speed;
            float result_speed, result_connect, result_server, result_tran, result_total;
            if (exitValue == 0) {
                connect = time1 - time0;
                server = time2 - time1;
                tran = time3 - time2;
            } else {
                connect = 0;
                server = 0;
                tran = 0;
            }

            total = time3;
            speed = Float.parseFloat(message[4]);

            BigDecimal bg1 = new BigDecimal(speed);
            BigDecimal bg2 = new BigDecimal(connect);
            BigDecimal bg3 = new BigDecimal(server);
            BigDecimal bg4 = new BigDecimal(tran);
            BigDecimal bg5 = new BigDecimal(total);

            result_speed = bg1.setScale(3, RoundingMode.HALF_UP).floatValue();
            result_connect = bg2.setScale(3, RoundingMode.HALF_UP).floatValue();
            result_server = bg3.setScale(3, RoundingMode.HALF_UP).floatValue();
            result_tran = bg4.setScale(3, RoundingMode.HALF_UP).floatValue();
            result_total = bg5.setScale(3, RoundingMode.HALF_UP).floatValue();

            cr.setContentType(data_type);
            cr.setTime_connect(result_connect);
            cr.setTime_transfer(result_tran);
            cr.setTime_server(result_server);
            cr.setTime_total(result_total);
            cr.setData_size(Integer.parseInt(message[6]));
            cr.setDownload_speed(result_speed);

            if (exitValue != 0)//IOException, curl failed
            {
                cr.setStatusCode(exitValue);
            } else {//curl succes
                if ((code == 200) && (!valid))//invalid services or layers
                {
                    cr.setStatusCode(600);;//self-defined statusCode
                } else//valid services or layers
                {
                    cr.setStatusCode(code);
                }
            }
            return cr;
        }
    }
}