124 lines
2.2 KiB
Go
124 lines
2.2 KiB
Go
package weight
|
|
|
|
import (
|
|
"sort"
|
|
)
|
|
|
|
// Weight 权重
|
|
type Weight struct {
|
|
// 权重值
|
|
Value float64
|
|
// 权重名称
|
|
Name string
|
|
}
|
|
|
|
// WeightList 权重列表
|
|
type WeightList []Weight
|
|
|
|
// Len 长度
|
|
func (w WeightList) Len() int {
|
|
return len(w)
|
|
}
|
|
|
|
// Less 小于
|
|
func (w WeightList) Less(i, j int) bool {
|
|
return w[i].Value < w[j].Value
|
|
}
|
|
|
|
// Swap 交换
|
|
func (w WeightList) Swap(i, j int) {
|
|
w[i], w[j] = w[j], w[i]
|
|
}
|
|
|
|
// Sort 排序
|
|
func (w WeightList) Sort() {
|
|
sort.Sort(w)
|
|
}
|
|
|
|
// WeightModel 权重模型
|
|
type WeightModel struct {
|
|
// 权重列表
|
|
Weights WeightList
|
|
// 权重总和
|
|
Total float64
|
|
}
|
|
|
|
// NewWeightModel 创建权重模型
|
|
func NewWeightModel() *WeightModel {
|
|
return &WeightModel{
|
|
Weights: make(WeightList, 0),
|
|
}
|
|
}
|
|
|
|
// Add 添加权重
|
|
func (w *WeightModel) Add(name string, value float64) {
|
|
w.Weights = append(w.Weights, Weight{
|
|
Name: name,
|
|
Value: value,
|
|
})
|
|
w.Total += value
|
|
}
|
|
|
|
// Get 获取权重
|
|
func (w *WeightModel) Get(name string) float64 {
|
|
for _, weight := range w.Weights {
|
|
if weight.Name == name {
|
|
return weight.Value
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
// GetWeight 获取权重
|
|
func (w *WeightModel) GetWeight(name string) float64 {
|
|
for _, weight := range w.Weights {
|
|
if weight.Name == name {
|
|
return weight.Value / w.Total
|
|
}
|
|
}
|
|
return 0
|
|
}
|
|
|
|
// GetWeightList 获取权重列表
|
|
func (w *WeightModel) GetWeightList() WeightList {
|
|
w.Weights.Sort()
|
|
return w.Weights
|
|
}
|
|
|
|
// GetWeightListByValue 获取权重列表
|
|
func (w *WeightModel) GetWeightListByValue() WeightList {
|
|
w.Weights.Sort()
|
|
return w.Weights
|
|
}
|
|
|
|
// GetWeightListByValueDesc 获取权重列表
|
|
func (w *WeightModel) GetWeightListByValueDesc() WeightList {
|
|
w.Weights.Sort()
|
|
return w.Weights
|
|
}
|
|
|
|
// GetWeightListByName 获取权重列表
|
|
func (w *WeightModel) GetWeightListByName() WeightList {
|
|
w.Weights.Sort()
|
|
return w.Weights
|
|
}
|
|
|
|
// GetWeightListByNameDesc 获取权重列表
|
|
func (w *WeightModel) GetWeightListByNameDesc() WeightList {
|
|
w.Weights.Sort()
|
|
return w.Weights
|
|
}
|
|
|
|
// GetWeightListByWeight 获取权重列表
|
|
func (w *WeightModel) GetWeightListByWeight() WeightList {
|
|
w.Weights.Sort()
|
|
return w.Weights
|
|
}
|
|
|
|
// GetWeightListByWeightDesc 获取权重列表
|
|
func (w *WeightModel) GetWeightListByWeightDesc() WeightList {
|
|
w.Weights.Sort()
|
|
return w.Weights
|
|
}
|
|
|