package test

import (
	"context"
	"joybuywork/internal/repository/cache"
	"joybuywork/internal/repository/finder"
	"runtime"
	"sync"
	"testing"
	"time"
)

func TestFinder(t *testing.T) {

	t.Run("func", func(t *testing.T) {

		var m1, m2 runtime.MemStats

		runtime.ReadMemStats(&m1)

		data, _ := cache.NewProject(context.Background()).FindById(143)

		t.Logf("%+v", data)

		runtime.ReadMemStats(&m2)

		t.Logf("Alloc: %v B, TotalAlloc: %v B, HeapAlloc: %v B",
			m2.Alloc-m1.Alloc,
			m2.TotalAlloc-m1.TotalAlloc,
			m2.HeapAlloc-m1.HeapAlloc)
	})

	t.Run("call", func(t *testing.T) {
		var m1, m2 runtime.MemStats

		runtime.ReadMemStats(&m1)

		data, _ := finder.Project(context.Background(), int64(143))

		t.Logf("%+v", data)

		runtime.ReadMemStats(&m2)

		t.Logf("Alloc: %v, TotalAlloc: %v, HeapAlloc: %v",
			m2.Alloc-m1.Alloc,
			m2.TotalAlloc-m1.TotalAlloc,
			m2.HeapAlloc-m1.HeapAlloc)
	})

	var ms runtime.MemStats
	runtime.ReadMemStats(&ms)
	t.Logf("堆内存: %d MiB, map size: %d MiB, 总申请：%d Mib\n", ms.Alloc/1024/1024, ms.Sys/1024/1024, ms.TotalAlloc/1024/1024)
	t.Run("wait", func(t *testing.T) {

		go func() {
			t1 := time.NewTicker(time.Second)
			for {
				select {
				case <-t1.C:

					wg := sync.WaitGroup{}
					for i := 0; i < 10; i++ {
						wg.Add(1)
						go func() {
							defer wg.Done()
							cache.NewProject(context.Background()).FindById(143)

						}()
					}

					wg.Wait()
				}
			}
		}()

	})

	t.Run("wait2", func(t *testing.T) {
		t2 := time.NewTicker(time.Second)
		for {
			select {
			case <-t2.C:
				runtime.ReadMemStats(&ms)
				t.Logf("堆内存: %d MiB, map size: %d MiB, 总申请：%d Mib\n", ms.Alloc/1024/1024, ms.Sys/1024/1024, ms.TotalAlloc/1024/1024)
			}
		}
	})

}

func TestAllocations(t *testing.T) {
	allocs := testing.AllocsPerRun(100, func() {
		finder.Project(context.Background(), 143)
	})
	if allocs > 0 {
		t.Errorf("代码发生了 %v 次内存分配", allocs)
	}
}

func BenchmarkSomething(b *testing.B) {
	b.ReportAllocs() // 报告内存分配情况
	for i := 0; i < b.N; i++ {
		finder.Project(context.Background(), int64(143))
	}
}