Update TemplatesPanel.java

#36
Yong-An-Dang · Nov 22, 2024 · 2fae126 · 2fae126
1 parent 4144009
commit 2fae126
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/nuclei/src/main/java/com/g3g4x5x6/nuclei/panel/tab/TemplatesPanel.java b/nuclei/src/main/java/com/g3g4x5x6/nuclei/panel/tab/TemplatesPanel.java
@@ -457,6 +457,15 @@ public void menuCanceled(MenuEvent e) {
     private void refreshDataForTable() {
         // 搜索功能
         sorter = new TableRowSorter<>(tableModel);
+        sorter.setComparator(0, (o1, o2) -> {
+            try {
+                int num1 = Integer.parseInt(o1.toString());
+                int num2 = Integer.parseInt(o2.toString());
+                return Integer.compare(num1, num2);
+            } catch (NumberFormatException ex) {
+                return o1.toString().compareTo(o2.toString());
+            }
+        });
         templatesTable.setRowSorter(sorter);
         new Thread(() -> {
             NucleiFrame.setStatusProgressBar(L.M("bar.tool.status.progressbar.text", "正在刷新模板列表，请稍后..."));