mirror of
https://github.com/vmware-tanzu/velero.git
synced 2026-01-07 05:46:37 +00:00
4543258970
* add pkg/builder with BackupBuilder and ObjectMeta functional options Signed-off-by: Steve Kriss <krisss@vmware.com>
98 lines
2.9 KiB
Go
98 lines
2.9 KiB
Go
/*
|
|
Copyright 2019 the Velero contributors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package serverstatus
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/watch"
|
|
|
|
velerov1api "github.com/heptio/velero/pkg/apis/velero/v1"
|
|
"github.com/heptio/velero/pkg/builder"
|
|
velerov1client "github.com/heptio/velero/pkg/generated/clientset/versioned/typed/velero/v1"
|
|
)
|
|
|
|
type ServerStatusGetter interface {
|
|
GetServerStatus(client velerov1client.ServerStatusRequestsGetter) (*velerov1api.ServerStatusRequest, error)
|
|
}
|
|
|
|
type DefaultServerStatusGetter struct {
|
|
Namespace string
|
|
Timeout time.Duration
|
|
}
|
|
|
|
func (g *DefaultServerStatusGetter) GetServerStatus(client velerov1client.ServerStatusRequestsGetter) (*velerov1api.ServerStatusRequest, error) {
|
|
req := builder.ForServerStatusRequest(g.Namespace, "").
|
|
ObjectMeta(
|
|
builder.WithGenerateName("velero-cli-"),
|
|
).Result()
|
|
|
|
created, err := client.ServerStatusRequests(g.Namespace).Create(req)
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
defer client.ServerStatusRequests(g.Namespace).Delete(created.Name, nil)
|
|
|
|
listOptions := metav1.ListOptions{
|
|
// TODO: once the minimum supported Kubernetes version is v1.9.0, uncomment the following line.
|
|
// See http://issue.k8s.io/51046 for details.
|
|
//FieldSelector: "metadata.name=" + req.Name
|
|
ResourceVersion: created.ResourceVersion,
|
|
}
|
|
watcher, err := client.ServerStatusRequests(g.Namespace).Watch(listOptions)
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
defer watcher.Stop()
|
|
|
|
expired := time.NewTimer(g.Timeout)
|
|
defer expired.Stop()
|
|
|
|
Loop:
|
|
for {
|
|
select {
|
|
case <-expired.C:
|
|
return nil, errors.New("timed out waiting for server status request to be processed")
|
|
case e := <-watcher.ResultChan():
|
|
updated, ok := e.Object.(*velerov1api.ServerStatusRequest)
|
|
if !ok {
|
|
return nil, errors.Errorf("unexpected type %T", e.Object)
|
|
}
|
|
|
|
// TODO: once the minimum supported Kubernetes version is v1.9.0, remove the following check.
|
|
// See http://issue.k8s.io/51046 for details.
|
|
if updated.Name != created.Name {
|
|
continue
|
|
}
|
|
|
|
switch e.Type {
|
|
case watch.Deleted:
|
|
return nil, errors.New("server status request was unexpectedly deleted")
|
|
case watch.Modified:
|
|
if updated.Status.Phase == velerov1api.ServerStatusRequestPhaseProcessed {
|
|
req = updated
|
|
break Loop
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return req, nil
|
|
}
|