Kirs
3 years ago
committed by
GitHub
52 changed files with 414 additions and 552 deletions
@ -1,56 +0,0 @@ |
|||||||
/* |
|
||||||
* 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.dolphinscheduler.server.worker.cache; |
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.server.entity.TaskExecutionContext; |
|
||||||
|
|
||||||
/** |
|
||||||
* TaskExecutionContextCacheManager |
|
||||||
*/ |
|
||||||
public interface TaskExecutionContextCacheManager { |
|
||||||
|
|
||||||
/** |
|
||||||
* get taskInstance by taskInstance id |
|
||||||
* |
|
||||||
* @param taskInstanceId taskInstanceId |
|
||||||
* @return taskInstance |
|
||||||
*/ |
|
||||||
TaskExecutionContext getByTaskInstanceId(Integer taskInstanceId); |
|
||||||
|
|
||||||
/** |
|
||||||
* cache taskInstance |
|
||||||
* |
|
||||||
* @param taskExecutionContext taskExecutionContext |
|
||||||
*/ |
|
||||||
void cacheTaskExecutionContext(TaskExecutionContext taskExecutionContext); |
|
||||||
|
|
||||||
/** |
|
||||||
* remove taskInstance by taskInstanceId |
|
||||||
* |
|
||||||
* @param taskInstanceId taskInstanceId |
|
||||||
*/ |
|
||||||
void removeByTaskInstanceId(Integer taskInstanceId); |
|
||||||
|
|
||||||
/** |
|
||||||
* If the value for the specified key is present and non-null,then perform the update，otherwise it will return false |
|
||||||
* |
|
||||||
* @param taskExecutionContext taskExecutionContext |
|
||||||
* @return status |
|
||||||
*/ |
|
||||||
boolean updateTaskExecutionContext(TaskExecutionContext taskExecutionContext); |
|
||||||
} |
|
||||||
@ -1,76 +0,0 @@ |
|||||||
/* |
|
||||||
* 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.dolphinscheduler.server.worker.cache.impl; |
|
||||||
|
|
||||||
import org.apache.dolphinscheduler.server.entity.TaskExecutionContext; |
|
||||||
import org.apache.dolphinscheduler.server.worker.cache.TaskExecutionContextCacheManager; |
|
||||||
|
|
||||||
import java.util.Map; |
|
||||||
import java.util.concurrent.ConcurrentHashMap; |
|
||||||
|
|
||||||
import org.springframework.stereotype.Service; |
|
||||||
|
|
||||||
/** |
|
||||||
* TaskExecutionContextCache |
|
||||||
*/ |
|
||||||
@Service |
|
||||||
public class TaskExecutionContextCacheManagerImpl implements TaskExecutionContextCacheManager { |
|
||||||
|
|
||||||
|
|
||||||
/** |
|
||||||
* taskInstance cache |
|
||||||
*/ |
|
||||||
private Map<Integer, TaskExecutionContext> taskExecutionContextCache = new ConcurrentHashMap<>(); |
|
||||||
|
|
||||||
/** |
|
||||||
* get taskInstance by taskInstance id |
|
||||||
* |
|
||||||
* @param taskInstanceId taskInstanceId |
|
||||||
* @return taskInstance |
|
||||||
*/ |
|
||||||
@Override |
|
||||||
public TaskExecutionContext getByTaskInstanceId(Integer taskInstanceId) { |
|
||||||
return taskExecutionContextCache.get(taskInstanceId); |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* cache taskInstance |
|
||||||
* |
|
||||||
* @param taskExecutionContext taskExecutionContext |
|
||||||
*/ |
|
||||||
@Override |
|
||||||
public void cacheTaskExecutionContext(TaskExecutionContext taskExecutionContext) { |
|
||||||
taskExecutionContextCache.put(taskExecutionContext.getTaskInstanceId(), taskExecutionContext); |
|
||||||
} |
|
||||||
|
|
||||||
/** |
|
||||||
* remove taskInstance by taskInstanceId |
|
||||||
* |
|
||||||
* @param taskInstanceId taskInstanceId |
|
||||||
*/ |
|
||||||
@Override |
|
||||||
public void removeByTaskInstanceId(Integer taskInstanceId) { |
|
||||||
taskExecutionContextCache.remove(taskInstanceId); |
|
||||||
} |
|
||||||
|
|
||||||
@Override |
|
||||||
public boolean updateTaskExecutionContext(TaskExecutionContext taskExecutionContext) { |
|
||||||
taskExecutionContextCache.computeIfPresent(taskExecutionContext.getTaskInstanceId(), (k, v) -> taskExecutionContext); |
|
||||||
return taskExecutionContextCache.containsKey(taskExecutionContext.getTaskInstanceId()); |
|
||||||
} |
|
||||||
} |
|
||||||
Loading…
Reference in new issue