mirror of
https://github.com/sablierapp/sablier.git
synced 2025-12-27 15:41:41 +01:00
b1e8fc2f20
* refactor: move app/theme to pkg/theme * refactor * wip * use dockerindocker * add providertest * wip * wip * test(docker): get state now uses dind container to test against a real provider * test(docker): move to docker_test package * refactor(docker): create container_inspect.go * test(docker): add more dind test * test(docker): event test now use docker in docker * refactor: remove unused instance type props * refactor test docker * fix instance list sort * stabilize test * remove testcontainers custom config
101 lines
2.5 KiB
Go
101 lines
2.5 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"context"
|
|
"github.com/sablierapp/sablier/app/discovery"
|
|
"github.com/sablierapp/sablier/app/types"
|
|
"github.com/sablierapp/sablier/pkg/provider"
|
|
v1 "k8s.io/api/apps/v1"
|
|
core_v1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"strings"
|
|
)
|
|
|
|
func (p *KubernetesProvider) InstanceList(ctx context.Context, options provider.InstanceListOptions) ([]types.Instance, error) {
|
|
deployments, err := p.deploymentList(ctx, options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
statefulSets, err := p.statefulSetList(ctx, options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return append(deployments, statefulSets...), nil
|
|
}
|
|
|
|
func (p *KubernetesProvider) deploymentList(ctx context.Context, options provider.InstanceListOptions) ([]types.Instance, error) {
|
|
deployments, err := p.Client.AppsV1().Deployments(core_v1.NamespaceAll).List(ctx, metav1.ListOptions{
|
|
LabelSelector: strings.Join(options.Labels, ","),
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
instances := make([]types.Instance, 0, len(deployments.Items))
|
|
for _, d := range deployments.Items {
|
|
instance := p.deploymentToInstance(d)
|
|
instances = append(instances, instance)
|
|
}
|
|
|
|
return instances, nil
|
|
}
|
|
|
|
func (p *KubernetesProvider) deploymentToInstance(d v1.Deployment) types.Instance {
|
|
var group string
|
|
|
|
if _, ok := d.Labels[discovery.LabelEnable]; ok {
|
|
if g, ok := d.Labels[discovery.LabelGroup]; ok {
|
|
group = g
|
|
} else {
|
|
group = discovery.LabelGroupDefaultValue
|
|
}
|
|
}
|
|
|
|
parsed := DeploymentName(d, ParseOptions{Delimiter: p.delimiter})
|
|
|
|
return types.Instance{
|
|
Name: parsed.Original,
|
|
Group: group,
|
|
}
|
|
}
|
|
|
|
func (p *KubernetesProvider) statefulSetList(ctx context.Context, options provider.InstanceListOptions) ([]types.Instance, error) {
|
|
statefulSets, err := p.Client.AppsV1().StatefulSets(core_v1.NamespaceAll).List(ctx, metav1.ListOptions{
|
|
LabelSelector: strings.Join(options.Labels, ","),
|
|
})
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
instances := make([]types.Instance, 0, len(statefulSets.Items))
|
|
for _, ss := range statefulSets.Items {
|
|
instance := p.statefulSetToInstance(ss)
|
|
instances = append(instances, instance)
|
|
}
|
|
|
|
return instances, nil
|
|
}
|
|
|
|
func (p *KubernetesProvider) statefulSetToInstance(ss v1.StatefulSet) types.Instance {
|
|
var group string
|
|
|
|
if _, ok := ss.Labels[discovery.LabelEnable]; ok {
|
|
if g, ok := ss.Labels[discovery.LabelGroup]; ok {
|
|
group = g
|
|
} else {
|
|
group = discovery.LabelGroupDefaultValue
|
|
}
|
|
}
|
|
|
|
parsed := StatefulSetName(ss, ParseOptions{Delimiter: p.delimiter})
|
|
|
|
return types.Instance{
|
|
Name: parsed.Original,
|
|
Group: group,
|
|
}
|
|
}
|