com.taotao.boot.monitor.dump.DumpProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of taotao-boot-starter-monitor Show documentation
taotao-boot-starter-monitor
There is a newer version: 2025.01
Show newest version
/*
 * Copyright (c) 2020-2030, Shuigedeng ([email protected] & https://blog.taotaocloud.top/).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.taotao.boot.monitor.dump;

import com.taotao.boot.common.constant.StarterName;
import com.taotao.boot.common.utils.log.LogUtils;
import com.taotao.boot.common.utils.servlet.RequestUtils;
import com.taotao.boot.monitor.exception.HealthException;
import com.taotao.boot.monitor.utils.ProcessUtils;
import jakarta.servlet.http.HttpServletResponse;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.concurrent.TimeUnit;

/**
 * DumpProvider
 *
 * @author shuigedeng
 * @version 2022.04
 * @since 2022-04-27 17:26:52
 */
public class DumpProvider {

    /** 最后把时间 */
    private static Long lastDumpTime = 0L;

    /**
     * getList
     *
     * @return {@link File[] }
     * @since 2022-04-27 17:26:52
     */
    public File[] getList() {
        File file = new File(".");
        return file.listFiles((dir, name) -> name.contains(".hprof"));
    }

    /**
     * list
     *
     * @since 2022-04-27 17:26:52
     */
    public void list() {
        StringBuilder stringBuilder = new StringBuilder();
        stringBuilder.append(("不要多次dump,一分钟仅限一次,仅限linux系统有效
立即Dump
"));

        for (File f : getList()) {
            if (!f.getName().endsWith(".tar")) {
                stringBuilder.append(String.format(
                        "%s (%s M)压缩
",
                        f.getName(), f.length() / 1024 / 1024, f.getName()));
            } else {
                stringBuilder.append(String.format(
                        "%s (%s M)下载
",
                        f.getName(), f.length() / 1024 / 1024, f.getName()));
            }
        }
        response(stringBuilder.toString());
    }

    /**
     * zip
     *
     * @param name name
     * @since 2022-04-27 17:26:52
     */
    public void zip(String name) {
        for (File f : getList()) {
            if (name != null && name.equals(f.getName())) {
                try {
                    Runtime.getRuntime().exec(String.format("tar -zcvf %s.tar %s", name, name), null, null);
                    response("压缩成功,请等待耐心等待,不要重复执行!");
                } catch (Exception exp) {
                    LogUtils.error(StarterName.MONITOR_STARTER, "zip 出错", exp);
                    response("压缩出错:" + exp.getMessage());
                }
            }
        }
    }

    /**
     * dump
     *
     * @since 2022-04-27 17:26:52
     */
    public void dump() {
        try {
            if (System.currentTimeMillis() - lastDumpTime < TimeUnit.MINUTES.toMillis(1)) {
                throw new HealthException("dump过于频繁,请等待后1分钟重试");
            }

            Runtime.getRuntime()
                    .exec(String.format(
                            "jmap -dump:format=b,file=heap.%s.hprof %s",
                            new SimpleDateFormat("yyyyMMddHHmmssSSS").format(new Date()), ProcessUtils.getProcessID()));

            lastDumpTime = System.currentTimeMillis();
            response("dump成功,请等待耐心等待,不要重复执行!");
        } catch (Exception exp) {
            LogUtils.error(StarterName.MONITOR_STARTER, "dump 出错", exp);
            response("dump出错:" + exp.getMessage());
        }
    }

    /**
     * download
     *
     * @param name name
     * @since 2022-04-27 17:26:52
     */
    public void download(String name) {
        for (File f : getList()) {
            if (name != null && name.equals(f.getName())) {
                HttpServletResponse response = RequestUtils.getResponse();
                response.reset();
                response.setContentType("application/x-download");
                response.addHeader("Content-Disposition", "attachment;filename=" + f.getName());
                response.addHeader("Content-Length", "" + f.length());
                response.setHeader("Content-type", "");
                try {
                    try (FileInputStream fs = new FileInputStream(f)) {
                        try (InputStream fis = new BufferedInputStream(fs)) {
                            try (OutputStream out = new BufferedOutputStream(response.getOutputStream())) {
                                response.setContentType("application/octet-stream");
                                byte[] buffer = new byte[1024];
                                int i;
                                while ((i = fis.read(buffer)) != -1) {
                                    out.write(buffer, 0, i);
                                }
                                fis.close();
                                out.flush();
                            }
                        }
                    }
                } catch (Exception exp) {
                    LogUtils.error(StarterName.MONITOR_STARTER, "download 出错", exp);
                    response("下载出错:" + exp.getMessage());
                }
            }
        }
    }

    /**
     * response
     *
     * @param html html
     * @since 2022-04-27 17:26:52
     */
    private void response(String html) {
        try {
            HttpServletResponse response = RequestUtils.getResponse();
            response.reset();
            response.setHeader("Content-type", "text/html;charset=UTF-8");
            response.setCharacterEncoding("UTF-8");
            response.getWriter().append(html);
            response.getWriter().flush();
            response.getWriter().close();
        } catch (Exception e) {
            LogUtils.error(e);
        }
    }
}