com.athena.meerkat.agent.netty.MeerkatClient.java Source code

Introduction

Here is the source code for com.athena.meerkat.agent.netty.MeerkatClient.java

Source

/*
 * Copyright 2013 The Athena-Peacock Project.
 *
 * 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
 *
 *      http://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.
 *
 * Revision History
 * Author         Date            Description
 * ---------------   ----------------   ------------
 * Sang-cheon Park   2013. 7. 17.      First Draft.
 */
package com.athena.meerkat.agent.netty;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;

import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import javax.inject.Inject;
import javax.inject.Named;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

import com.athena.meerkat.agent.util.AgentConfigUtil;
import com.athena.meerkat.common.constant.MeerkatConstant;
import com.athena.meerkat.common.provider.AppContext;

/**
 * <pre>
 * Netty Client
 * </pre>
 * 
 * @author Sang-cheon Park
 * @version 1.0
 */
@Component
@Qualifier("meerkatClient")
public class MeerkatClient implements ApplicationContextAware {

    private final String[] hosts = AgentConfigUtil.getConfig(MeerkatConstant.SERVER_IP).split(",");
    private final int port = Integer.parseInt(AgentConfigUtil.getConfig(MeerkatConstant.SERVER_PORT));

    @Inject
    @Named("eventLoopGroup")
    private EventLoopGroup group;

    @Inject
    @Named("meerkatClientInitializer")
    private MeerkatClientInitializer initializer;

    @Inject
    @Named("meerkatClientHandler")
    private MeerkatClientHandler handler;

    /**
     * <pre>
     * Netty Server   Bootstrap?  .
     * </pre>
     * 
     * @param bootstrap
     * @param group
     * @return
     */
    public Bootstrap createBootstrap(Bootstrap bootstrap, EventLoopGroup group, String host) {
        if (bootstrap != null) {
            bootstrap.group(group).channel(NioSocketChannel.class).option(ChannelOption.SO_KEEPALIVE, true)
                    .handler(new LoggingHandler(LogLevel.WARN)).handler(initializer).remoteAddress(host, port)
                    .connect().addListener(new MeerkatClientListener(this, host));
        }

        return bootstrap;
    }// end of createBootstrap()

    /**
     * <pre>
     * Bean ?  ?  Server? ? .
     * </pre>
     * 
     * @throws Exception
     */
    @PostConstruct
    public void start() {
        for (String host : hosts) {
            createBootstrap(new Bootstrap(), group, host);
        }
    }// end of start()

    /**
     * <pre>
     * Bean   ?  Server? ? .
     * </pre>
     */
    @PreDestroy
    public void stop() {
        if (handler.isConnected()) {
            handler.close();
        }
    }// end of stop()

    public void setApplicationContext(ApplicationContext ctx) throws BeansException {
        if (AppContext.getApplicationContext() == null) {
            AppContext.setApplicationContext(ctx);
        }

        //      String[] beans = ctx.getBeanDefinitionNames();
        //      
        //      System.out.println("==========================================");
        //      for (String name : beans) 
        //         System.out.println(name);
    }
}
// end of MeerkatClient.java