zer0/harmony
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Merge pull request #59 from Snowflake107/main

Browse source 
update collection methods
This commit is contained in:
Helloyunho 2020-12-14 19:27:20 +09:00 • committed by GitHub
commit 67c361b9cd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

37
src/utils/collection.ts
View file

@ -10,20 +10,37 @@ export class Collection<K = string, V = any> extends Map<K, V> {
return [...this.values()] return [...this.values()]
} }
/** Get first value in Collection */ /** Get first value(s) in Collection */
first(): V { first(): V | undefined;
return this.values().next().value first(amount: number): V[];
first(amount?: number): V | V[] | undefined {
if (typeof amount === 'undefined') return this.values().next().value
if (amount < 0) return this.last(amount * -1)
amount = Math.min(this.size, amount)
const iter = this.values()
return Array.from({ length: amount }, (): V => iter.next().value)
} }
/** Get last value in Collection */ /** Get last value(s) in Collection */
last(): V { last(): V | undefined;
return [...this.values()][this.size - 1] last(amount: number): V[];
last(amount?: number): V | V[] | undefined {
const arr = this.array()
if (typeof amount === 'undefined') return arr[arr.length - 1]
if (amount < 0) return this.first(amount * -1)
if (!amount) return [] // eslint-disable-line
return arr.slice(-amount)
} }
/** Get a random value from Collection */ /** Get random value(s) from Collection */
random(): V { random(): V;
const arr = [...this.values()] random(amount: number): V[];
return arr[Math.floor(Math.random() * arr.length)] random(amount?: number): V | V[] {
let arr = this.array()
if (typeof amount === 'undefined') return arr[Math.floor(Math.random() * arr.length)]
if (arr.length === 0 || !amount) return [] // eslint-disable-line
arr = arr.slice()
return Array.from({ length: amount }, (): V => arr.splice(Math.floor(Math.random() * arr.length), 1)[0])
} }
/** Find a value from Collection using callback */ /** Find a value from Collection using callback */