Marcel S. Henselin
68e4c137f1
## Description
<!-- **Please link some issue here describing what you are trying to achieve.**
In case there is no issue present for your PR, please consider creating one.
At least please give us some description what you are trying to achieve and why your change is needed. -->
relates to #1234
## Checklist
- [ ] Issue was linked above
- [ ] Code format was applied: `make fmt`
- [ ] Examples were added / adjusted (see `examples/` directory)
- [x] Docs are up-to-date: `make generate-docs` (will be checked by CI)
- [ ] Unit tests got implemented or updated
- [ ] Acceptance tests got implemented or updated (see e.g. [here](f5f99d1709/stackit/internal/services/dns/dns_acc_test.go))
- [x] Unit tests are passing: `make test` (will be checked by CI)
- [x] No linter issues: `make lint` (will be checked by CI)
Reviewed-on: #38
Co-authored-by: Marcel S. Henselin <marcel.henselin@stackit.cloud>
Co-committed-by: Marcel S. Henselin <marcel.henselin@stackit.cloud>
91 lines
2.6 KiB
Go
91 lines
2.6 KiB
Go
package postgresflexalpha
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
postgresflex "tf-provider.git.onstackit.cloud/stackit-dev-tools/terraform-provider-stackitprivatepreview/pkg_gen/postgresflexalpha"
|
|
)
|
|
|
|
// databaseClientReader represents the contract to listing databases from postgresflex.APIClient.
|
|
type databaseClientReader interface {
|
|
ListDatabasesRequest(
|
|
ctx context.Context,
|
|
projectId string,
|
|
region string,
|
|
instanceId string,
|
|
) postgresflex.ApiListDatabasesRequestRequest
|
|
}
|
|
|
|
// getDatabaseById gets a database by its ID.
|
|
func getDatabaseById(
|
|
ctx context.Context,
|
|
client databaseClientReader,
|
|
projectId, region, instanceId string,
|
|
databaseId int64,
|
|
) (*postgresflex.ListDatabase, error) {
|
|
filter := func(db postgresflex.ListDatabase) bool {
|
|
return db.Id != nil && *db.Id == databaseId
|
|
}
|
|
return getDatabase(ctx, client, projectId, region, instanceId, filter)
|
|
}
|
|
|
|
// getDatabaseByName gets a database by its name.
|
|
func getDatabaseByName(
|
|
ctx context.Context,
|
|
client databaseClientReader,
|
|
projectId, region, instanceId, databaseName string,
|
|
) (*postgresflex.ListDatabase, error) {
|
|
filter := func(db postgresflex.ListDatabase) bool {
|
|
return db.Name != nil && *db.Name == databaseName
|
|
}
|
|
return getDatabase(ctx, client, projectId, region, instanceId, filter)
|
|
}
|
|
|
|
// getDatabase is a helper function to retrieve a database using a filter function.
|
|
// Hint: The API does not have a GetDatabase endpoint, only ListDatabases
|
|
func getDatabase(
|
|
ctx context.Context,
|
|
client databaseClientReader,
|
|
projectId, region, instanceId string,
|
|
filter func(db postgresflex.ListDatabase) bool,
|
|
) (*postgresflex.ListDatabase, error) {
|
|
if projectId == "" || region == "" || instanceId == "" {
|
|
return nil, fmt.Errorf("all parameters (project, region, instance) are required")
|
|
}
|
|
|
|
const pageSize = 25
|
|
|
|
for page := int32(1); ; page++ {
|
|
res, err := client.ListDatabasesRequest(ctx, projectId, region, instanceId).
|
|
Page(page).Size(pageSize).Sort(postgresflex.DATABASESORT_DATABASE_ID_ASC).Execute()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("requesting database list (page %d): %w", page, err)
|
|
}
|
|
|
|
// If the API returns no databases, we have reached the end of the list.
|
|
if res.Databases == nil || len(*res.Databases) == 0 {
|
|
break
|
|
}
|
|
|
|
// Iterate over databases to find a match
|
|
for _, db := range *res.Databases {
|
|
if filter(db) {
|
|
foundDb := db
|
|
return &foundDb, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil, fmt.Errorf("database not found for instance %s", instanceId)
|
|
}
|
|
|
|
// cleanString removes leading and trailing quotes which are sometimes returned by the API.
|
|
func cleanString(s *string) *string {
|
|
if s == nil {
|
|
return nil
|
|
}
|
|
res := strings.Trim(*s, "\"")
|
|
return &res
|
|
}
|