package compare

import (
	"basic.com/pubsub/protomsg.git"
	"sdkCompare/cache"
	"sort"
)

func realNamePersonsCompere(args protomsg.CompareArgs, fFeature []float32, baseScore float32) protomsg.SdkCompareResult {
	var scr protomsg.SdkCompareResult
	targets := cache.RealNameDbMap.Walk(DoSdkCompare, fFeature, baseScore)
	if len(targets) > 0 {
		sort.Slice(targets, func(i, j int) bool {
			return targets[i].CompareScore > targets[j].CompareScore
		})

		// 截取前5个元素
		topTargets := targets
		if len(targets) > 5 {
			topTargets = targets[:5] // 只取前5个元素
		}

		for idx, _ := range topTargets {
			scr.CompareResult = append(scr.CompareResult, topTargets[idx])
		}
	}

	return scr
}