add valitation tag

linkernetworks · Jul 16, 2018 · 1537f90 · 1537f90
1 parent ddf35c0
commit 1537f90
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/src/entity/volume.go b/src/entity/volume.go
@@ -18,12 +18,12 @@ const (
 	So the Volume will create a PVC type and connect to a known StorageClass
 */
 type Volume struct {
-	ID          bson.ObjectId                     `bson:"_id,omitempty" json:"id"`
-	Name        string                            `bson:"name" json:"name"`
-	StorageName string                            `bson:"storageName" json:"storageName"`
-	AccessMode  corev1.PersistentVolumeAccessMode `bson:"accessMode" json:"accessMode"`
-	Capacity    string                            `bson:"capacity" json:"capacity"`
-	CreatedAt   *time.Time                        `bson:"createdAt,omitempty" json:"createdAt,omitempty"`
+	ID          bson.ObjectId                     `bson:"_id,omitempty" json:"id" validate:"-"`
+	Name        string                            `bson:"name" json:"name" validate:"required"`
+	StorageName string                            `bson:"storageName" json:"storageName" validate:"required"`
+	AccessMode  corev1.PersistentVolumeAccessMode `bson:"accessMode" json:"accessMode" validate:"required"`
+	Capacity    string                            `bson:"capacity" json:"capacity" validate:"required"`
+	CreatedAt   *time.Time                        `bson:"createdAt,omitempty" json:"createdAt,omitempty" validate:"-"`
 }
 
 //GetCollection - get model mongo collection name.

diff --git a/src/server/handler_volume.go b/src/server/handler_volume.go
@@ -25,6 +25,11 @@ func createVolume(ctx *web.Context) {
 		return
 	}
 
+	if err := sp.Validator.Struct(v); err != nil {
+		response.BadRequest(req.Request, resp.ResponseWriter, err)
+		return
+	}
+
 	session := sp.Mongo.NewSession()
 	session.C(entity.VolumeCollectionName).EnsureIndex(mgo.Index{
 		Key:    []string{"name"},