RabbitMQ之(1)连接工具

川长思鸟来 2021-09-28 03:48 585阅读 0赞

package rabbitmq;
    
    import java.io.IOException;
    
    import org.apache.log4j.LogManager;
    import org.apache.log4j.Logger;
    
    import com.rabbitmq.client.Channel;
    import com.rabbitmq.client.Connection;
    import com.rabbitmq.client.ConnectionFactory;
    /**
     * @ClassName: RabbitMQConnection 
     * @Description: RabbitMQ连接工具
     * @author wxy 
     * @date 2019年1月24日 下午1:31:12 
     */
    public class RabbitMQConnection {
    	protected final Logger LOG = LogManager.getLogger(RabbitMQConnection.class);
    	//是否已初始化
    	private boolean isInit = false;
    	//mq地址
    	private String rabbitmq_url = "localhost";
    	//端口号
    	private int rabbitmq_port = 80;
    	//用户名
    	private String userName = "zxt";
    	//密码
    	private String password = "123456";
    	//队列标识
    	protected String rabbit_queue_name = "rabbit_mq_zxt";
    	//连接工厂
    	private ConnectionFactory connectionFactory;
    	//连接
    	private Connection connection;
    	//渠道
    	protected Channel channel;
    	
    	public RabbitMQConnection() {
    	}
    
    	public RabbitMQConnection(String rabbitmq_url, int rabbitmq_port, String userName, String password,
    			String rabbit_queue_name) {
    		LOG.info("MQ地址：" + rabbitmq_url);
    		LOG.info("端口：" + rabbitmq_port);
    		LOG.info("用户名：" + userName);
    		LOG.info("密码：" + password);
    		LOG.info("队列标识：" + rabbit_queue_name);
    		this.rabbitmq_url = rabbitmq_url;
    		this.rabbitmq_port = rabbitmq_port;
    		this.userName = userName;
    		this.password = password;
    		this.rabbit_queue_name = rabbit_queue_name;
    	}
    	
    	//初始化
    	public void initConnection() {
    		if(!isInit) {
    			LOG.info("开始初始化");
    			isInit = true;
    			connectionFactory = new ConnectionFactory();
    			connectionFactory.setConnectionTimeout(8000);
    			connectionFactory.setHost(rabbitmq_url);
    			connectionFactory.setPassword(password);
    			connectionFactory.setPort(rabbitmq_port);
    			connectionFactory.setUsername(userName);
    			LOG.info("开始创建渠道");
    			getChannel();
    		}
    		LOG.info("无需初始化");
    	}
    	
    	//生成渠道
    	public Channel getChannel() {
    		if(null == connection) {
    			try {
    				//生成一个新的连接
    				connection = connectionFactory.newConnection();
    			} catch (Exception e) {
    				e.printStackTrace();
    				LOG.error(e);
    			}
    		}
    		
    		if(null == channel) {
    			try {
    				//通过连接生成一个新的渠道
    				channel = connection.createChannel();
    			} catch (Exception e) {
    				LOG.error(e);
    			}
    		}
    		
    		if(null != channel) {
    			try {
    				channel.queueDeclare(rabbit_queue_name, true, false, false, null);
    			} catch (IOException e) {
    				e.printStackTrace();
    			}
    		}
    		
    		return channel;
    	}
    	
    	//关闭渠道、连接
    	public void close() {
    		//关闭渠道
    		if(null != channel) {
    			try {
    				channel.abort();
    			} catch (Exception e) {
    				e.printStackTrace();
    			}
    			
    			try {
    				channel.close();
    			} catch (Exception e) {
    				e.printStackTrace();
    			}
    			
    			channel = null;
    		}
    		//关闭连接
    		if(null != connection) {
    			try {
    				connection.abort();
    			} catch (Exception e) {
    				e.printStackTrace();
    			}
    			
    			try {
    				connection.close();
    			} catch (Exception e) {
    				e.printStackTrace();
    			}
    			
    			connection = null;
    		}
    	}
    	
    }