| /* |
| * Copyright 1999-2015 dangdang.com. |
| * <p> |
| * 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. |
| * </p> |
| */ |
| |
| package io.elasticjob.cloud.util.config; |
| |
| import com.google.common.base.Strings; |
| import io.elasticjob.cloud.exception.JobConfigurationException; |
| import lombok.AllArgsConstructor; |
| import lombok.Getter; |
| |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| /** |
| * 分片序列号个性化参数. |
| * |
| * @author zhangliang |
| */ |
| @Getter |
| public final class ShardingItemParameters { |
| |
| private static final String PARAMETER_DELIMITER = ","; |
| |
| private static final String KEY_VALUE_DELIMITER = "="; |
| |
| private final Map<Integer, String> map; |
| |
| public ShardingItemParameters(final String shardingItemParameters) { |
| map = toMap(shardingItemParameters); |
| } |
| |
| private Map<Integer, String> toMap(final String originalShardingItemParameters) { |
| if (Strings.isNullOrEmpty(originalShardingItemParameters)) { |
| return Collections.emptyMap(); |
| } |
| String[] shardingItemParameters = originalShardingItemParameters.split(PARAMETER_DELIMITER); |
| Map<Integer, String> result = new HashMap<>(shardingItemParameters.length); |
| for (String each : shardingItemParameters) { |
| ShardingItem shardingItem = parse(each, originalShardingItemParameters); |
| result.put(shardingItem.item, shardingItem.parameter); |
| } |
| return result; |
| } |
| |
| private ShardingItem parse(final String shardingItemParameter, final String originalShardingItemParameters) { |
| String[] pair = shardingItemParameter.trim().split(KEY_VALUE_DELIMITER); |
| if (2 != pair.length) { |
| throw new JobConfigurationException("Sharding item parameters '%s' format error, should be int=xx,int=xx", originalShardingItemParameters); |
| } |
| try { |
| return new ShardingItem(Integer.parseInt(pair[0].trim()), pair[1].trim()); |
| } catch (final NumberFormatException ex) { |
| throw new JobConfigurationException("Sharding item parameters key '%s' is not an integer.", pair[0]); |
| } |
| } |
| |
| /** |
| * 分片项. |
| */ |
| @AllArgsConstructor |
| private static final class ShardingItem { |
| |
| private final int item; |
| |
| private final String parameter; |
| } |
| } |