db.sql.api.impl.cmd.executor.DbSelector Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mybatis-mp-sql-api-impl Show documentation

There is a newer version: 1.7.6-RC2

Show newest version

package db.sql.api.impl.cmd.executor;

import db.sql.api.DbType;
import db.sql.api.cmd.executor.Runnable;
import db.sql.api.impl.tookit.Objects;

import java.util.HashMap;
import java.util.Map;

public class DbSelector implements Selector {

    private final Map consumers = new HashMap<>();

    private Runnable otherwise;

    public DbSelector when(DbType dbType, Runnable runnable) {
        consumers.put(dbType, runnable);
        return this;
    }

    public DbSelector when(DbType[] dbTypes, Runnable runnable) {
        for (DbType dbType : dbTypes) {
            consumers.put(dbType, runnable);
        }
        return this;
    }

    public DbSelector otherwise(Runnable runnable) {
        if (Objects.nonNull(this.otherwise)) {
            throw new RuntimeException("The method of 'otherwise' has already called");
        }
        this.otherwise = runnable;
        return this;
    }

    public DbSelector otherwise() {
        return this.otherwise(() -> {
        });
    }

    public void dbExecute(DbType dbType) {
        Runnable runnable = consumers.get(dbType);
        if (Objects.nonNull(runnable)) {
            runnable.run();
            return;
        }
        if (Objects.nonNull(this.otherwise)) {
            this.otherwise.run();
            return;
        }
        throw new RuntimeException("Not adapted to DbType " + dbType);
    }
}