feature: set quota id for a container

Specified quota id for a container, if id < 0, it means pouchd
alloc a unique quota id. If specified quota id, it will ignore the
quota's regular expression, but the volume will keep its size as before.

Signed-off-by: Rudy Zhang <[email protected]>

Author: rudyfly

cli/common_flags.go

@@ -92,6 +92,7 @@ func addCommonFlags(flagSet *pflag.FlagSet) *container {
 
 	// disk quota
 	flagSet.StringSliceVar(&c.diskQuota, "disk-quota", nil, "Set disk quota for container")
+	flagSet.StringVar(&c.quotaID, "quota-id", "", "Specified quota id, if id < 0, it means pouchd alloc a unique quota id")
 
 	// additional runtime spec annotations
 	flagSet.StringSliceVar(&c.specAnnotation, "annotation", nil, "Additional annotation for runtime")

cli/container.go

@@ -62,6 +62,7 @@ type container struct {
 	capDrop        []string
 	IntelRdtL3Cbm  string
 	diskQuota      []string
+	quotaID        string
 	oomScoreAdj    int64
 	specAnnotation []string
 	cgroupParent   string
@@ -180,6 +181,7 @@ func (c *container) config() (*types.ContainerCreateConfig, error) {
 			InitScript:     c.initScript,
 			ExposedPorts:   ports,
 			DiskQuota:      diskQuota,
+			QuotaID:        c.quotaID,
 			SpecAnnotation: specAnnotation,
 		},
 

daemon/mgr/container.go

@@ -1497,17 +1497,43 @@ func (mgr *ContainerManager) parseBinds(ctx context.Context, meta *ContainerMeta
 }
 
 func (mgr *ContainerManager) setMountPointDiskQuota(ctx context.Context, c *ContainerMeta) error {
-	qid := 0
+	var (
+		qid        uint32
+		setQuotaID bool
+	)
+
 	if c.Config.QuotaID != "" {
-		var err error
-		qid, err = strconv.Atoi(c.Config.QuotaID)
+		id, err := strconv.Atoi(c.Config.QuotaID)
 		if err != nil {
 			return errors.Wrapf(err, "invalid argument, QuotaID: %s", c.Config.QuotaID)
 		}
+
+		// if QuotaID is < 0, it means pouchd alloc a unique quota id.
+		if id < 0 {
+			qid, err = quota.GetNextQuatoID()
+			if err != nil {
+				return errors.Wrap(err, "failed to get next quota id")
+			}
+
+			// update QuotaID
+			c.Config.QuotaID = strconv.Itoa(int(qid))
+		} else {
+			qid = uint32(id)
+		}
 	}
 
-	// parse diskquota regexe
+	if qid > 0 {
+		setQuotaID = true
+	}
+
+	// get rootfs quota
 	quotas := c.Config.DiskQuota
+	defaultQuota := quota.GetDefaultQuota(quotas)
+	if setQuotaID && defaultQuota == "" {
+		return fmt.Errorf("set quota id but have no set quota size")
+	}
+
+	// parse diskquota regexe
 	res := make([]*quota.RegExp, 0)
 	for path, size := range quotas {
 		re := regexp.MustCompile(path)
@@ -1537,11 +1563,15 @@ func (mgr *ContainerManager) setMountPointDiskQuota(ctx context.Context, c *Cont
 			}
 		}
 
-		if matched {
-			err := quota.SetDiskQuota(mp.Source, quotas[mp.Destination], uint32(qid))
-			if err != nil {
-				return err
-			}
+		size := ""
+		if matched && !setQuotaID {
+			size = quotas[mp.Destination]
+		} else {
+			size = defaultQuota
+		}
+		err := quota.SetDiskQuota(mp.Source, size, qid)
+		if err != nil {
+			return err
 		}
 	}
 

pkg/quota/quota.go

@@ -120,3 +120,22 @@ func SetFileAttrNoOutput(dir string, id uint32) {
 func GetNextQuatoID() (uint32, error) {
 	return Gquota.GetNextQuatoID()
 }
+
+//GetDefaultQuota returns the default quota.
+func GetDefaultQuota(quotas map[string]string) string {
+	if quotas == nil {
+		return ""
+	}
+
+	quota, ok := quotas["/"]
+	if ok && quota != "" {
+		return quota
+	}
+
+	quota, ok = quotas[".*"]
+	if ok && quota != "" {
+		return quota
+	}
+
+	return ""
+}