diff --git a/api/rdb/v1/rdb_utils.go b/api/rdb/v1/rdb_utils.go index e0178206..f30b58af 100644 --- a/api/rdb/v1/rdb_utils.go +++ b/api/rdb/v1/rdb_utils.go @@ -108,3 +108,49 @@ func (s *API) WaitForDatabaseBackup(req *WaitForDatabaseBackupRequest) (*Databas } return backup.(*DatabaseBackup), nil } + +type WaitForInstanceLogRequest struct { + InstanceLogID string + Region scw.Region + Timeout *time.Duration + RetryInterval *time.Duration +} + +// WaitForInstanceLog waits for the instance logs to be in a "terminal state" before returning. +// This function can be used to wait for an instance logs to be ready for example. +func (s *API) WaitForInstanceLog(req *WaitForInstanceLogRequest) (*InstanceLog, error) { + timeout := defaultTimeout + if req.Timeout != nil { + timeout = *req.Timeout + } + retryInterval := defaultRetryInterval + if req.RetryInterval != nil { + retryInterval = *req.RetryInterval + } + + terminalStatus := map[InstanceLogStatus]struct{}{ + InstanceLogStatusReady: {}, + InstanceLogStatusError: {}, + } + + logs, err := async.WaitSync(&async.WaitSyncConfig{ + Get: func() (interface{}, bool, error) { + res, err := s.GetInstanceLog(&GetInstanceLogRequest{ + Region: req.Region, + }) + + if err != nil { + return nil, false, err + } + _, isTerminal := terminalStatus[res.Status] + + return res, isTerminal, nil + }, + Timeout: timeout, + IntervalStrategy: async.LinearIntervalStrategy(retryInterval), + }) + if err != nil { + return nil, errors.Wrap(err, "waiting for instance logs failed") + } + return logs.(*InstanceLog), nil +}