-
Notifications
You must be signed in to change notification settings - Fork 9
/
readiness.go
56 lines (47 loc) · 1.69 KB
/
readiness.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Copyright 2022 Namespace Labs Inc; All rights reserved.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
package kubernetes
import (
"context"
"fmt"
"net"
"time"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/kubernetes"
"namespacelabs.dev/foundation/framework/kubernetes/kubedef"
"namespacelabs.dev/foundation/framework/kubernetes/kubeobj"
"namespacelabs.dev/foundation/internal/fnerrors"
"namespacelabs.dev/foundation/internal/runtime"
"namespacelabs.dev/foundation/internal/runtime/kubernetes/client"
)
type ServiceReadiness struct {
Ready bool
Message string
}
func AreServicesReady(ctx context.Context, cli *kubernetes.Clientset, namespace string, srv runtime.Deployable) (ServiceReadiness, error) {
if !client.IsInclusterClient(cli) {
return ServiceReadiness{}, fnerrors.InternalError("cannot check service readiness for remote kubernetes cluster")
}
// TODO only check services that are required
services, err := cli.CoreV1().Services(namespace).List(ctx, metav1.ListOptions{
LabelSelector: kubeobj.SerializeSelector(kubedef.SelectById(srv)),
})
if err != nil {
return ServiceReadiness{}, err
}
for _, s := range services.Items {
for _, port := range s.Spec.Ports {
addr := fmt.Sprintf("%s.%s.svc.cluster.local:%d", s.Name, s.Namespace, port.Port)
conn, err := net.DialTimeout("tcp", addr, 100*time.Millisecond)
if err != nil {
return ServiceReadiness{
Ready: false,
Message: fmt.Sprintf("%q not ready: failed to dial %s:%d: %v", srv.GetName(), s.Name, port.Port, err),
}, nil
}
conn.Close()
}
}
return ServiceReadiness{Ready: true}, nil
}