com.lwq.fast.log.fastlogclient.log4j2.appender.RabbitMqAppender Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fast-log-client Show documentation
fast-log-client
The newest version!
package com.lwq.fast.log.fastlogclient.log4j2.appender;

import cn.hutool.core.util.ObjectUtil;
import com.alibaba.fastjson.JSON;
import com.lwq.fast.log.fastlogclient.log4j2.appender.util.MessageUtil;
import com.lwq.fast.log.fastlogcore.client.rabbitmq.RabbitMqClient;
import com.lwq.fast.log.fastlogcore.client.redis.RedisClient;
import com.lwq.fast.log.fastlogcore.entity.Message;
import com.lwq.fast.log.fastlogcore.util.ThreadPoolUtil;
import com.lwq.fast.log.fastlogcore.work.MessageCollection;
import org.apache.logging.log4j.core.Filter;
import org.apache.logging.log4j.core.Layout;
import org.apache.logging.log4j.core.LogEvent;
import org.apache.logging.log4j.core.appender.AbstractAppender;
import org.apache.logging.log4j.core.config.Property;
import org.apache.logging.log4j.core.config.plugins.Plugin;
import org.apache.logging.log4j.core.config.plugins.PluginAttribute;
import org.apache.logging.log4j.core.config.plugins.PluginElement;
import org.apache.logging.log4j.core.config.plugins.PluginFactory;
import org.apache.logging.log4j.core.layout.PatternLayout;

import java.io.Serializable;
import java.util.concurrent.ThreadPoolExecutor;

/**
 * 重写 appender
 * @author 刘文强
 */
@Plugin(name = "Log4jRabbitMqAppender", category = "Core", elementType = "appender", printObject = true)
public class RabbitMqAppender extends AbstractAppender {

    /**
     * 应用名称
     */
    private String appName;
    /**
     * 环境信息  如： dev， test, prod 等
     */
    private String env;
    private String host;
    private int port;
    private String virtualHost;
    private String username;
    private String password;



    protected RabbitMqAppender(String name, Filter filter, Layout layout,
                            boolean ignoreExceptions, String appName, String env, String host,
                            int port, String virtualHost, String username, String password) {
        super(name, filter, layout, ignoreExceptions, Property.EMPTY_ARRAY);
        this.appName = appName;
        this.env = env;
        this.host = host;
        this.port = port;
        this.virtualHost = virtualHost;
        this.username = username;
        this.password = password;

        //初始化
        RabbitMqClient instance = RabbitMqClient.getInstance(host, port, virtualHost, username, password);
        if (ObjectUtil.isNotNull(instance)){
            // 从队列中获取消息，发送到 MQ中
            ThreadPoolExecutor threadPoolExecutor = ThreadPoolUtil.getThreadPoolExecutor();
            threadPoolExecutor.execute(() -> {
                MessageCollection.messageCollect(instance);
            });
        }
    }

    @Override
    public void append(LogEvent logEvent) {
        Message message = MessageUtil.formatMessage(appName, env, logEvent);
        if (ObjectUtil.isNotNull(message)){
            MessageCollection.addMessage2Queue(JSON.toJSONString(message));
        }
    }
    @PluginFactory
    public static RabbitMqAppender createAppender(@PluginAttribute("name") String name,
                                               @PluginElement("filter") Filter filter,
                                               @PluginElement("layout") Layout layout,
                                               @PluginAttribute("ignoreExceptions") boolean ignoreExceptions,
                                               @PluginAttribute("appName") String appName,
                                               @PluginAttribute("env") String env,
                                               @PluginAttribute("host") String host,
                                               @PluginAttribute("port") int port,
                                               @PluginAttribute("virtualHost") String virtualHost,
                                               @PluginAttribute("username") String username,
                                               @PluginAttribute("password") String password
    ) {
        if (null == name) {
            LOGGER.error("name 为空");
            return null;
        }
        if (null == layout) {
            layout = PatternLayout.createDefaultLayout();
        }
        return new RabbitMqAppender(name, filter, layout, ignoreExceptions, appName, env, host, port, virtualHost, username, password);
    }





    public String getAppName() {
        return appName;
    }

    public void setAppName(String appName) {
        this.appName = appName;
    }

    public String getEnv() {
        return env;
    }

    public void setEnv(String env) {
        this.env = env;
    }

    public String getHost() {
        return host;
    }

    public void setHost(String host) {
        this.host = host;
    }

    public int getPort() {
        return port;
    }

    public void setPort(int port) {
        this.port = port;
    }

    public String getVirtualHost() {
        return virtualHost;
    }

    public void setVirtualHost(String virtualHost) {
        this.virtualHost = virtualHost;
    }

    public String getUsername() {
        return username;
    }

    public void setUsername(String username) {
        this.username = username;
    }

    public String getPassword() {
        return password;
    }

    public void setPassword(String password) {
        this.password = password;
    }
}