io.edurt.datacap.plugin.jdbc.db2.Db2Plugin Maven / Gradle / Ivy

Go to download
package io.edurt.datacap.plugin.jdbc.db2;

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import io.edurt.datacap.spi.Plugin;
import io.edurt.datacap.spi.PluginType;
import io.edurt.datacap.spi.adapter.JdbcAdapter;
import io.edurt.datacap.spi.connection.JdbcConfigure;
import io.edurt.datacap.spi.connection.JdbcConnection;
import io.edurt.datacap.spi.model.Configure;
import io.edurt.datacap.spi.model.Response;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.beanutils.BeanUtils;
import org.apache.commons.lang3.ObjectUtils;

@Slf4j
@SuppressFBWarnings(value = {"EI_EXPOSE_REP"})
public class Db2Plugin
        implements Plugin
{
    private JdbcConfigure jdbcConfigure;
    private JdbcConnection connection;
    private Response response;

    @Override
    public String validator()
    {
        return "SELECT\n" +
                "  replace(SERVICE_LEVEL, 'DB2 v', '') AS version\n" +
                "FROM\n" +
                "  SYSIBMADM.ENV_INST_INFO";
    }

    @Override
    public String name()
    {
        return "DB2";
    }

    @Override
    public String description()
    {
        return "Integrate DB2 data sources";
    }

    @Override
    public PluginType type()
    {
        return PluginType.JDBC;
    }

    @Override
    public void connect(Configure configure)
    {
        try {
            this.response = new Response();
            this.jdbcConfigure = new JdbcConfigure();
            BeanUtils.copyProperties(this.jdbcConfigure, configure);
            this.jdbcConfigure.setJdbcDriver("com.ibm.db2.jcc.DB2Driver");
            this.jdbcConfigure.setJdbcType("db2");
            this.connection = new JdbcConnection(this.jdbcConfigure, this.response);
        }
        catch (Exception ex) {
            this.response.setIsConnected(Boolean.FALSE);
            this.response.setMessage(ex.getMessage());
        }
    }

    @Override
    public Response execute(String content)
    {
        if (ObjectUtils.isNotEmpty(this.connection)) {
            log.info("Execute db2 plugin logic started");
            this.response = this.connection.getResponse();
            JdbcAdapter processor = new Db2Adapter(this.connection);
            this.response = processor.handlerExecute(content);
            log.info("Execute db2 plugin logic end");
        }
        return this.response;
    }

    @Override
    public void destroy()
    {
        if (ObjectUtils.isNotEmpty(this.connection)) {
            this.connection.destroy();
        }
    }
}