move logic to broadcast-table-example

/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

/*

 * Please make sure master-slave data sync on MySQL is running correctly. Otherwise this example will query empty data from slave.

 */

package org.apache.shardingsphere.example.broadcast.table.raw.jdbc;

import org.apache.shardingsphere.example.broadcast.table.raw.jdbc.factory.DataSourceFactory;

import org.apache.shardingsphere.example.common.jdbc.repository.CountryRepositroyImpl;

import org.apache.shardingsphere.example.common.jdbc.service.CountryServiceImpl;

import org.apache.shardingsphere.example.common.service.CommonService;

import org.apache.shardingsphere.example.type.ShardingType;

import javax.sql.DataSource;

import java.sql.SQLException;

public class JavaConfigurationExample {

    private static ShardingType shardingType = ShardingType.SHARDING_DATABASES;

//    private static ShardingType shardingType = ShardingType.MASTER_SLAVE;

    public static void main(final String[] args) throws SQLException {

        DataSource dataSource = DataSourceFactory.newInstance(shardingType);

        CommonService countryService = getCountryService(dataSource);

        countryService.initEnvironment();

        countryService.processSuccess();

        countryService.cleanEnvironment();

    }

    private static CommonService getCountryService(final DataSource dataSource) {

        return new CountryServiceImpl(new CountryRepositroyImpl(dataSource));

    }

}

/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package org.apache.shardingsphere.example.broadcast.table.raw.jdbc;

import org.apache.shardingsphere.example.broadcast.table.raw.jdbc.factory.YamlDataSourceFactory;

import org.apache.shardingsphere.example.common.jdbc.repository.CountryRepositroyImpl;

import org.apache.shardingsphere.example.common.jdbc.service.CountryServiceImpl;

import org.apache.shardingsphere.example.common.service.CommonService;

import org.apache.shardingsphere.example.type.ShardingType;

import javax.sql.DataSource;

import java.io.IOException;

import java.sql.SQLException;

/*

 * Please make sure master-slave data sync on MySQL is running correctly. Otherwise this example will query empty data from slave.

 */

public class YamlConfigurationExample {

    private static ShardingType shardingType = ShardingType.SHARDING_DATABASES;

//    private static ShardingType shardingType = ShardingType.MASTER_SLAVE;

    public static void main(final String[] args) throws SQLException, IOException {

        DataSource dataSource = YamlDataSourceFactory.newInstance(shardingType);

        CommonService countryService = getCountryService(dataSource);

        countryService.initEnvironment();

        countryService.processSuccess();

        countryService.cleanEnvironment();

    }

    private static CommonService getCountryService(final DataSource dataSource) {

        return new CountryServiceImpl(new CountryRepositroyImpl(dataSource));

    }

}

/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package org.apache.shardingsphere.example.broadcast.table.raw.jdbc.config;

import org.apache.shardingsphere.api.config.masterslave.MasterSlaveRuleConfiguration;

import org.apache.shardingsphere.example.common.DataSourceUtil;

import org.apache.shardingsphere.example.config.ExampleConfiguration;

import org.apache.shardingsphere.shardingjdbc.api.MasterSlaveDataSourceFactory;

import javax.sql.DataSource;

import java.sql.SQLException;

import java.util.Arrays;

import java.util.HashMap;

import java.util.Map;

import java.util.Properties;

public class MasterSlaveConfiguration implements ExampleConfiguration {

    @Override

    public DataSource getDataSource() throws SQLException {

        MasterSlaveRuleConfiguration masterSlaveRuleConfig = new MasterSlaveRuleConfiguration("demo_ds_master_slave", "demo_ds_master", Arrays.asList("demo_ds_slave_0", "demo_ds_slave_1"));

        return MasterSlaveDataSourceFactory.createDataSource(createDataSourceMap(), masterSlaveRuleConfig, new Properties());

    }

    private Map<String, DataSource> createDataSourceMap() {

        Map<String, DataSource> result = new HashMap<>();

        result.put("demo_ds_master", DataSourceUtil.createDataSource("demo_ds_master"));

        result.put("demo_ds_slave_0", DataSourceUtil.createDataSource("demo_ds_slave_0"));

        result.put("demo_ds_slave_1", DataSourceUtil.createDataSource("demo_ds_slave_1"));

        return result;

    }

}

/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package org.apache.shardingsphere.example.broadcast.table.raw.jdbc.config;

import org.apache.shardingsphere.api.config.sharding.ShardingRuleConfiguration;

import org.apache.shardingsphere.example.common.DataSourceUtil;

import org.apache.shardingsphere.example.config.ExampleConfiguration;

import org.apache.shardingsphere.shardingjdbc.api.ShardingDataSourceFactory;

import javax.sql.DataSource;

import java.sql.SQLException;

import java.util.HashMap;

import java.util.Map;

import java.util.Properties;

public class ShardingDatabasesConfigurationPrecise implements ExampleConfiguration {

    @Override

    public DataSource getDataSource() throws SQLException {

        ShardingRuleConfiguration shardingRuleConfig = new ShardingRuleConfiguration();

        shardingRuleConfig.getBroadcastTables().add("t_country");

        return ShardingDataSourceFactory.createDataSource(createDataSourceMap(), shardingRuleConfig, new Properties());

    }

    private static Map<String, DataSource> createDataSourceMap() {

        Map<String, DataSource> result = new HashMap<>();

        result.put("demo_ds_0", DataSourceUtil.createDataSource("demo_ds_0"));

        result.put("demo_ds_1", DataSourceUtil.createDataSource("demo_ds_1"));

        return result;

    }

}

/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You under the Apache License, Version 2.0

 * (the "License"); you may not use this file except in compliance with

 * the License.  You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package org.apache.shardingsphere.example.broadcast.table.raw.jdbc.factory;

import org.apache.shardingsphere.example.broadcast.table.raw.jdbc.config.MasterSlaveConfiguration;

import org.apache.shardingsphere.example.broadcast.table.raw.jdbc.config.ShardingDatabasesConfigurationPrecise;

import org.apache.shardingsphere.example.type.ShardingType;

import javax.sql.DataSource;

import java.sql.SQLException;

public class DataSourceFactory {

    public static DataSource newInstance(final ShardingType shardingType) throws SQLException {

        switch (shardingType) {

            case SHARDING_DATABASES:

                return new ShardingDatabasesConfigurationPrecise().getDataSource();

            case MASTER_SLAVE:

                return new MasterSlaveConfiguration().getDataSource();

            default:

                throw new UnsupportedOperationException(shardingType.name());

        }

    }

}