[JBPM-9964] Avoid cast exception when priority is integer

kiegroup · Jan 13, 2022 · 04f4492 · 04f4492
1 parent cad581b
commit 04f4492
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/...an-task-workitems/src/main/java/org/jbpm/services/task/wih/AbstractHTWorkItemHandler.java b/...an-task-workitems/src/main/java/org/jbpm/services/task/wih/AbstractHTWorkItemHandler.java
@@ -128,16 +128,17 @@ protected Task createTaskBasedOnWorkItemParams(KieSession session, WorkItem work
 
         task.setSubject(comment);
 
-        String priorityString = (String) workItem.getParameter("Priority");
-        int priority = 0;
-        if (priorityString != null) {
+        Object priorityObject = workItem.getParameter("Priority");
+
+        if (priorityObject instanceof Number) {
+            task.setPriority(((Number) priorityObject).intValue());
+        } else if (priorityObject != null) {
             try {
-                priority = new Integer(priorityString);
+                task.setPriority(Integer.parseInt(priorityObject.toString()));
             } catch (NumberFormatException e) {
                 // do nothing
             }
         }
-        task.setPriority(priority);
 
         InternalTaskData taskData = (InternalTaskData) TaskModelProvider.getFactory().newTaskData();        
         taskData.setWorkItemId(workItem.getId());