com.jn.sqlhelper.cli.commands.JdbcConnectionCommands Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of sqlhelper-cli Show documentation
There is a newer version: 5.0.7
package com.jn.sqlhelper.cli.commands;

import com.jn.langx.configuration.file.directoryfile.DirectoryBasedFileConfigurationRepository;
import com.jn.langx.util.Preconditions;
import com.jn.langx.util.Strings;
import com.jn.langx.util.collection.Collects;
import com.jn.langx.util.collection.StringMap;
import com.jn.langx.util.function.Consumer2;
import com.jn.sqlhelper.common.connection.NamedConnectionConfiguration;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.shell.standard.ShellComponent;
import org.springframework.shell.standard.ShellMethod;
import org.springframework.shell.standard.ShellOption;

import java.util.List;
import java.util.Properties;

@ShellComponent
public class JdbcConnectionCommands {

    @Autowired
    DirectoryBasedFileConfigurationRepository repository;

    @ShellMethod(key = "jdbc list", value = "list all connection configuration names")
    public List getConnectionNames() {
        return Collects.asList(repository.getAll().keySet());
    }

    @ShellMethod(key = "jdbc create", value = "create an new connection configuration")
    public NamedConnectionConfiguration addConnection(
            @ShellOption() String name,
            @ShellOption() String driver,
            @ShellOption() String url,
            @ShellOption String username,
            @ShellOption(defaultValue = "") String password,
            @ShellOption(defaultValue = "") String catalog,
            @ShellOption(defaultValue = "") String schema,
            @ShellOption(defaultValue = "") String props) {

        Preconditions.checkTrue(Strings.isNotEmpty(name));
        Preconditions.checkTrue(Strings.isNotEmpty(driver));
        Preconditions.checkTrue(Strings.isNotEmpty(url));
        Preconditions.checkTrue(Strings.isNotEmpty(username));


        StringMap propsMap = new StringMap(props, "=", "&");
        final Properties properties = new Properties();
        Collects.forEach(propsMap, new Consumer2() {
            @Override
            public void accept(String key, String value) {
                properties.setProperty(key, value);
            }
        });

        NamedConnectionConfiguration configuration = new NamedConnectionConfiguration();
        configuration.setName(name);
        configuration.setDriver(driver);
        configuration.setUrl(url);
        configuration.setUser(username);
        configuration.setPassword(password);
        configuration.setDriverProps(properties);

        properties.setProperty("user", username);
        properties.setProperty("password", password);

        repository.add(configuration);
        return configuration;
    }

    @ShellMethod(key = "jdbc get", value = "get a connection configuration by name")
    public NamedConnectionConfiguration getConnection(String name) {
        return repository.getById(name);
    }

    @ShellMethod(key = "jdbc remove", value = "remove a connection configuration by name")
    public boolean removeConnection(String name) {
        repository.removeById(name);
        return true;
    }


}