fr.putnami.pwt.plugin.spring.rpc.server.service.CommandServiceImpl.java Source code

Introduction

Here is the source code for fr.putnami.pwt.plugin.spring.rpc.server.service.CommandServiceImpl.java

Source

/**
 * This file is part of pwt.
 *
 * pwt is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser
 * General Public License as published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * pwt is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the
 * implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
 * General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along with pwt. If not,
 * see <http://www.gnu.org/licenses/>.
 */
package fr.putnami.pwt.plugin.spring.rpc.server.service;

import com.google.common.collect.Lists;

import org.springframework.aop.support.AopUtils;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.ApplicationContext;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.stereotype.Service;

import java.util.List;

import javax.annotation.PostConstruct;

import fr.putnami.pwt.core.service.server.service.CommandExecutor;
import fr.putnami.pwt.core.service.server.service.CommandExecutorRegistry;
import fr.putnami.pwt.core.service.server.service.CommandExecutorRegistryImpl;
import fr.putnami.pwt.core.service.shared.domain.CommandRequest;
import fr.putnami.pwt.core.service.shared.domain.CommandResponse;
import fr.putnami.pwt.core.service.shared.service.CommandService;

public class CommandServiceImpl implements CommandService, BeanPostProcessor {

    @Autowired
    private ApplicationContext applicationContext;

    private final CommandExecutorRegistry executorRegistry = new CommandExecutorRegistryImpl();

    @PostConstruct
    public void afterPropertySet() {
        for (String beanName : this.applicationContext.getBeanDefinitionNames()) {
            this.scanBean(this.applicationContext.getBean(beanName), beanName);
        }
    }

    @Override
    public List<CommandResponse> executeCommands(List<CommandRequest> commands) {
        List<CommandResponse> result = Lists.newArrayList();
        for (CommandRequest request : commands) {
            CommandExecutor executor = this.executorRegistry.resolveCommandExecutor(request.getCommandDefinition());
            result.add(executor.executeCommand(request));
        }
        return result;
    }

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        this.scanBean(bean, beanName);
        return bean;
    }

    protected void injectService(Class<?> serviceInterface, Object service) {
        this.executorRegistry.injectService(serviceInterface, service);
    }

    private void scanBean(Object bean, String name) {
        Class<?> implClass = bean.getClass();
        if (AopUtils.isAopProxy(bean)) {
            implClass = AopUtils.getTargetClass(bean);
        }
        Service serviceAnnotation = AnnotationUtils.findAnnotation(implClass, Service.class);
        if (serviceAnnotation != null) {
            for (Class<?> inter : implClass.getInterfaces()) {
                this.injectService(inter, bean);
            }
        }
    }

}