first

lib/cache/memory_test.go

@@ -0,0 +1,107 @@
+package cache
+
+import (
+	"fmt"
+	"testing"
+	"time"
+)
+
+func TestMemorySet(t *testing.T) {
+	mc := NewMemoryCache(0)
+	err := mc.Set("123", "44567", 0)
+	if err != nil {
+		fmt.Println(err.Error())
+		t.Fatalf("写入失败")
+	}
+}
+
+func TestMemoryGet(t *testing.T) {
+	mc := NewMemoryCache(0)
+	mc.Set("123", "44567", 0)
+	res := ""
+	err := mc.Get("123", &res)
+	fmt.Println("res", res)
+	if err != nil {
+		t.Fatalf("读取失败 " + err.Error())
+	}
+	if res != "44567" {
+		t.Fatalf("读取错误")
+	}
+
+}
+
+func TestMemorySetExpGet(t *testing.T) {
+	mc := NewMemoryCache(0)
+	//mc.stopEviction()
+	mc.Set("1", "10", 10)
+	mc.Set("2", "5", 5)
+	err := mc.Set("3", "3", 3)
+	if err != nil {
+		t.Fatalf("写入失败")
+	}
+
+	res := ""
+	err = mc.Get("3", &res)
+	if err != nil {
+		t.Fatalf("读取失败" + err.Error())
+	}
+	fmt.Println("res 3", res)
+	time.Sleep(4 * time.Second)
+	//res = ""
+	err = mc.Get("3", &res)
+	if err != nil {
+		t.Fatalf("读取失败" + err.Error())
+	}
+	fmt.Println("res 3", res)
+	err = mc.Get("2", &res)
+	if err != nil {
+		t.Fatalf("读取失败" + err.Error())
+	}
+	fmt.Println("res 2", res)
+	err = mc.Get("1", &res)
+	if err != nil {
+		t.Fatalf("读取失败" + err.Error())
+	}
+	fmt.Println("res 1", res)
+
+}
+func TestMemoryLru(t *testing.T) {
+	mc := NewMemoryCache(18)
+	mc.Set("1", "1111", 10)
+	mc.Set("2", "2222", 5)
+	//读取一次，2就会被放到最后
+	mc.Get("1", nil)
+	err := mc.Set("3", "三", 3)
+	if err != nil {
+		//t.Fatalf("写入失败")
+	}
+
+	res := ""
+	err = mc.Get("3", &res)
+	if err != nil {
+		t.Fatalf("读取失败" + err.Error())
+	}
+	fmt.Println("res3", res)
+	res = ""
+	err = mc.Get("2", &res)
+	if err != nil {
+		t.Fatalf("读取失败" + err.Error())
+	}
+	fmt.Println("res2", res)
+	res = ""
+	err = mc.Get("1", &res)
+	if err != nil {
+		t.Fatalf("读取失败" + err.Error())
+	}
+	fmt.Println("res1", res)
+
+}
+func BenchmarkMemorySet(b *testing.B) {
+	mc := NewMemoryCache(0)
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		key := fmt.Sprintf("key%d", i)
+		value := fmt.Sprintf("value%d", i)
+		mc.Set(key, value, 1000)
+	}
+}