コード例 #1
0
ファイル: volume_service.go プロジェクト: APTrust/exchange
// Returns a Volume object with info about the volume at the specified
// mount point. The mount point should be the path to a disk or partition.
// For example, "/", "/mnt/data", etc.
func (service *VolumeService) getVolume(path string) *models.Volume {
	mountpoint, err := platform.GetMountPointFromPath(path)
	if err != nil {
		mountpoint = "/"
		service.logger.Error("Cannot determine mountpoint of file '%s': %v",
			path, err)
	}
	if _, keyExists := service.volumes[mountpoint]; !keyExists {
		service.volumes[mountpoint] = models.NewVolume(mountpoint)
	}
	return service.volumes[mountpoint]
}
コード例 #2
0
ファイル: volume_test.go プロジェクト: APTrust/exchange
func TestClaimedReserveReleasePath(t *testing.T) {
	_, filename, _, _ := runtime.Caller(0)
	volume := models.NewVolume(filename)
	assert.EqualValues(t, 0, volume.ClaimedSpace())
	assert.Equal(t, filename, volume.MountPoint())

	err := volume.Reserve("/path/to/file_0", 1000)
	require.Nil(t, err)
	assert.EqualValues(t, 1000, volume.ClaimedSpace())

	volume.Release("/this/file/was/never/reserved")
	assert.EqualValues(t, 1000, volume.ClaimedSpace())

	volume.Release("/path/to/file_0")
	assert.EqualValues(t, 0, volume.ClaimedSpace())
}
コード例 #3
0
ファイル: volume_test.go プロジェクト: APTrust/exchange
func TestReservations(t *testing.T) {
	_, filename, _, _ := runtime.Caller(0)
	volume := models.NewVolume(filename)

	paths := []string{"p1", "p2", "p3", "p4", "p5"}
	for i, path := range paths {
		err := volume.Reserve(path, uint64(1000+i))
		assert.Nil(t, err)
	}
	reservations := volume.Reservations()
	require.Equal(t, len(paths), len(reservations))
	for i, path := range paths {
		bytes, keyExists := reservations[path]
		assert.True(t, keyExists)
		assert.EqualValues(t, uint64(1000+i), bytes)
		// Releasing path should remove it from reservations
		volume.Release(path)
	}
	assert.Empty(t, volume.Reservations())
}
コード例 #4
0
ファイル: volume_test.go プロジェクト: APTrust/exchange
// This functional/behavioral test goes through some more realistic
// usage scenarios.
func TestVolume(t *testing.T) {
	_, filename, _, _ := runtime.Caller(0)
	volume := models.NewVolume(filename)

	// Make sure we can reserve space that's actually there.
	initialSpace, err := volume.AvailableSpace()
	require.Nil(t, err)
	numBytes := initialSpace / 3
	err = volume.Reserve("/path/to/file_1", numBytes)
	require.Nil(t, err)
	err = volume.Reserve("/path/to/file_2", numBytes)
	require.Nil(t, err)

	// Make sure we're tracking the available space correctly.
	bytesAvailable, err := volume.AvailableSpace()
	require.Nil(t, err)
	expectedBytesAvailable := (initialSpace - (2 * numBytes))
	assert.Equal(t, expectedBytesAvailable, bytesAvailable)

	// Make sure a request for too much space is rejected
	err = volume.Reserve("/path/to/file_3", numBytes*2)
	require.NotNil(t, err)

	// Free the two chunks of space we just requested.
	volume.Release("/path/to/file_1")
	volume.Release("/path/to/file_2")

	// Make sure it was freed.
	bytesAvailable, err = volume.AvailableSpace()
	require.Nil(t, err)
	assert.Equal(t, initialSpace, bytesAvailable)

	// Now we should have enough space for this.
	err = volume.Reserve("/path/to/file_4", numBytes*2)
	require.Nil(t, err)
}