go init

weight.go

@@ -0,0 +1,123 @@
+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
+}
+