com.fluxtion.server.plugin.jdbc.impl.JdbcConnectionLoaderService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of serverplugin-jdbc Show documentation

There is a newer version: 0.1.19

/*
 * SPDX-FileCopyrightText: © 2024 Gregory Higgins 
 * SPDX-License-Identifier: AGPL-3.0-only
 */

package com.fluxtion.server.plugin.jdbc.impl;

import com.fluxtion.runtime.lifecycle.Lifecycle;
import com.fluxtion.server.plugin.jdbc.JdbcConnectionLoader;
import lombok.Data;
import lombok.extern.log4j.Log4j2;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

@Data
@Log4j2
public class JdbcConnectionLoaderService implements Lifecycle, JdbcConnectionLoader {

    private Map connections = new HashMap<>();
    private boolean testConnection = false;
    private boolean fastFail = false;

    @Override
    public void init() {
        log.info("registered connections {}", connections);
    }

    @Override
    public void start() {
        if (testConnection) {
            log.info("starting connection test {}", connections);
            for (String name : connections.keySet()) {
                try {
                    log.info("testing connection {}", name);
                    getConnection(name);
                } catch (SQLException e) {
                    log.info("failed to connect to {}", name, e);
                    if (fastFail) {
                        throw new RuntimeException("failed to connect to " + name, e);
                    }
                }
            }
        }
    }

    @Override
    public Connection getConnection(String name) throws SQLException {

        Connection conn = null;
        if (connections.containsKey(name)) {
            JdbcConnectionConfig jdbcConnectionConfig = connections.get(name);
            var connUrl = jdbcConnectionConfig.getUrl();
            try {
                conn = DriverManager.getConnection(
                        connUrl,
                        jdbcConnectionConfig.getUsername(),
                        jdbcConnectionConfig.getPassword());
                log.info("Connected to {}", jdbcConnectionConfig);
            } catch (SQLException e) {
                log.error("failed to get jdbc connection:{}", jdbcConnectionConfig, e);
            }
        } else {
            log.warn("no connection registered with name:{}", name);
        }
        return conn;
    }

    @Override
    public void tearDown() {

    }
}