Java tutorial
/* * Licensed to David Pilato (the "Author") under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. Author 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 fr.pilato.spring.elasticsearch.xml; import fr.pilato.spring.elasticsearch.ElasticsearchClientFactoryBean; import fr.pilato.spring.elasticsearch.ElasticsearchTransportClientFactoryBean; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.springframework.beans.factory.config.BeanDefinition; import org.springframework.beans.factory.support.BeanDefinitionBuilder; import org.springframework.beans.factory.support.GenericBeanDefinition; import org.springframework.beans.factory.xml.BeanDefinitionParser; import org.springframework.beans.factory.xml.ParserContext; import org.w3c.dom.Element; public class ClientBeanDefinitionParser implements BeanDefinitionParser { protected static final Log logger = LogFactory.getLog(ClientBeanDefinitionParser.class); public BeanDefinition parse(Element element, ParserContext parserContext) { // When node is not null, we should build a client. // When node is null, we want to build a transport client. String id = XMLParserUtil.getElementStringValue(element, "id"); String name = XMLParserUtil.getElementStringValue(element, "name"); String node = XMLParserUtil.getElementStringValue(element, "node"); String esNodes = XMLParserUtil.getElementStringValue(element, "esNodes"); String settingsFile = XMLParserUtil.getElementStringValue(element, "settingsFile"); String properties = XMLParserUtil.getElementStringValue(element, "properties"); boolean forceMapping = XMLParserUtil.getElementBooleanValue(element, "forceMapping"); boolean forceTemplate = XMLParserUtil.getElementBooleanValue(element, "forceTemplate"); boolean mergeMapping = XMLParserUtil.getElementBooleanValue(element, "mergeMapping"); boolean mergeSettings = XMLParserUtil.getElementBooleanValue(element, "mergeSettings"); boolean autoscan = XMLParserUtil.getElementBooleanValue(element, "autoscan", true); String classpathRoot = XMLParserUtil.getElementStringValue(element, "classpathRoot"); String mappings = XMLParserUtil.getElementStringValue(element, "mappings"); String aliases = XMLParserUtil.getElementStringValue(element, "aliases"); String templates = XMLParserUtil.getElementStringValue(element, "templates"); String taskExecutor = XMLParserUtil.getElementStringValue(element, "taskExecutor"); String async = XMLParserUtil.getElementStringValue(element, "async"); // Checking bean definition boolean isClientNode = (node != null && node.length() > 0); boolean isEsNodesEmpty = (esNodes == null || esNodes.length() == 0); if (isClientNode && !isEsNodesEmpty) { throw new RuntimeException("Incorrect settings. You should not set esNodes when using a client node."); } if (!isClientNode && isEsNodesEmpty) { throw new RuntimeException( "Incorrect settings. You must set esNodes when creating a transport client."); } BeanDefinition client; GenericBeanDefinition bdef = new GenericBeanDefinition(); if (isClientNode) { bdef.setBeanClass(ElasticsearchClientFactoryBean.class); BeanDefinitionBuilder clientBuilder = startClientBuilder(ElasticsearchClientFactoryBean.class, settingsFile, properties, forceMapping, forceTemplate, mergeMapping, mergeSettings, autoscan, classpathRoot, mappings, aliases, templates, async, taskExecutor); client = ClientBeanDefinitionParser.buildClientDef(clientBuilder, node); } else { bdef.setBeanClass(ElasticsearchTransportClientFactoryBean.class); BeanDefinitionBuilder clientBuilder = startClientBuilder(ElasticsearchTransportClientFactoryBean.class, settingsFile, properties, forceMapping, forceTemplate, mergeMapping, mergeSettings, autoscan, classpathRoot, mappings, aliases, templates, async, taskExecutor); client = ClientBeanDefinitionParser.buildTransportClientDef(clientBuilder, esNodes); } // Register NodeBeanDefinition if (id != null && id.length() > 0) { parserContext.getRegistry().registerBeanDefinition(id, client); } else { parserContext.getRegistry().registerBeanDefinition(name, client); } return bdef; } /** * Managing common properties for NodeClient and TransportClient * * * @param beanClass * @param settingsFile * @param properties * @param forceMapping * @param forceTemplate * @param mergeMapping * @param mergeSettings * @param autoscan * @param classpathRoot * @param mappings * @param aliases * @param templates * @return */ public static BeanDefinitionBuilder startClientBuilder(Class beanClass, String settingsFile, String properties, boolean forceMapping, boolean forceTemplate, boolean mergeMapping, boolean mergeSettings, boolean autoscan, String classpathRoot, String mappings, String aliases, String templates, String async, String taskExecutor) { BeanDefinitionBuilder nodeFactory = BeanDefinitionBuilder.rootBeanDefinition(beanClass); if (settingsFile != null && settingsFile.length() > 0) { logger.warn( "settingsFile is deprecated. Use properties attribute instead. See issue #15: https://github.com/dadoonet/spring-elasticsearch/issues/15."); nodeFactory.addPropertyValue("settingsFile", settingsFile); } if (properties != null && properties.length() > 0) { nodeFactory.addPropertyReference("properties", properties); } nodeFactory.addPropertyValue("forceMapping", forceMapping); nodeFactory.addPropertyValue("forceTemplate", forceTemplate); nodeFactory.addPropertyValue("mergeMapping", mergeMapping); nodeFactory.addPropertyValue("mergeSettings", mergeSettings); nodeFactory.addPropertyValue("autoscan", autoscan); if (classpathRoot != null && classpathRoot.length() > 0) { nodeFactory.addPropertyValue("classpathRoot", classpathRoot); } if (mappings != null && mappings.length() > 0) { nodeFactory.addPropertyValue("mappings", mappings); } if (aliases != null && aliases.length() > 0) { nodeFactory.addPropertyValue("aliases", aliases); } if (templates != null && templates.length() > 0) { nodeFactory.addPropertyValue("templates", templates); } if (async != null && async.length() > 0) { nodeFactory.addPropertyValue("async", async); } if (taskExecutor != null && taskExecutor.length() > 0) { nodeFactory.addPropertyReference("taskExecutor", taskExecutor); } return nodeFactory; } public static BeanDefinition buildClientDef(BeanDefinitionBuilder nodeFactory, String node) { if (node != null && node.length() > 0) { nodeFactory.addPropertyReference("node", node); } return nodeFactory.getBeanDefinition(); } public static BeanDefinition buildTransportClientDef(BeanDefinitionBuilder nodeFactory, String esNodes) { if (esNodes != null && esNodes.length() > 0) { nodeFactory.addPropertyValue("esNodes", esNodes); } return nodeFactory.getBeanDefinition(); } }