| /* |
| * 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.elasticjob.cloud.context; |
| |
| import com.google.common.base.Function; |
| import com.google.common.base.Joiner; |
| import com.google.common.base.Preconditions; |
| import com.google.common.base.Splitter; |
| import com.google.common.collect.Lists; |
| import lombok.EqualsAndHashCode; |
| import lombok.Getter; |
| import lombok.RequiredArgsConstructor; |
| import lombok.Setter; |
| import lombok.ToString; |
| |
| import java.util.Collections; |
| import java.util.List; |
| import java.util.UUID; |
| |
| /** |
| * Task runtime context. |
| */ |
| @Getter |
| @EqualsAndHashCode(of = "id") |
| @ToString(of = "id") |
| public final class TaskContext { |
| |
| private static final String DELIMITER = "@-@"; |
| |
| private static final String UNASSIGNED_SLAVE_ID = "unassigned-slave"; |
| |
| private String id; |
| |
| private final MetaInfo metaInfo; |
| |
| private final ExecutionType type; |
| |
| private String slaveId; |
| |
| @Setter |
| private boolean idle; |
| |
| public TaskContext(final String jobName, final List<Integer> shardingItem, final ExecutionType type) { |
| this(jobName, shardingItem, type, UNASSIGNED_SLAVE_ID); |
| } |
| |
| public TaskContext(final String jobName, final List<Integer> shardingItem, final ExecutionType type, final String slaveId) { |
| metaInfo = new MetaInfo(jobName, shardingItem); |
| this.type = type; |
| this.slaveId = slaveId; |
| id = Joiner.on(DELIMITER).join(metaInfo, type, slaveId, UUID.randomUUID().toString()); |
| } |
| |
| private TaskContext(final String id, final MetaInfo metaInfo, final ExecutionType type, final String slaveId) { |
| this.id = id; |
| this.metaInfo = metaInfo; |
| this.type = type; |
| this.slaveId = slaveId; |
| } |
| |
| /** |
| * Get task context via task ID. |
| * |
| * @param id task ID |
| * @return task context |
| */ |
| public static TaskContext from(final String id) { |
| String[] result = id.split(DELIMITER); |
| Preconditions.checkState(5 == result.length); |
| return new TaskContext(id, MetaInfo.from(result[0] + DELIMITER + result[1]), ExecutionType.valueOf(result[2]), result[3]); |
| } |
| |
| /** |
| * Get unassigned task ID before job execute. |
| * |
| * @param id task ID |
| * @return unassigned task ID before job execute |
| */ |
| public static String getIdForUnassignedSlave(final String id) { |
| return id.replaceAll(TaskContext.from(id).getSlaveId(), UNASSIGNED_SLAVE_ID); |
| } |
| |
| /** |
| * Set job server ID. |
| * |
| * @param slaveId job server ID |
| */ |
| public void setSlaveId(final String slaveId) { |
| id = id.replaceAll(this.slaveId, slaveId); |
| this.slaveId = slaveId; |
| } |
| |
| /** |
| * Get task name. |
| * |
| * @return task name |
| */ |
| public String getTaskName() { |
| return Joiner.on(DELIMITER).join(metaInfo, type, slaveId); |
| } |
| |
| /** |
| * Get executor ID. |
| * |
| * @param appName application name |
| * @return executor ID |
| */ |
| public String getExecutorId(final String appName) { |
| return Joiner.on(DELIMITER).join(appName, slaveId); |
| } |
| |
| /** |
| * Task meta data. |
| */ |
| @RequiredArgsConstructor |
| @Getter |
| @EqualsAndHashCode |
| public static class MetaInfo { |
| |
| private final String jobName; |
| |
| private final List<Integer> shardingItems; |
| |
| /** |
| * Get task meta data info via string. |
| * |
| * @param value task meta data info string |
| * @return task meta data info |
| */ |
| public static MetaInfo from(final String value) { |
| String[] result = value.split(DELIMITER); |
| Preconditions.checkState(1 == result.length || 2 == result.length || 5 == result.length); |
| return new MetaInfo(result[0], 1 == result.length || "".equals(result[1]) ? Collections.<Integer>emptyList() : Lists.transform(Splitter.on(",").splitToList(result[1]), |
| new Function<String, Integer>() { |
| |
| @Override |
| public Integer apply(final String input) { |
| return Integer.parseInt(input); |
| } |
| })); |
| } |
| |
| @Override |
| public String toString() { |
| return Joiner.on(DELIMITER).join(jobName, Joiner.on(",").join(shardingItems)); |
| } |
| } |
| } |