package prom

import (
	"math/rand"
	"testing"
	"time"
)

func TestProm(t *testing.T) {
	go func() {
		NewEngine(true).Run(":9008")
	}()

	// 模拟gate进程1
	ccuOnlineCounter := NewGauge("ccu").InitDefaultLabels(map[string]string{
		"app_id": "1",
		"app":    "gate",
	}, []string{"nation"})

	// 模拟gate进程2
	ccuOnlineCounter1 := NewGauge("ccu1").InitDefaultLabels(map[string]string{
		"app_id": "2",
		"app":    "gate",
	}, []string{"nation"})

	// 模拟比赛进程1
	ccuBattleCounter := NewGauge("ccu_battle").InitDefaultLabels(map[string]string{
		"app_id": "1",
		"app":    "battle",
	}, []string{"nation"})

	go func() {
		for {
			// 模拟统计gate1间隔收集在线人数
			num := float64(rand.Int31n(100))
			ccuOnlineCounter.LabelValues("cn").Add(num)

			// 模拟统计gate2间隔收集在线人数
			ccuOnlineCounter1.LabelValues("cn").Add(num * 2)

			// 模拟统计比赛1间隔收集比赛人数
			num = float64(rand.Int31n(100))
			ccuBattleCounter.LabelValues("uk").Add(num)

			time.Sleep(time.Second * 5)
		}
	}()

	select {}
}