55 lines
1.6 KiB
TypeScript
55 lines
1.6 KiB
TypeScript
import { DataSource } from '@angular/cdk/collections';
|
|
import { MatPaginator } from '@angular/material/paginator';
|
|
import { MatSort } from '@angular/material/sort';
|
|
import { merge, Observable, of as observableOf } from 'rxjs';
|
|
import { map } from 'rxjs/operators';
|
|
|
|
import { User } from '../../core/user';
|
|
|
|
/** Simple sort comparator for example ID/Name columns (for client-side sorting). */
|
|
function compare(a, b, isAsc) {
|
|
return (a < b ? -1 : 1) * (isAsc ? 1 : -1);
|
|
}
|
|
|
|
export class UserListDataSource extends DataSource<User> {
|
|
constructor(private paginator: MatPaginator, private sort: MatSort, public data: User[]) {
|
|
super();
|
|
}
|
|
|
|
connect(): Observable<User[]> {
|
|
const dataMutations = [observableOf(this.data), this.paginator.page, this.sort.sortChange];
|
|
|
|
// Set the paginators length
|
|
this.paginator.length = this.data.length;
|
|
|
|
return merge(...dataMutations).pipe(
|
|
map(() => this.getPagedData(this.getSortedData([...this.data]))),
|
|
);
|
|
}
|
|
|
|
disconnect() {}
|
|
|
|
private getPagedData(data: User[]) {
|
|
const startIndex = this.paginator.pageIndex * this.paginator.pageSize;
|
|
return data.splice(startIndex, this.paginator.pageSize);
|
|
}
|
|
|
|
private getSortedData(data: User[]) {
|
|
if (!this.sort.active || this.sort.direction === '') {
|
|
return data;
|
|
}
|
|
|
|
return data.sort((a, b) => {
|
|
const isAsc = this.sort.direction === 'asc';
|
|
switch (this.sort.active) {
|
|
case 'name':
|
|
return compare(a.name, b.name, isAsc);
|
|
case 'id':
|
|
return compare(+a.id, +b.id, isAsc);
|
|
default:
|
|
return 0;
|
|
}
|
|
});
|
|
}
|
|
}
|