com.landawn.abacus.dataSource.SimpleSourceSelector Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of abacus-android Show documentation

A general and simple library for Android

There is a newer version: 1.10.1

/*
 * Copyright (c) 2015, Haiyang Li. All rights reserved.
 */

package com.landawn.abacus.dataSource;

import java.util.List;
import java.util.Map;

import com.landawn.abacus.DataSource;
import com.landawn.abacus.DataSourceManager;
import com.landawn.abacus.DataSourceSelector;

/**
 * 
 * @since 0.8
 * 
 * @author Haiyang Li
 */
public class SimpleSourceSelector implements DataSourceSelector {
    private static final String QUERY_WITH_DATA_SOURCE = "queryWithDataSource";

    @Override
    public DataSource select(DataSourceManager dataSourceManager, String entityName, String sql, Object[] parameters, Map options) {
        return getDataSource(dataSourceManager, options);
    }

    @Override
    public DataSource select(DataSourceManager dataSourceManager, String entityName, String sql, List parameters, Map options) {
        return getDataSource(dataSourceManager, options);
    }

    private DataSource getDataSource(DataSourceManager dataSourceManager, Map options) {
        String dataSourceName = (options == null) ? null : (String) options.get(QUERY_WITH_DATA_SOURCE);

        DataSource ds = (dataSourceName == null) ? dataSourceManager.getPrimaryDataSource() : dataSourceManager.getActiveDataSources().get(dataSourceName);

        if (ds == null) {
            throw new IllegalArgumentException("No data source is available with name: " + dataSourceName);
        }

        return ds;
    }
}