typescript学习：泛型函数 + 函数重载，实现多功能排序器

本文内容如下

泛型函数 + 函数重载，实现多功能排序器

如果你都有了答案，可以忽略本文章，或去TS学习地图寻找更多答案

---

快排算法

用于字母 和 数字排序

function quickSort<T>(arr: T[]): T[] {
    if (arr.length < 2) return arr

    let small: T[] = []
    let big: T[] = []
    let target = arr[0]

    for (let i = 1; i < arr.length; i++) {
        if (arr[i] > target) {
            big.push(arr[i])
        } else {
            small.push(arr[i])
        }
    }

    return [...quickSort(small), target, ...quickSort(big)]
}

中文排序

function sortChinese(array: string[]): any[] {
    return array.sort((a, b) => {
        return a.localeCompare(b)
    })
}

字符串排序

function strSort(str: string, count?: number): string {
    let result = quickSort(str.split('')).join('')
    if (count) {
        return result.substr(0, count) + '...'
    }
    return result

}

用于判断是否是中文

function isChinese<T>(array: T[]): boolean {
    let pattern = /[\u4e00-\u9fa5]+/g
    return array.some((item) => pattern.test(item as any))
}

---

实现排序器

中文，英文，数字数组排序，内部字符串排序，字符串排序

function sort<T>(data: T, count: number = 5): T[] | string {
    if (typeof data === 'string') {
        return strSort(data, count)
    }

    if (data instanceof Array) {
        if (isChinese(data)) {
            return sortChinese(data)
        }

        // 内部item的字母排序
        let result: any = data.map((item) => typeof item === 'string' ? strSort(item) : item)

        return quickSort(result)
    }

    return []
}

console.log(sort('adcbegf', 6));   
//abcdef...
 
let chineseArr = ['武汉', '郑州', '大连', '上海', '北京', '玉林']
console.log(sort(chineseArr));
//["北京", "大连", "上海", "武汉", "玉林", "郑州"]

console.log(sort(['bp', 'da', 'lt', 'zy']));
//["ad", "bp", "lt", "yz"]

console.log(sort([5, 1, 9, 2, 5, 4, 6, 9, 10, 65]));
//[1, 2, 4, 5, 5, 6, 9, 9, 10, 65]

该版本问题：

• 功能运行期间正常
• 在编译期间，没有提示，无法获取属性，如数组类型无法获取forEach方法等等

返回联合类型 string | number[][]
返回二维数组的原因是：ts自动推断的泛型为sort<number[]>()，然后返回值是：T[]，加起来就是number[][]

let result = sort([5, 1, 9, 2, 5, 4, 6, 9, 10, 65]) as number[]  // 无法断言为数组
result.f    //没有提示语法

---

最终：函数重载版

function sort(str: string, count?: number): string
function sort<T>(data: T[], count?: number): T[]
function sort(data: any, count: number = 5): any {
    if (typeof data === 'string') {
        return strSort(data, count)
    }

    if (data instanceof Array) {
        if (isChinese(data)) {
            return sortChinese(data)
        }

        let result = data.map((item) => typeof item === 'string' ? strSort(item) : item)

        return quickSort(result)
    }
}

// 返回number[]类型，ts自动推断泛型为sort<number>()，返回T[]，就是正确的
let result = sort([5, 1, 9, 2, 5, 4, 6, 9, 10, 65])
result.forEach  //数组，字符串都有正确的代码提示

---

学习更多

TS学习地图