package shardmap

import (
	"github.com/goinbox/crypto"
	"github.com/goinbox/gomisc"

	"strconv"
	"sync"
	"testing"
)

var shardMap *ShardMap
var syncMap *sync.Map
var simpleMap map[string]int64

func init() {
	shardMap = New(32)
	syncMap = new(sync.Map)
	simpleMap = make(map[string]int64)
}

func TestSetGet(t *testing.T) {
	for i := 0; i < 10000; i++ {
		key := getIntMd5(i)
		shardMap.Set(key, i)

		v, ok := shardMap.Get(key)
		if !ok || v != i {
			t.Error(v, ok)
		}
	}
}

func TestWalkDel(t *testing.T) {
	shardMap.Walk(func(k string, v interface{}) {
		//t.Log(k, v)

		shardMap.Del(k)

		_, ok := shardMap.Get(k)
		if ok {
			t.Error(v, ok)
		}
	})
}

func BenchmarkShardMapWrite(b *testing.B) {
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			ki := gomisc.RandByTime(nil)
			k := strconv.FormatInt(ki, 10)
			shardMap.Set(k, ki)
		}
	})
}

func getIntMd5(i int) string {
	return crypto.Md5String([]byte(strconv.Itoa(i)))
}

func BenchmarkSyncMapWrite(b *testing.B) {
	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			ki := gomisc.RandByTime(nil)
			k := strconv.FormatInt(ki, 10)
			syncMap.Store(k, ki)
		}
	})
}

func BenchmarkSimpleMapWrite(b *testing.B) {
	lock := new(sync.Mutex)

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			ki := gomisc.RandByTime(nil)
			k := strconv.FormatInt(ki, 10)
			lock.Lock()
			simpleMap[k] = ki
			lock.Unlock()
		}
	})
}