net.phoenix.thrift.xml.ArgsBeanDefinitionParser.java Source code

Introduction

Here is the source code for net.phoenix.thrift.xml.ArgsBeanDefinitionParser.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */
package net.phoenix.thrift.xml;

import java.net.InetSocketAddress;

import org.apache.log4j.Logger;
import org.apache.thrift.server.AbstractNonblockingServer;
import org.apache.thrift.transport.TNonblockingServerSocket;
import org.apache.thrift.transport.TServerSocket;
import org.apache.thrift.transport.TServerTransport;
import org.springframework.beans.factory.BeanCreationException;
import org.springframework.beans.factory.BeanDefinitionStoreException;
import org.springframework.beans.factory.support.AbstractBeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.xml.ParserContext;
import org.springframework.util.StringUtils;
import org.springframework.util.xml.DomUtils;
import org.w3c.dom.Element;

/**
 *
 * @author Shamphone Lee<shamphone@gmail.com>
 *
 */
public class ArgsBeanDefinitionParser extends ComplexBeanDefinitionParser {
    private static Logger LOG = Logger.getLogger(ProcessorBeanDefinitionParser.class);
    protected static final String BEAN_NAME_SUFFIX = "-args";

    @Override
    protected String getBeanClassName(Element argsElement) {
        String className = argsElement.getAttribute("class");
        if (StringUtils.isEmpty(className)) {
            Element parent = (Element) argsElement.getParentNode();
            className = parent.getAttribute("class") + ".Args";
        }
        LOG.info("Using '" + className + "' for server args classname.");
        return className;
    }

    @Override
    protected boolean shouldGenerateId() {
        return true;
    }

    @Override
    protected String resolveId(Element element, AbstractBeanDefinition definition, ParserContext parserContext)
            throws BeanDefinitionStoreException {
        return element.getAttribute("id");
    }

    @Override
    protected void preParse(Element element, ParserContext parserContext, BeanDefinitionBuilder builder) {
        Element parent = (Element) element.getParentNode();
        String name = parent.getAttribute("name") + BEAN_NAME_SUFFIX;
        if (StringUtils.isEmpty(element.getAttribute("id"))) {
            element.setAttribute("id", name);
        }
        if (StringUtils.isEmpty(element.getAttribute("name"))) {
            element.setAttribute("name", name);
        }
        builder.addConstructorArgValue(this.buildSocket(element));
    }

    /**
     * ?arg
     */
    @Override
    protected void postParse(Element element, ParserContext parserContext, AbstractBeanDefinition current) {
        for (Element argElement : DomUtils.getChildElementsByTagName(element, "arg")) {
            parserContext.getDelegate().parseCustomElement(argElement, current);
        }
    }

    /**
     * server socket args ?
     *
     * @param argsElement
     * @param parserContext
     * @return
     */
    protected AbstractBeanDefinition buildSocket(Element argsElement) {
        int timeout = Integer.parseInt(argsElement.getAttribute("timeout"));
        BeanDefinitionBuilder builder = BeanDefinitionBuilder
                .genericBeanDefinition(this.parseSocketClass(argsElement));
        builder.addConstructorArgValue(this.buildInetSocketAddress(argsElement));
        if (timeout > 0)
            builder.addConstructorArgValue(timeout);
        return builder.getBeanDefinition();
    }

    /**
     * socketNonblockingServerTNonblockingServerSocket
     * @param element
     * @return
     */
    private Class<? extends TServerTransport> parseSocketClass(Element argsElement) {
        Element parent = (Element) argsElement.getParentNode();
        String serverClassName = parent.getAttribute("class");
        Class<?> serverClass;
        try {
            serverClass = Class.forName(serverClassName);
        } catch (ClassNotFoundException e) {
            throw new BeanCreationException("Could not found thrift server with class '" + serverClassName + "'.");
        }

        if (AbstractNonblockingServer.class.isAssignableFrom(serverClass)) {
            LOG.info("Using '" + TNonblockingServerSocket.class + "' for server transport.");
            return TNonblockingServerSocket.class;
        } else {
            LOG.info("Using '" + TServerSocket.class + "' for server transport.");
            return TServerSocket.class;
        }
    }

    /**
     *  InetSocketAddress server socket?
     *
     * @param element
     * @param parserContext
     * @return
     */
    protected AbstractBeanDefinition buildInetSocketAddress(Element element) {
        int port = Integer.parseInt(element.getAttribute("port"));
        String hostname = element.getAttribute("hostname");
        BeanDefinitionBuilder builder = BeanDefinitionBuilder.genericBeanDefinition(InetSocketAddress.class);
        if (!StringUtils.isEmpty(hostname))
            builder.addConstructorArgValue(hostname);
        builder.addConstructorArgValue(port);
        return builder.getBeanDefinition();
    }

}