/* * Copyright 1999-2015 dangdang.com. *
* Licensed 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 io.shardingjdbc.orchestration.internal; import com.google.common.base.Preconditions; import io.shardingjdbc.core.api.config.MasterSlaveRuleConfiguration; import io.shardingjdbc.core.api.config.ShardingRuleConfiguration; import io.shardingjdbc.core.jdbc.core.datasource.MasterSlaveDataSource; import io.shardingjdbc.core.jdbc.core.datasource.ShardingDataSource; import io.shardingjdbc.orchestration.api.config.OrchestrationConfiguration; import io.shardingjdbc.orchestration.internal.config.ConfigurationService; import io.shardingjdbc.orchestration.internal.listener.ListenerFactory; import io.shardingjdbc.orchestration.internal.state.datasource.DataSourceService; import io.shardingjdbc.orchestration.internal.state.instance.InstanceStateService; import io.shardingjdbc.orchestration.reg.api.RegistryCenter; import io.shardingjdbc.orchestration.reg.api.RegistryCenterConfiguration; import io.shardingjdbc.orchestration.reg.etcd.EtcdConfiguration; import io.shardingjdbc.orchestration.reg.etcd.EtcdRegistryCenter; import io.shardingjdbc.orchestration.reg.zookeeper.ZookeeperConfiguration; import io.shardingjdbc.orchestration.reg.zookeeper.ZookeeperRegistryCenter; import lombok.extern.slf4j.Slf4j; import javax.sql.DataSource; import java.util.LinkedHashMap; import java.util.Map; import java.util.Map.Entry; import java.util.Properties; /** * Orchestration service facade. * * @author zhangliang * @author caohao */ @Slf4j public final class OrchestrationFacade implements AutoCloseable { private final boolean isOverwrite; private final ConfigurationService configService; private final InstanceStateService instanceStateService; private final DataSourceService dataSourceService; private final ListenerFactory listenerManager; private final RegistryCenter regCenter; public OrchestrationFacade(final OrchestrationConfiguration config) { regCenter = createRegistryCenter(config.getRegCenterConfig()); isOverwrite = config.isOverwrite(); configService = new ConfigurationService(config.getName(), regCenter); instanceStateService = new InstanceStateService(config.getName(), regCenter); dataSourceService = new DataSourceService(config.getName(), regCenter); listenerManager = new ListenerFactory(config.getName(), regCenter); } private RegistryCenter createRegistryCenter(final RegistryCenterConfiguration regCenterConfig) { Preconditions.checkNotNull(regCenterConfig, "Registry center configuration cannot be null."); if (regCenterConfig instanceof ZookeeperConfiguration) { return new ZookeeperRegistryCenter((ZookeeperConfiguration) regCenterConfig); } if (regCenterConfig instanceof EtcdConfiguration) { return new EtcdRegistryCenter((EtcdConfiguration) regCenterConfig); } throw new UnsupportedOperationException(regCenterConfig.getClass().getName()); } /** * Initialize for sharding orchestration. * * @param dataSourceMap data source map * @param shardingRuleConfig sharding rule configuration * @param configMap config map * @param props sharding properties * @param shardingDataSource sharding data source */ public void init(final Map