upstream-seaweedfs/weed/topology/crowded_writable_test.go

package topology

import (
	"sync"
	"testing"
	"time"

	"github.com/seaweedfs/seaweedfs/weed/pb/master_pb"
	"github.com/seaweedfs/seaweedfs/weed/storage/needle"
	"github.com/seaweedfs/seaweedfs/weed/storage/super_block"
	"github.com/seaweedfs/seaweedfs/weed/storage/types"
)

// Crowding asks for more room to write into, so a read-only volume has nothing
// to say about it. Growth already discounts them by intersecting with the
// writable list, so an entry for one is weight and nothing else.
func TestCrowdedVolumesAreOnesThatCanTakeWrites(t *testing.T) {
	topo := NewTopology("crowd", nil, 10000, 5, false)
	dn := topo.GetOrCreateDataCenter("dc1").GetOrCreateRack("rack1").
		GetOrCreateDataNode("127.0.0.1", 8080, 18080, "", "", map[string]uint32{"": 100})

	// Both are past the growth threshold; only one can be written to.
	topo.SyncDataNodeRegistration([]*master_pb.VolumeInformationMessage{
		{Id: 1, Size: 9500, Collection: "c", Version: 3},
		{Id: 2, Size: 9500, Collection: "c", Version: 3, ReadOnly: true},
	}, dn)
	dn.LastSeen = time.Now().Unix()

	// The channels are unbuffered and normally drained by the writable-refresh
	// loop, which is not running here.
	crowded := map[needle.VolumeId]bool{}
	stop, done := make(chan struct{}), make(chan struct{})
	go func() {
		defer close(done)
		for {
			select {
			case v := <-topo.chanCrowdedVolumes:
				crowded[v.Id] = true
			case <-topo.chanFullVolumes:
			case <-stop:
				return
			}
		}
	}()
	// The channels are unbuffered, so every send has been received by the time
	// this returns; waiting for the collector to finish covers the recording.
	topo.CollectDeadNodeAndFullVolumes(time.Now().Unix()-1, topo.volumeSizeLimit, 0.9)
	close(stop)
	<-done // the collector owns the map until it returns

	if !crowded[needle.VolumeId(1)] {
		t.Error("a writable volume past the threshold was not reported crowded")
	}
	if crowded[needle.VolumeId(2)] {
		t.Error("a read-only volume was reported crowded, which only adds an entry growth then discounts")
	}
}

// The count growth decides on has to be unchanged by this.
func TestCrowdedCountStillMatchesWritables(t *testing.T) {
	topo := NewTopology("crowd", nil, 10000, 5, false)
	dn := topo.GetOrCreateDataCenter("dc1").GetOrCreateRack("rack1").
		GetOrCreateDataNode("127.0.0.1", 8080, 18080, "", "", map[string]uint32{"": 100})
	topo.SyncDataNodeRegistration([]*master_pb.VolumeInformationMessage{
		{Id: 1, Size: 9500, Collection: "c", Version: 3},
	}, dn)

	rp, _ := super_block.NewReplicaPlacementFromString("000")
	vl := topo.GetVolumeLayout("c", rp, needle.EMPTY_TTL, types.HardDriveType)
	vl.SetVolumeCrowded(needle.VolumeId(1))

	writable, crowded := vl.GetWritableVolumeCount()
	if writable != 1 || crowded != 1 {
		t.Errorf("writable=%d crowded=%d, want 1 and 1", writable, crowded)
	}
}

// SetVolumeCrowded mutates the crowded map while GetWritableVolumeCount reads
// it on the Assign hot path. Under RLock both could run at once and the Go
// runtime aborts with "concurrent map read and map write". Run with -race to
// reproduce the original bug; the write lock makes this pass.
func TestSetVolumeCrowdedNoRaceWithGetWritableVolumeCount(t *testing.T) {
	rp, _ := super_block.NewReplicaPlacementFromString("000")
	vl := NewVolumeLayout(rp, needle.EMPTY_TTL, types.HardDriveType, 10000, false)
	// GetWritableVolumeCount iterates vl.writables, so give it something to
	// read while the crowded map is being mutated.
	vl.writables = []needle.VolumeId{1, 2, 3}

	var wg sync.WaitGroup
	stop := make(chan struct{})
	for range 8 {
		wg.Add(1)
		go func() {
			defer wg.Done()
			for {
				select {
				case <-stop:
					return
				default:
					vl.SetVolumeCrowded(needle.VolumeId(2))
				}
			}
		}()
	}
	for range 8 {
		wg.Add(1)
		go func() {
			defer wg.Done()
			for {
				select {
				case <-stop:
					return
				default:
					vl.GetWritableVolumeCount()
				}
			}
		}()
	}

	// The race detector needs only a brief window to flag a concurrent
	// map read/write; let the goroutines hammer the lock briefly.
	time.Sleep(50 * time.Millisecond)
	close(stop)
	wg.Wait()
}