Kirs
4 years ago
committed by
GitHub
16 changed files with 430 additions and 136 deletions
@ -0,0 +1,56 @@
|
||||
/* |
||||
* 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.remote.command; |
||||
|
||||
import org.apache.dolphinscheduler.common.utils.JSONUtils; |
||||
|
||||
import java.io.Serializable; |
||||
import java.util.LinkedHashMap; |
||||
import java.util.Map; |
||||
|
||||
/** |
||||
* command context |
||||
*/ |
||||
public class CommandContext implements Serializable { |
||||
|
||||
private Map<String, String> items = new LinkedHashMap<>(); |
||||
|
||||
public Map<String, String> getItems() { |
||||
return items; |
||||
} |
||||
|
||||
public void setItems(Map<String, String> items) { |
||||
this.items = items; |
||||
} |
||||
|
||||
public void put(String key, String value) { |
||||
items.put(key, value); |
||||
} |
||||
|
||||
public String get(String key) { |
||||
return items.get(key); |
||||
} |
||||
|
||||
public byte[] toBytes() { |
||||
return JSONUtils.toJsonByteArray(this); |
||||
} |
||||
|
||||
public static CommandContext valueOf(byte[] src) { |
||||
return JSONUtils.parseObject(src, CommandContext.class); |
||||
} |
||||
} |
||||
@ -0,0 +1,44 @@
|
||||
/* |
||||
* 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.service.exceptions; |
||||
|
||||
/** |
||||
* task priority queue exception |
||||
*/ |
||||
public class TaskPriorityQueueException extends Exception { |
||||
|
||||
/** |
||||
* Construct a new runtime exception with the detail message |
||||
* |
||||
* @param message message |
||||
*/ |
||||
public TaskPriorityQueueException(String message) { |
||||
super(message); |
||||
} |
||||
|
||||
/** |
||||
* Construct a new runtime exception with the detail message and cause |
||||
* |
||||
* @param message message |
||||
* @param cause cause |
||||
*/ |
||||
public TaskPriorityQueueException(String message, Throwable cause) { |
||||
super(message, cause); |
||||
} |
||||
|
||||
} |
||||
@ -0,0 +1,81 @@
|
||||
/* |
||||
* 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.service.queue; |
||||
|
||||
import java.util.Arrays; |
||||
import java.util.Collections; |
||||
import java.util.List; |
||||
|
||||
import org.junit.Assert; |
||||
import org.junit.Test; |
||||
|
||||
public class TaskPriorityTest { |
||||
|
||||
@Test |
||||
public void testSort() { |
||||
TaskPriority priorityOne = new TaskPriority(1, 0, 0, 0, "default"); |
||||
TaskPriority priorityTwo = new TaskPriority(2, 0, 0, 0, "default"); |
||||
TaskPriority priorityThree = new TaskPriority(3, 0, 0, 0, "default"); |
||||
List<TaskPriority> taskPrioritys = Arrays.asList(priorityOne, priorityThree, priorityTwo); |
||||
Collections.sort(taskPrioritys); |
||||
Assert.assertEquals( |
||||
Arrays.asList(priorityOne, priorityTwo, priorityThree), |
||||
taskPrioritys |
||||
); |
||||
|
||||
priorityOne = new TaskPriority(0, 1, 0, 0, "default"); |
||||
priorityTwo = new TaskPriority(0, 2, 0, 0, "default"); |
||||
priorityThree = new TaskPriority(0, 3, 0, 0, "default"); |
||||
taskPrioritys = Arrays.asList(priorityOne, priorityThree, priorityTwo); |
||||
Collections.sort(taskPrioritys); |
||||
Assert.assertEquals( |
||||
Arrays.asList(priorityOne, priorityTwo, priorityThree), |
||||
taskPrioritys |
||||
); |
||||
|
||||
priorityOne = new TaskPriority(0, 0, 1, 0, "default"); |
||||
priorityTwo = new TaskPriority(0, 0, 2, 0, "default"); |
||||
priorityThree = new TaskPriority(0, 0, 3, 0, "default"); |
||||
taskPrioritys = Arrays.asList(priorityOne, priorityThree, priorityTwo); |
||||
Collections.sort(taskPrioritys); |
||||
Assert.assertEquals( |
||||
Arrays.asList(priorityOne, priorityTwo, priorityThree), |
||||
taskPrioritys |
||||
); |
||||
|
||||
priorityOne = new TaskPriority(0, 0, 0, 1, "default"); |
||||
priorityTwo = new TaskPriority(0, 0, 0, 2, "default"); |
||||
priorityThree = new TaskPriority(0, 0, 0, 3, "default"); |
||||
taskPrioritys = Arrays.asList(priorityOne, priorityThree, priorityTwo); |
||||
Collections.sort(taskPrioritys); |
||||
Assert.assertEquals( |
||||
Arrays.asList(priorityOne, priorityTwo, priorityThree), |
||||
taskPrioritys |
||||
); |
||||
|
||||
priorityOne = new TaskPriority(0, 0, 0, 0, "default_1"); |
||||
priorityTwo = new TaskPriority(0, 0, 0, 0, "default_2"); |
||||
priorityThree = new TaskPriority(0, 0, 0, 0, "default_3"); |
||||
taskPrioritys = Arrays.asList(priorityOne, priorityThree, priorityTwo); |
||||
Collections.sort(taskPrioritys); |
||||
Assert.assertEquals( |
||||
Arrays.asList(priorityOne, priorityTwo, priorityThree), |
||||
taskPrioritys |
||||
); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue