golang對結構體排序(struct),重寫sort

package main

import (
    "fmt"
    "sort"
)

type Log struct {
    UserID     int
    Message    string
    Num        float64
    CreateTime string
}

type Wrapper struct {
    log []Log
    by  func(p, q *Log) bool
}

type SortBy func(p, q *Log) bool

func (pw Wrapper) Len() int { // 重寫 Len() 方法
    return len(pw.log)
}
func (pw Wrapper) Swap(i, j int) { // 重寫 Swap() 方法
    pw.log[i], pw.log[j] = pw.log[j], pw.log[i]
}
func (pw Wrapper) Less(i, j int) bool { // 重寫 Less() 方法
    return pw.by(&pw.log[i], &pw.log[j])
}

// 封裝成 SortLog 方法
func SortLog(log []Log, by SortBy) {
    sort.Sort(Wrapper{log, by})
}

func main() {
    log := []Log{
        {1, "簽到", 1, "1563935120"},
        {1, "充值", 100, "1563935320"},
    }

    fmt.Println(log)
    //調用wrapper
    sort.Sort(Wrapper{log, func(p, q *Log) bool {
        return q.Num < p.Num // Num 遞減排序
    }})

    fmt.Println(log)
    //間接封裝
    SortLog(log, func(p, q *Log) bool {
        return p.CreateTime < q.CreateTime // CreateTime 遞增排序
    })

    fmt.Println(log)

}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章