All Downloads are FREE. Search and download functionalities are using the official Maven repository.

tech.guyi.ipojo.module.h2.executor.JdbcExecutor Maven / Gradle / Ivy

Go to download

The newest version!
package tech.guyi.ipojo.module.h2.executor;

import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.dbutils.ResultSetHandler;
import org.h2.jdbcx.JdbcDataSource;
import org.osgi.framework.BundleContext;
import org.osgi.service.log.Logger;
import tech.guyi.ipojo.application.ApplicationContext;
import tech.guyi.ipojo.application.annotation.Component;
import tech.guyi.ipojo.application.annotation.Resource;
import tech.guyi.ipojo.application.bean.interfaces.ApplicationStartSuccessEvent;
import tech.guyi.ipojo.application.bean.interfaces.InitializingBean;
import tech.guyi.ipojo.application.osgi.log.Log;
import tech.guyi.ipojo.module.h2.datasource.JdbcDataSourceProvider;
import tech.guyi.ipojo.module.stream.Mono;
import tech.guyi.ipojo.module.stream.producer.Producer;
import tech.guyi.ipojo.module.stream.publisher.Publisher;
import tech.guyi.ipojo.module.stream.subscriber.Subscriber;

import java.sql.SQLException;
import java.util.LinkedList;
import java.util.List;

@Component
public class JdbcExecutor implements InitializingBean, ApplicationStartSuccessEvent {

    @Resource
    private JdbcDataSourceProvider provider;

    @Log
    private Logger logger;
    public void setLogger(Logger logger) {
        this.logger = logger;
    }

    private JdbcDataSource dataSource;
    private QueryRunner runner;

    private Mono readyMono;
    private Publisher _publisher;

    public boolean isReady(){
        return this.runner != null;
    }

    public void await(Subscriber subscriber){
        if (this.readyMono == null || this.isReady()){
            subscriber.subscription(true);
        }else{
            this.readyMono.subscription(subscriber);
        }
    }

    @Override
    public void afterPropertiesSet() {
        this.readyMono = Mono.create(new Producer() {
            @Override
            public void produce(Publisher publisher) {
                _publisher = publisher;
            }
        });
        this.readyMono.open();
    }

    private List filterNull(List args){
        List result = new LinkedList<>();
        for (Object arg : args) {
            if (arg != null){
                result.add(arg);
            }
        }
        return result;
    }

    public synchronized  T execute(JdbcInvoker invoker) throws SQLException {
        return invoker.invoke(this.runner);
    }

    public int update(final String sql, final List args) throws SQLException {
        return this.execute(new JdbcInvoker(){
            @Override
            public Integer invoke(QueryRunner runner) throws SQLException {
                List parameters = filterNull(args);
                logger.debug("execute sql [{}] Parameters {}",sql,parameters);
                return runner.update(sql,parameters.toArray());
            }
        });
    }

    public  T query(final String sql, final ResultSetHandler handler, final List args) throws SQLException {
        return this.execute(new JdbcInvoker() {
            @Override
            public T invoke(QueryRunner runner) throws SQLException {
                List parameters = filterNull(args);
                logger.debug("execute sql [{}] Parameters {}",sql,parameters);
                return runner.query(sql,handler,parameters.toArray());
            }
        });
    }

    @Override
    public void onStartSuccess(ApplicationContext applicationContext, BundleContext bundleContext) throws Exception {
        Mono.create(new Producer() {
            @Override
            public void produce(Publisher publisher) {
                provider.provide(publisher);
            }
        }).subscription(new Subscriber() {
            @Override
            public void subscription(JdbcDataSource value) {
                dataSource = value;
                runner = new QueryRunner(dataSource);
                _publisher.publish(true);
                readyMono = null;
                _publisher = null;
            }
        }).open();
    }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy