X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/8adcf378579655c4653cbcdae92c9d9eec154ea8..a1e644bd3b63d97fab7ed1a1d66e00e6dea5fa1e:/sdk/go/keepclient/keepclient.go

diff --git a/sdk/go/keepclient/keepclient.go b/sdk/go/keepclient/keepclient.go
index 2cd6bb4d43..68ac886ddd 100644
--- a/sdk/go/keepclient/keepclient.go
+++ b/sdk/go/keepclient/keepclient.go
@@ -69,11 +69,11 @@ type ErrNotFound struct {
 	multipleResponseError
 }
 
-type InsufficientReplicasError error
+type InsufficientReplicasError struct{ error }
 
-type OversizeBlockError error
+type OversizeBlockError struct{ error }
 
-var ErrOversizeBlock = OversizeBlockError(errors.New("Exceeded maximum block size (" + strconv.Itoa(BLOCKSIZE) + ")"))
+var ErrOversizeBlock = OversizeBlockError{error: errors.New("Exceeded maximum block size (" + strconv.Itoa(BLOCKSIZE) + ")")}
 var MissingArvadosApiHost = errors.New("Missing required environment variable ARVADOS_API_HOST")
 var MissingArvadosApiToken = errors.New("Missing required environment variable ARVADOS_API_TOKEN")
 var InvalidLocatorError = errors.New("Invalid locator")
@@ -97,17 +97,18 @@ type HTTPClient interface {
 
 // KeepClient holds information about Arvados and Keep servers.
 type KeepClient struct {
-	Arvados            *arvadosclient.ArvadosClient
-	Want_replicas      int
-	localRoots         map[string]string
-	writableLocalRoots map[string]string
-	gatewayRoots       map[string]string
-	lock               sync.RWMutex
-	HTTPClient         HTTPClient
-	Retries            int
-	BlockCache         *BlockCache
-	RequestID          string
-	StorageClasses     []string
+	Arvados               *arvadosclient.ArvadosClient
+	Want_replicas         int
+	localRoots            map[string]string
+	writableLocalRoots    map[string]string
+	gatewayRoots          map[string]string
+	lock                  sync.RWMutex
+	HTTPClient            HTTPClient
+	Retries               int
+	BlockCache            *BlockCache
+	RequestID             string
+	StorageClasses        []string
+	DefaultStorageClasses []string // Set by cluster's exported config
 
 	// set to 1 if all writable services are of disk type, otherwise 0
 	replicasPerService int
@@ -119,7 +120,23 @@ type KeepClient struct {
 	disableDiscovery bool
 }
 
-// MakeKeepClient creates a new KeepClient, calls
+func (kc *KeepClient) loadDefaultClasses() error {
+	scData, err := kc.Arvados.ClusterConfig("StorageClasses")
+	if err != nil {
+		return err
+	}
+	classes := scData.(map[string]interface{})
+	for scName := range classes {
+		scConf, _ := classes[scName].(map[string]interface{})
+		isDefault, ok := scConf["Default"].(bool)
+		if ok && isDefault {
+			kc.DefaultStorageClasses = append(kc.DefaultStorageClasses, scName)
+		}
+	}
+	return nil
+}
+
+// MakeKeepClient creates a new KeepClient, loads default storage classes, calls
 // DiscoverKeepServices(), and returns when the client is ready to
 // use.
 func MakeKeepClient(arv *arvadosclient.ArvadosClient) (*KeepClient, error) {
@@ -138,11 +155,16 @@ func New(arv *arvadosclient.ArvadosClient) *KeepClient {
 			defaultReplicationLevel = int(v)
 		}
 	}
-	return &KeepClient{
+	kc := &KeepClient{
 		Arvados:       arv,
 		Want_replicas: defaultReplicationLevel,
 		Retries:       2,
 	}
+	err = kc.loadDefaultClasses()
+	if err != nil {
+		DebugPrintf("DEBUG: Unable to load the default storage classes cluster config")
+	}
+	return kc
 }
 
 // PutHR puts a block given the block hash, a reader, and the number of bytes