fix: list all databases in the project

Author: ecmadao

VERSION

@@ -1 +1 @@
-1.0.9
+1.0.10

api/client.go

@@ -56,7 +56,7 @@ type Client interface {
 	// GetDatabase gets the database by instance resource id and the database name.
 	GetDatabase(ctx context.Context, databaseName string) (*v1pb.Database, error)
 	// ListDatabase list the databases.
-	ListDatabase(ctx context.Context, instanceID, filter string) (*v1pb.ListDatabasesResponse, error)
+	ListDatabase(ctx context.Context, instanceID, filter string) ([]*v1pb.Database, error)
 	// UpdateDatabase patches the database.
 	UpdateDatabase(ctx context.Context, patch *v1pb.Database, updateMasks []string) (*v1pb.Database, error)
 	// BatchUpdateDatabases batch updates databases.

client/database.go

@@ -26,12 +26,37 @@ func (c *client) GetDatabase(ctx context.Context, databaseName string) (*v1pb.Da
 	return &res, nil
 }
 
-// ListDatabase list the databases.
-func (c *client) ListDatabase(ctx context.Context, instanceID, filter string) (*v1pb.ListDatabasesResponse, error) {
-	requestURL := fmt.Sprintf("%s/%s/instances/%s/databases", c.url, c.version, instanceID)
-	if filter != "" {
-		requestURL = fmt.Sprintf("%s?filter=%s", requestURL, url.QueryEscape(filter))
-	}
+// ListDatabase list all databases.
+func (c *client) ListDatabase(ctx context.Context, instanceID, filter string) ([]*v1pb.Database, error) {
+	res := []*v1pb.Database{}
+	pageToken := ""
+
+	for true {
+		resp, err := c.listDatabase(ctx, instanceID, filter, pageToken, 500)
+		if err != nil {
+			return nil, err
+		}
+		res = append(res, resp.Databases...)
+		pageToken = resp.NextPageToken
+		if pageToken == "" {
+			break
+		}
+	}
+
+	return res, nil
+}
+
+// listDatabase list the databases.
+func (c *client) listDatabase(ctx context.Context, instanceID, filter, pageToken string, pageSize int) (*v1pb.ListDatabasesResponse, error) {
+	requestURL := fmt.Sprintf(
+		"%s/%s/instances/%s/databases?filter=%s&page_size=%d&page_token=%s",
+		c.url,
+		c.version,
+		instanceID,
+		url.QueryEscape(filter),
+		pageSize,
+		pageToken,
+	)
 
 	req, err := http.NewRequestWithContext(ctx, "GET", requestURL, nil)
 	if err != nil {

provider/data_source_project.go

@@ -300,7 +300,7 @@ func setProject(
 	project *v1pb.Project,
 ) diag.Diagnostics {
 	filter := fmt.Sprintf(`project == "%s"`, project.Name)
-	listDBResponse, err := client.ListDatabase(ctx, "-", filter)
+	databases, err := client.ListDatabase(ctx, "-", filter)
 	if err != nil {
 		return diag.FromErr(err)
 	}
@@ -353,7 +353,7 @@ func setProject(
 		return diag.Errorf("cannot set postgres_database_tenant_mode for project: %s", err.Error())
 	}
 
-	databaseList := flattenDatabaseList(listDBResponse.Databases)
+	databaseList := flattenDatabaseList(databases)
 	if err := d.Set("databases", schema.NewSet(databaseHash, databaseList)); err != nil {
 		return diag.Errorf("cannot set databases for project: %s", err.Error())
 	}

provider/data_source_project_list.go

@@ -125,12 +125,12 @@ func dataSourceProjectListRead(ctx context.Context, d *schema.ResourceData, m in
 		proj["postgres_database_tenant_mode"] = project.PostgresDatabaseTenantMode
 
 		filter := fmt.Sprintf(`project == "%s"`, project.Name)
-		response, err := c.ListDatabase(ctx, "-", filter)
+		databases, err := c.ListDatabase(ctx, "-", filter)
 		if err != nil {
 			return diag.FromErr(err)
 		}
 
-		proj["databases"] = flattenDatabaseList(response.Databases)
+		proj["databases"] = flattenDatabaseList(databases)
 
 		iamPolicy, err := c.GetProjectIAMPolicy(ctx, project.Name)
 		if err != nil {

provider/internal/mock_client.go

@@ -375,7 +375,7 @@ func (c *mockClient) GetDatabase(_ context.Context, databaseName string) (*v1pb.
 }
 
 // ListDatabase list the databases.
-func (c *mockClient) ListDatabase(_ context.Context, instaceID, filter string) (*v1pb.ListDatabasesResponse, error) {
+func (c *mockClient) ListDatabase(_ context.Context, instaceID, filter string) ([]*v1pb.Database, error) {
 	projectID := "-"
 	if strings.HasPrefix(filter, "project == ") {
 		projectID = strings.Split(filter, "project == ")[1]
@@ -391,9 +391,7 @@ func (c *mockClient) ListDatabase(_ context.Context, instaceID, filter string) (
 		databases = append(databases, db)
 	}
 
-	return &v1pb.ListDatabasesResponse{
-		Databases: databases,
-	}, nil
+	return databases, nil
 }
 
 // UpdateDatabase patches the database.

provider/resource_project.go

@@ -367,12 +367,12 @@ func resourceProjectDelete(ctx context.Context, d *schema.ResourceData, m interf
 
 func updateDatabasesInProject(ctx context.Context, d *schema.ResourceData, client api.Client, projectName string) diag.Diagnostics {
 	filter := fmt.Sprintf(`project == "%s"`, projectName)
-	listDB, err := client.ListDatabase(ctx, "-", filter)
+	databases, err := client.ListDatabase(ctx, "-", filter)
 	if err != nil {
 		return diag.Errorf("failed to list database with error: %v", err)
 	}
 	existedDBMap := map[string]*v1pb.Database{}
-	for _, db := range listDB.Databases {
+	for _, db := range databases {
 		existedDBMap[db.Name] = db
 	}