com.payu.ratel.config.beans.InMemoryRegistryBeanProviderFactory.java Source code

Introduction

Here is the source code for com.payu.ratel.config.beans.InMemoryRegistryBeanProviderFactory.java

Source

/*
 * Copyright 2015 PayU
 *
 * 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.payu.ratel.config.beans;

import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.core.env.Environment;
import org.springframework.scheduling.TaskScheduler;
import org.springframework.scheduling.concurrent.ThreadPoolTaskScheduler;

import com.payu.ratel.client.ClientProxyGenerator;
import com.payu.ratel.client.inmemory.DiscoveryClient;
import com.payu.ratel.client.inmemory.RatelServerFetcher;
import com.payu.ratel.client.inmemory.RatelServerProxyGenerator;
import com.payu.ratel.register.inmemory.RatelServerRegistry;
import com.payu.ratel.register.inmemory.RemoteRestDiscoveryServer;

public class InMemoryRegistryBeanProviderFactory implements RegistryStrategiesProvider, InitializingBean {

    private static final String DEFAULT_DISCOVERY_URL = "http://localhost:8090/server/discovery";

    private final ConfigurableListableBeanFactory beanFactory;

    private RatelServerRegistry ratelServerRegistry;
    private ClientProxyGenerator ratelServerProxyGenerator;
    private RatelServerFetcher ratelServerFetcher;

    public InMemoryRegistryBeanProviderFactory(ConfigurableListableBeanFactory beanFactory) {
        this.beanFactory = beanFactory;
    }

    @Override
    public void afterPropertiesSet() {
        final Environment env = beanFactory.getBean(Environment.class);
        TaskScheduler taskScheduler = new ThreadPoolTaskScheduler();
        final String schedulerBeanName = taskScheduler.getClass().getName();
        beanFactory.registerSingleton(schedulerBeanName, taskScheduler);
        beanFactory.initializeBean(taskScheduler, schedulerBeanName);

        final String inMemoryServerAddress = getRatelServiceAddress(env);
        final DiscoveryClient discoveryClient = new DiscoveryClient(inMemoryServerAddress);
        this.ratelServerFetcher = new RatelServerFetcher(discoveryClient);

        this.ratelServerProxyGenerator = new RatelServerProxyGenerator(beanFactory);

        this.ratelServerRegistry = new RatelServerRegistry(new RemoteRestDiscoveryServer(inMemoryServerAddress),
                taskScheduler);
    }

    private String getRatelServiceAddress(final Environment env) {
        final String inMemoryServerAddress = env.getProperty(RegistryBeanProviderFactory.SERVICE_DISCOVERY_ADDRESS,
                DEFAULT_DISCOVERY_URL);
        return inMemoryServerAddress;
    }

    public RatelServerRegistry getRegisterStrategy() {
        return ratelServerRegistry;
    }

    public ClientProxyGenerator getClientProxyGenerator() {
        return ratelServerProxyGenerator;
    }

    public RatelServerFetcher getFetchStrategy() {
        return ratelServerFetcher;
    }

    @SuppressWarnings("PMD.SignatureDeclareThrowsException")
    @Override
    public void destroy() throws Exception {
        //Not supported for this strategy
    }
}