packages/core/utils/src/common/partition-array.ts - develop - dimamir/medusa

packages/core/utils/src/common/partition-array.ts

30 строк699 B

Adrien de Peretti

chore(): start moving some packages to the core directory (#7215)

03 май 2024, 14:37

03 май 2024, 14:37bbccd64

/**
 * Partitions an array into two arrays based on a predicate function

 * @example
 * const result = partitionArray([1, 2, 3, 4, 5], (x) => x % 2 === 0)
 * 
 * console.log(result)
 * 
 * // output: [[2, 4], [1, 3, 5]]
 *
 * @param {T} input input array of type T
 * @param {(T) => boolean} predicate function to use when split array elements
 */
export const partitionArray = <T>(
  input: T[],
  predicate: (T) => boolean
): [T[], T[]] => {
  return input.reduce(
    ([pos, neg], currentElement) => {
      if (predicate(currentElement)) {
        pos.push(currentElement)
      } else {
        neg.push(currentElement)
      }

      return [pos, neg]
    },
    [[], []] as [T[], T[]]
  )
}