Improve orderBy(), orderByDescending()

emonkak · Mar 21, 2018 · e647b68 · e647b68
1 parent b8558ae
commit e647b68
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/orderBy.ts b/src/orderBy.ts
@@ -6,5 +6,8 @@ export default function orderBy<TSource, TKey>(this: Iterable<TSource>, keySelec
     if (keySelector == null) {
         keySelector = value => value as any;
     }
+    if (this instanceof OrderedEnumerable) {
+        return (this as any).thenBy(keySelector);
+    }
     return new OrderedEnumerable(this, keySelector, false);
 }
diff --git a/src/orderByDescending.ts b/src/orderByDescending.ts
@@ -6,5 +6,8 @@ export default function orderByDescending<TSource, TKey>(this: Iterable<TSource>
     if (keySelector == null) {
         keySelector = value => value as any;
     }
+    if (this instanceof OrderedEnumerable) {
+        return (this as any).thenByDescending(keySelector);
+    }
     return new OrderedEnumerable(this, keySelector, true);
 }