1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one or more
3 * contributor license agreements. See the NOTICE file distributed with
4 * this work for additional information regarding copyright ownership.
5 * The ASF licenses this file to You under the Apache License, Version 2.0
6 * (the "License"); you may not use this file except in compliance with
7 * the License. You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 */
17
18 package org.apache.shardingsphere.sharding.yaml.swapper;
19
20 import lombok.AccessLevel;
21 import lombok.NoArgsConstructor;
22 import org.apache.shardingsphere.infra.yaml.config.pojo.rule.YamlRuleConfiguration;
23 import org.apache.shardingsphere.sharding.api.config.ShardingRuleConfiguration;
24 import org.apache.shardingsphere.sharding.yaml.config.YamlShardingRuleConfiguration;
25
26 import java.util.Collection;
27 import java.util.Optional;
28
29 /**
30 * Sharding rule configuration converter.
31 */
32 // TODO Move to pipeline module
33 @NoArgsConstructor(access = AccessLevel.PRIVATE)
34 public final class ShardingRuleConfigurationConverter {
35
36 /**
37 * Find and convert sharding rule configuration from YAML.
38 *
39 * @param yamlRuleConfigs YAML rule configurations
40 * @return sharding rule configuration
41 * @throws IllegalStateException if there is no available sharding rule
42 */
43 public static Optional<ShardingRuleConfiguration> findAndConvertShardingRuleConfiguration(final Collection<YamlRuleConfiguration> yamlRuleConfigs) {
44 return findYamlShardingRuleConfiguration(yamlRuleConfigs).map(each -> new YamlShardingRuleConfigurationSwapper().swapToObject(each));
45 }
46
47 /**
48 * Find YAML sharding rule configuration.
49 *
50 * @param yamlRuleConfigs YAML rule configurations
51 * @return YAML sharding rule configuration
52 * @throws IllegalStateException if there is no available sharding rule
53 */
54 public static Optional<YamlShardingRuleConfiguration> findYamlShardingRuleConfiguration(final Collection<YamlRuleConfiguration> yamlRuleConfigs) {
55 return yamlRuleConfigs.stream().filter(YamlShardingRuleConfiguration.class::isInstance).findFirst().map(YamlShardingRuleConfiguration.class::cast);
56 }
57 }