修复缓存

2026-03-06 21:58:29 -08:00
parent 19932bb9f7
commit 4d913c1728
175 changed files with 437 additions and 12104 deletions
@@ -1,155 +0,0 @@
-package hash
-
-import (
-	"fmt"
-	"strconv"
-	"testing"
-
-	"github.com/stretchr/testify/assert"
-)
-
-const (
-	keySize     = 20
-	requestSize = 1000
-)
-
-func BenchmarkConsistentHashGet(b *testing.B) {
-	ch := NewConsistentHash()
-	for i := 0; i < keySize; i++ {
-		ch.Add("localhost:" + strconv.Itoa(i))
-	}
-
-	for i := 0; i < b.N; i++ {
-		ch.Get(i)
-	}
-}
-
-func TestConsistentHashIncrementalTransfer(t *testing.T) {
-	prefix := "anything"
-	create := func() *ConsistentHash {
-		ch := NewConsistentHash()
-		for i := 0; i < keySize; i++ {
-			ch.Add(prefix + strconv.Itoa(i))
-		}
-		return ch
-	}
-
-	originCh := create()
-	keys := make(map[int]string, requestSize)
-	for i := 0; i < requestSize; i++ {
-		key, ok := originCh.Get(requestSize + i)
-		assert.True(t, ok)
-		assert.NotNil(t, key)
-		keys[i] = key.(string)
-	}
-
-	node := fmt.Sprintf("%s%d", prefix, keySize)
-	for i := 0; i < 10; i++ {
-		laterCh := create()
-		laterCh.AddWithWeight(node, 10*(i+1))
-
-		for j := 0; j < requestSize; j++ {
-			key, ok := laterCh.Get(requestSize + j)
-			assert.True(t, ok)
-			assert.NotNil(t, key)
-			value := key.(string)
-			assert.True(t, value == keys[j] || value == node)
-		}
-	}
-}
-
-func TestConsistentHashTransferOnFailure(t *testing.T) {
-	index := 41
-	keys, newKeys := getKeysBeforeAndAfterFailure(t, "localhost:", index)
-	var transferred int
-	for k, v := range newKeys {
-		if v != keys[k] {
-			transferred++
-		}
-	}
-
-	ratio := float32(transferred) / float32(requestSize)
-	assert.True(t, ratio < 2.5/float32(keySize), fmt.Sprintf("%d: %f", index, ratio))
-}
-
-func TestConsistentHashLeastTransferOnFailure(t *testing.T) {
-	prefix := "localhost:"
-	index := 41
-	keys, newKeys := getKeysBeforeAndAfterFailure(t, prefix, index)
-	for k, v := range keys {
-		newV := newKeys[k]
-		if v != prefix+strconv.Itoa(index) {
-			assert.Equal(t, v, newV)
-		}
-	}
-}
-
-func TestConsistentHash_Remove(t *testing.T) {
-	ch := NewConsistentHash()
-	ch.Add("first")
-	ch.Add("second")
-	ch.Remove("first")
-	for i := 0; i < 100; i++ {
-		val, ok := ch.Get(i)
-		assert.True(t, ok)
-		assert.Equal(t, "second", val)
-	}
-}
-
-func TestConsistentHash_RemoveInterface(t *testing.T) {
-	const key = "any"
-	ch := NewConsistentHash()
-	node1 := newMockNode(key, 1)
-	node2 := newMockNode(key, 2)
-	ch.AddWithWeight(node1, 80)
-	ch.AddWithWeight(node2, 50)
-	assert.Equal(t, 1, len(ch.nodes))
-	node, ok := ch.Get(1)
-	assert.True(t, ok)
-	assert.Equal(t, key, node.(*mockNode).addr)
-	assert.Equal(t, 2, node.(*mockNode).id)
-}
-
-func getKeysBeforeAndAfterFailure(t *testing.T, prefix string, index int) (map[int]string, map[int]string) {
-	ch := NewConsistentHash()
-	for i := 0; i < keySize; i++ {
-		ch.Add(prefix + strconv.Itoa(i))
-	}
-
-	keys := make(map[int]string, requestSize)
-	for i := 0; i < requestSize; i++ {
-		key, ok := ch.Get(requestSize + i)
-		assert.True(t, ok)
-		assert.NotNil(t, key)
-		keys[i] = key.(string)
-	}
-
-	remove := fmt.Sprintf("%s%d", prefix, index)
-	ch.Remove(remove)
-	newKeys := make(map[int]string, requestSize)
-	for i := 0; i < requestSize; i++ {
-		key, ok := ch.Get(requestSize + i)
-		assert.True(t, ok)
-		assert.NotNil(t, key)
-		assert.NotEqual(t, remove, key)
-		newKeys[i] = key.(string)
-	}
-
-	return keys, newKeys
-}
-
-type mockNode struct {
-	addr string
-	id   int
-}
-
-func newMockNode(addr string, id int) *mockNode {
-	return &mockNode{
-		addr: addr,
-		id:   id,
-	}
-}
-
-func (n *mockNode) String() string {
-	return n.addr
-}
@@ -1,47 +0,0 @@
-package hash
-
-import (
-	"crypto/md5"
-	"fmt"
-	"hash/fnv"
-	"math/big"
-	"testing"
-
-	"github.com/stretchr/testify/assert"
-)
-
-const (
-	text      = "hello, world!\n"
-	md5Digest = "910c8bc73110b0cd1bc5d2bcae782511"
-)
-
-func TestMd5(t *testing.T) {
-	actual := fmt.Sprintf("%x", Md5([]byte(text)))
-	assert.Equal(t, md5Digest, actual)
-}
-
-func TestMd5Hex(t *testing.T) {
-	actual := Md5Hex([]byte(text))
-	assert.Equal(t, md5Digest, actual)
-}
-
-func BenchmarkHashFnv(b *testing.B) {
-	for i := 0; i < b.N; i++ {
-		h := fnv.New32()
-		new(big.Int).SetBytes(h.Sum([]byte(text))).Int64()
-	}
-}
-
-func BenchmarkHashMd5(b *testing.B) {
-	for i := 0; i < b.N; i++ {
-		h := md5.New()
-		bytes := h.Sum([]byte(text))
-		new(big.Int).SetBytes(bytes).Int64()
-	}
-}
-
-func BenchmarkMurmur3(b *testing.B) {
-	for i := 0; i < b.N; i++ {
-		Hash([]byte(text))
-	}
-}