es-toolkit
Version:
A state-of-the-art, high-performance JavaScript utility library with a small bundle size and strong type annotations.
54 lines (53 loc) • 2.21 kB
JavaScript
const require_compareValues = require("../_internal/compareValues.js");
//#region src/array/orderBy.ts
/**
* Sorts an array of objects based on the given `criteria` and their corresponding order directions.
*
* - If you provide keys, it sorts the objects by the values of those keys.
* - If you provide functions, it sorts based on the values returned by those functions.
*
* The function returns the array of objects sorted in corresponding order directions.
* If two objects have the same value for the current criterion, it uses the next criterion to determine their order.
* If the number of orders is less than the number of criteria, it uses the last order for the rest of the criteria.
*
* @template T - The type of elements in the array.
* @param arr - The array of objects to be sorted.
* @param criteria - The criteria for sorting. This can be an array of object keys or functions that return values used for sorting.
* @param orders - An array of order directions ('asc' for ascending or 'desc' for descending).
* @returns The sorted array.
*
* @example
* // Sort an array of objects by 'user' in ascending order and 'age' in descending order.
* const users = [
* { user: 'fred', age: 48 },
* { user: 'barney', age: 34 },
* { user: 'fred', age: 40 },
* { user: 'barney', age: 36 },
* ];
*
* const result = orderBy(users, [obj => obj.user, 'age'], ['asc', 'desc']);
* // result will be:
* // [
* // { user: 'barney', age: 36 },
* // { user: 'barney', age: 34 },
* // { user: 'fred', age: 48 },
* // { user: 'fred', age: 40 },
* // ]
*/
function orderBy(arr, criteria, orders) {
return arr.slice().sort((a, b) => {
const ordersLength = orders.length;
for (let i = 0; i < criteria.length; i++) {
const order = ordersLength > i ? orders[i] : orders[ordersLength - 1];
const criterion = criteria[i];
const criterionIsFunction = typeof criterion === "function";
const valueA = criterionIsFunction ? criterion(a) : a[criterion];
const valueB = criterionIsFunction ? criterion(b) : b[criterion];
const result = require_compareValues.compareValues(valueA, valueB, order);
if (result !== 0) return result;
}
return 0;
});
}
//#endregion
exports.orderBy = orderBy;