reapply sort on data source when items change

haxeui · May 9, 2024 · 8a08e47 · 8a08e47
1 parent a7ec6b1
commit 8a08e47
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/haxe/ui/data/ArrayDataSource.hx b/haxe/ui/data/ArrayDataSource.hx
@@ -36,12 +36,11 @@ class ArrayDataSource<T> extends DataSource<T> {
         handleChanged();
     }
 
-    public override function sortCustom(fn:T->T->SortDirection->Int, direction:SortDirection = null) {
+    private override function handleSort(fn:T->T->SortDirection->Int, direction:SortDirection = null) {
         _array.sort(fn.bind(_, _, direction));
         if (_filteredArray != null) {
             _filteredArray.sort(fn.bind(_, _, direction));
         }
-        handleChanged();
     }
 
     private override function handleGetSize():Int {

diff --git a/haxe/ui/data/DataSource.hx b/haxe/ui/data/DataSource.hx
@@ -148,13 +148,25 @@ class DataSource<T> {
 
     private function handleChanged() {
         _changed = true;
+        if (_currentSortFn != null) {
+            handleSort(_currentSortFn, _currentSortDirection);
+        }
         if (_allowCallbacks == true) {
             _changed = false;
             onInternalChange();
         }
     }
 
+    private function handleSort(fn:T->T->SortDirection->Int, direction:SortDirection = null) {
+    }
+
+    private var _currentSortFn:T->T->SortDirection->Int = null;
+    private var _currentSortDirection:SortDirection = null;
     public function sortCustom(fn:T->T->SortDirection->Int, direction:SortDirection = null) {
+        _currentSortFn = fn;
+        _currentSortDirection = direction;
+        handleSort(fn, direction);
+        handleChanged();
     }
 
     public function sort(field:String = null, direction:SortDirection = null) {