X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/9eff4af096f60b302abc9ad591f8b5fed45945a4..051ad2017b69ca8e438396b461525e485a896321:/services/keepstore/volume.go

diff --git a/services/keepstore/volume.go b/services/keepstore/volume.go
index 07b497cc2a..6bce05bec0 100644
--- a/services/keepstore/volume.go
+++ b/services/keepstore/volume.go
@@ -1,3 +1,7 @@
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: AGPL-3.0
+
 package main
 
 import (
@@ -8,6 +12,8 @@ import (
 	"math/big"
 	"sync/atomic"
 	"time"
+
+	"git.curoverse.com/arvados.git/sdk/go/arvados"
 )
 
 type BlockWriter interface {
@@ -232,6 +238,13 @@ type Volume interface {
 	// EmptyTrash looks for trashed blocks that exceeded TrashLifetime
 	// and deletes them from the volume.
 	EmptyTrash()
+
+	// Return a globally unique ID of the underlying storage
+	// device if possible, otherwise "".
+	DeviceID() string
+
+	// Get the storage classes associated with this volume
+	GetStorageClasses() []string
 }
 
 // A VolumeWithExamples provides example configs to display in the
@@ -276,12 +289,8 @@ type VolumeManager interface {
 
 // A VolumeMount is an attachment of a Volume to a VolumeManager.
 type VolumeMount struct {
-	UUID        string
-	DeviceID    string
-	ReadOnly    bool
-	Replication int
-	Tier        int
-	volume      Volume
+	arvados.KeepMount
+	volume Volume
 }
 
 // Generate a UUID the way API server would for a "KeepVolumeMount"
@@ -318,18 +327,19 @@ func MakeRRVolumeManager(volumes []Volume) *RRVolumeManager {
 	}
 	vm.mountMap = make(map[string]*VolumeMount)
 	for _, v := range volumes {
-		mnt := &VolumeMount{
-			UUID:        (*VolumeMount)(nil).generateUUID(),
-			DeviceID:    "",
-			ReadOnly:    !v.Writable(),
-			Replication: v.Replication(),
-			Tier:        1,
-			volume:      v,
+		sc := v.GetStorageClasses()
+		if len(sc) == 0 {
+			sc = []string{"default"}
 		}
-		if v, ok := v.(interface {
-			DeviceID() string
-		}); ok {
-			mnt.DeviceID = v.DeviceID()
+		mnt := &VolumeMount{
+			KeepMount: arvados.KeepMount{
+				UUID:           (*VolumeMount)(nil).generateUUID(),
+				DeviceID:       v.DeviceID(),
+				ReadOnly:       !v.Writable(),
+				Replication:    v.Replication(),
+				StorageClasses: sc,
+			},
+			volume: v,
 		}
 		vm.iostats[v] = &ioStats{}
 		vm.mounts = append(vm.mounts, mnt)