From 1389332205f84311a6cd3cba0293eba62ae3824b Mon Sep 17 00:00:00 2001
From: richardxz <richardxiang@yunify.com>
Date: Wed, 19 Sep 2018 14:00:35 +0800
Subject: [PATCH] support return storageclass's provisioner

---
 pkg/models/controllers/storageclasses.go | 2 ++
 pkg/models/controllers/types.go          | 1 +
 2 files changed, 3 insertions(+)

diff --git a/pkg/models/controllers/storageclasses.go b/pkg/models/controllers/storageclasses.go
index 705ca32b3..b0bccc122 100644
--- a/pkg/models/controllers/storageclasses.go
+++ b/pkg/models/controllers/storageclasses.go
@@ -39,6 +39,7 @@ func (ctl *StorageClassCtl) generateObject(item v1.StorageClass) *StorageClass {
 	name := item.Name
 	createTime := item.CreationTimestamp.Time
 	isDefault := false
+	provisioner := item.Provisioner
 	if item.Annotations["storageclass.beta.kubernetes.io/is-default-class"] == "true" {
 		isDefault = true
 	}
@@ -53,6 +54,7 @@ func (ctl *StorageClassCtl) generateObject(item v1.StorageClass) *StorageClass {
 		CreateTime:  createTime,
 		IsDefault:   isDefault,
 		Annotation:  MapString{item.Annotations},
+		Provisioner: provisioner,
 	}
 
 	return object
diff --git a/pkg/models/controllers/types.go b/pkg/models/controllers/types.go
index 34e9ab55b..548a86b2d 100644
--- a/pkg/models/controllers/types.go
+++ b/pkg/models/controllers/types.go
@@ -279,6 +279,7 @@ type StorageClass struct {
 	CreateTime  time.Time `gorm:"column:createTime" json:"createTime,omitempty"`
 	IsDefault   bool      `json:"default"`
 	Count       int       `json:"count"`
+	Provisioner string    `json:"provisioner"`
 }
 
 type JobRevisions map[int]JobStatus