9f41c4da7f
## 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: #4
Reviewed-by: Andre_Harms <andre.harms@stackit.cloud>
Co-authored-by: Marcel S. Henselin <marcel.henselin@stackit.cloud>
Co-committed-by: Marcel S. Henselin <marcel.henselin@stackit.cloud>
85 lines
3.1 KiB
Go
85 lines
3.1 KiB
Go
package wait
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/stackitcloud/stackit-sdk-go/core/oapierror"
|
|
"github.com/stackitcloud/stackit-sdk-go/core/wait"
|
|
"github.com/stackitcloud/stackit-sdk-go/services/iaasalpha"
|
|
)
|
|
|
|
const (
|
|
CreateSuccess = "CREATED"
|
|
)
|
|
|
|
// Interfaces needed for tests
|
|
type APIClientInterface interface {
|
|
GetNetworkExecute(ctx context.Context, projectId, region, networkId string) (*iaasalpha.Network, error)
|
|
}
|
|
|
|
// CreateNetworkWaitHandler will wait for network creation using network id
|
|
func CreateNetworkWaitHandler(ctx context.Context, a APIClientInterface, projectId, region, networkId string) *wait.AsyncActionHandler[iaasalpha.Network] {
|
|
handler := wait.New(func() (waitFinished bool, response *iaasalpha.Network, err error) {
|
|
network, err := a.GetNetworkExecute(ctx, projectId, region, networkId)
|
|
if err != nil {
|
|
return false, network, err
|
|
}
|
|
if network.Id == nil || network.Status == nil {
|
|
return false, network, fmt.Errorf("create failed for network with id %s, the response is not valid: the id or the state are missing", networkId)
|
|
}
|
|
// The state returns to "CREATED" after a successful creation is completed
|
|
if *network.Id == networkId && *network.Status == CreateSuccess {
|
|
return true, network, nil
|
|
}
|
|
return false, network, nil
|
|
})
|
|
handler.SetSleepBeforeWait(2 * time.Second)
|
|
handler.SetTimeout(15 * time.Minute)
|
|
return handler
|
|
}
|
|
|
|
// UpdateNetworkWaitHandler will wait for network update
|
|
func UpdateNetworkWaitHandler(ctx context.Context, a APIClientInterface, projectId, region, networkId string) *wait.AsyncActionHandler[iaasalpha.Network] {
|
|
handler := wait.New(func() (waitFinished bool, response *iaasalpha.Network, err error) {
|
|
network, err := a.GetNetworkExecute(ctx, projectId, region, networkId)
|
|
if err != nil {
|
|
return false, network, err
|
|
}
|
|
if network.Id == nil || network.Status == nil {
|
|
return false, network, fmt.Errorf("update failed for network with id %s, the response is not valid: the id or the state are missing", networkId)
|
|
}
|
|
// The state returns to "CREATED" after a successful update is completed
|
|
if *network.Id == networkId && *network.Status == CreateSuccess {
|
|
return true, network, nil
|
|
}
|
|
return false, network, nil
|
|
})
|
|
handler.SetSleepBeforeWait(2 * time.Second)
|
|
handler.SetTimeout(15 * time.Minute)
|
|
return handler
|
|
}
|
|
|
|
// DeleteNetworkWaitHandler will wait for network deletion
|
|
func DeleteNetworkWaitHandler(ctx context.Context, a APIClientInterface, projectId, region, networkId string) *wait.AsyncActionHandler[iaasalpha.Network] {
|
|
handler := wait.New(func() (waitFinished bool, response *iaasalpha.Network, err error) {
|
|
network, err := a.GetNetworkExecute(ctx, projectId, region, networkId)
|
|
if err == nil {
|
|
return false, nil, nil
|
|
}
|
|
var oapiErr *oapierror.GenericOpenAPIError
|
|
ok := errors.As(err, &oapiErr)
|
|
if !ok {
|
|
return false, network, fmt.Errorf("could not convert error to oapierror.GenericOpenAPIError: %w", err)
|
|
}
|
|
if oapiErr.StatusCode != http.StatusNotFound {
|
|
return false, network, err
|
|
}
|
|
return true, nil, nil
|
|
})
|
|
handler.SetTimeout(15 * time.Minute)
|
|
return handler
|
|
}
|