fixes #419 : harbor_garbage_collection : add support for workers (#421)

Signed-off-by: flbla <[email protected]>

client/system.go

@@ -27,6 +27,7 @@ func GetSystemBoby(d *schema.ResourceData, scheduleType string) models.SystemBod
 	body.Schedule.Cron = CronStr
 	if scheduleType == "gc" {
 		body.Parameters.DeleteUntagged = d.Get("delete_untagged").(bool)
+		body.Parameters.Workers = d.Get("workers").(int)
 	} else if scheduleType == "purgeaudit" {
 		body.Parameters.AuditRetentionHour = d.Get("audit_retention_hour").(int)
 		body.Parameters.IncludeOperations = d.Get("include_operations").(string)

docs/resources/garbage_collection.md

@@ -14,8 +14,9 @@ description: |-
 
 ```terraform
 resource "harbor_garbage_collection" "main" {
-  schedule         = "Daily"
-  delete_untagged  = true
+  schedule        = "Daily"
+  delete_untagged = true
+  workers         = 1
 }
 ```
 
@@ -28,6 +29,7 @@ resource "harbor_garbage_collection" "main" {
 ### Optional
 
 - `delete_untagged` (Boolean) Allow garbage collection on untagged artifacts.
+- `workers` (Number) Number of workers to run the garbage collection, value must be between 1 and 5.
 
 ### Read-Only
 

examples/resources/harbor_garbage_collection/resource.tf

@@ -1,4 +1,5 @@
 resource "harbor_garbage_collection" "main" {
-  schedule         = "Daily"
-  delete_untagged  = true
+  schedule        = "Daily"
+  delete_untagged = true
+  workers         = 1
 }

models/system.go

@@ -12,6 +12,7 @@ type JobParameters struct {
 	DeleteUntagged     bool   `json:"delete_untagged,omitempty"`
 	AuditRetentionHour int    `json:"audit_retention_hour,omitempty"`
 	IncludeOperations  string `json:"include_operations,omitempty"`
+	Workers            int    `json:"workers,omitempty"`
 }
 
 type SystemBody struct {
@@ -31,6 +32,7 @@ type SystemBody struct {
 		DeleteUntagged     bool   `json:"delete_untagged,omitempty"`
 		AuditRetentionHour int    `json:"audit_retention_hour,omitempty"`
 		IncludeOperations  string `json:"include_operations,omitempty"`
+		Workers            int    `json:"workers,omitempty"`
 		AdditionalProp1    bool   `json:"additionalProp1,omitempty"`
 		AdditionalProp2    bool   `json:"additionalProp2,omitempty"`
 		AdditionalProp3    bool   `json:"additionalProp3,omitempty"`

provider/resource_garbage_collection.go

@@ -20,6 +20,18 @@ func resourceGC() *schema.Resource {
 				Type:     schema.TypeBool,
 				Optional: true,
 			},
+			"workers": {
+				Type:     schema.TypeInt,
+				Optional: true,
+				Default:  1,
+				ValidateFunc: func(i interface{}, k string) (ws []string, errors []error) {
+					value := i.(int)
+					if value < 1 || value > 5 {
+						errors = append(errors, fmt.Errorf("GC workers must be between 1 and 5"))
+					}
+					return
+				},
+			},
 		},
 		Create: resourceGCCreate,
 		Read:   resourceGCRead,
@@ -69,6 +81,7 @@ func resourceGCRead(d *schema.ResourceData, m interface{}) error {
 		d.Set("schedule", jsonData.Schedule.Type)
 	}
 	d.Set("delete_untagged", jsonJobParameters.DeleteUntagged)
+	d.Set("workers", jsonJobParameters.Workers)
 	return nil
 }
 

provider/resource_garbage_collection_test.go

@@ -76,6 +76,7 @@ func testAccCheckGCUpdate() string {
 	resource "harbor_garbage_collection" "main" {
 		schedule        = "Hourly"
 		delete_untagged = true
+		workers 			  = 2
 	}
 	`)
 }

templates/resources/garbage_collection.md.tmpl

@@ -27,6 +27,7 @@ For example, the {{ .SchemaMarkdown }} template can be used to replace manual sc
 ### Optional
 
 - `delete_untagged` (Boolean) Allow garbage collection on untagged artifacts.
+- `workers` (Number) Number of workers to run the garbage collection, value must be between 1 and 5.
 
 ### Read-Only