fix(database): set disk size in provision op

aptible · Jul 11, 2024 · ca0828b · ca0828b
1 parent c48bd10
commit ca0828b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/aptible/resource_database.go b/aptible/resource_database.go
@@ -110,7 +110,6 @@ func resourceDatabaseCreate(d *schema.ResourceData, meta interface{}) error {
 
 	create := aptibleapi.NewCreateDatabaseRequestWithDefaults()
 	create.SetHandle(handle)
-	createDb := client.DatabasesAPI.CreateDatabase(ctx, int32(envID))
 	create.SetType(databaseType)
 
 	if diskSize != 0 {
@@ -129,12 +128,16 @@ func resourceDatabaseCreate(d *schema.ResourceData, meta interface{}) error {
 		create.SetDatabaseImageId(int32(image.ID))
 	}
 
+	createDb := client.DatabasesAPI.CreateDatabase(ctx, int32(envID))
 	db, _, err := createDb.CreateDatabaseRequest(*create).Execute()
 	if err != nil {
 		return err
 	}
 
 	payload := aptibleapi.NewCreateOperationRequest("provision")
+	if diskSize != 0 {
+		payload.SetDiskSize(diskSize)
+	}
 	if containerSize != 0 {
 		payload.SetContainerSize(containerSize)
 	}