com.google.enterprise.connector.db.MockDBConnectorFactory.java Source code

Introduction

Here is the source code for com.google.enterprise.connector.db.MockDBConnectorFactory.java

Source

// Copyright 2011 Google Inc.
//
// 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.

package com.google.enterprise.connector.db;

import com.google.enterprise.connector.spi.Connector;
import com.google.enterprise.connector.spi.ConnectorFactory;
import com.google.enterprise.connector.spi.RepositoryException;
import com.google.enterprise.connector.util.diffing.DiffingConnector;

import junit.framework.Assert;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.PropertyPlaceholderConfigurer;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.beans.factory.xml.XmlBeanDefinitionReader;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;

import java.util.Arrays;
import java.util.Map;
import java.util.Properties;

public class MockDBConnectorFactory implements ConnectorFactory {
    public MockDBConnectorFactory() {
    }

    /**
     * Creates a database connector.
     *
     * @param config map of configuration values.
     */
    /* TODO(jlacey): Extract the Spring instantiation code in CM. */
    @Override
    public Connector makeConnector(Map<String, String> config) throws RepositoryException {
        // TODO(jlacey): The placeholder values are in the EPPC bean in
        // connectorDefaults.xml, but we're not loading that, and doing so
        // would unravel a ball of string: using setLocation instead of
        // setProperties (since the EPPC bean already has properties),
        // which in turn requires the ByteArrayResource machinery in
        // InstanceInfo or writing the properties to a file.
        Properties props = new Properties();
        for (String configKey : DBConnectorType.CONFIG_KEYS) {
            props.put(configKey, "");
        }
        // Escape MyBatis syntax that looks like a Spring placeholder.
        // See https://jira.springsource.org/browse/SPR-4953
        props.put("docIds", "#{'$'}{docIds}");
        props.putAll(config);

        Resource prototype = new ClassPathResource("config/connectorInstance.xml", MockDBConnectorFactory.class);
        Resource defaults = new ClassPathResource("config/connectorDefaults.xml", MockDBConnectorFactory.class);

        DefaultListableBeanFactory factory = new DefaultListableBeanFactory();
        XmlBeanDefinitionReader beanReader = new XmlBeanDefinitionReader(factory);
        try {
            beanReader.loadBeanDefinitions(prototype);
            beanReader.loadBeanDefinitions(defaults);
        } catch (BeansException e) {
            throw new RepositoryException(e);
        }

        PropertyPlaceholderConfigurer cfg = new PropertyPlaceholderConfigurer();
        cfg.setProperties(props);
        cfg.postProcessBeanFactory(factory);

        String[] beanList = factory.getBeanNamesForType(DiffingConnector.class);
        Assert.assertEquals(Arrays.asList(beanList).toString(), 1, beanList.length);
        return (Connector) factory.getBean(beanList[0]);
    }
}