-
Notifications
You must be signed in to change notification settings - Fork 0
/
job-deployer.go
62 lines (53 loc) · 1.84 KB
/
job-deployer.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
57
58
59
60
61
62
// Copyright (c) 2024 Benjamin Borbe All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package k8s
import (
"context"
"github.com/bborbe/errors"
"github.com/golang/glog"
batchv1 "k8s.io/api/batch/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/kubernetes"
)
var foreground = metav1.DeletePropagationForeground
//counterfeiter:generate -o mocks/k8s-job-deployer.go --fake-name K8sJobDeployer . JobDeployer
type JobDeployer interface {
Deploy(ctx context.Context, job batchv1.Job) error
Undeploy(ctx context.Context, namespace Namespace, name string) error
}
func NewJobDeployer(
clientset kubernetes.Interface,
) JobDeployer {
return &jobDeployer{
clientset: clientset,
}
}
type jobDeployer struct {
clientset kubernetes.Interface
}
func (s *jobDeployer) Deploy(ctx context.Context, job batchv1.Job) error {
glog.V(3).Infof("deploy %s started", job.Name)
if err := s.Undeploy(ctx, Namespace(job.Namespace), job.Name); err != nil {
return errors.Wrap(ctx, err, "undeploy failed")
}
if _, err := s.clientset.BatchV1().Jobs(job.Namespace).Create(ctx, &job, metav1.CreateOptions{}); err != nil {
return errors.Wrap(ctx, err, "create job failed")
}
glog.V(3).Infof("job %s created successful", job.Name)
return nil
}
func (s *jobDeployer) Undeploy(ctx context.Context, namespace Namespace, name string) error {
glog.V(3).Infof("delete %s started", name)
if _, err := s.clientset.BatchV1().Jobs(namespace.String()).Get(ctx, name, metav1.GetOptions{}); err != nil {
glog.V(4).Infof("job '%s' not found => skip", name)
return nil
}
if err := s.clientset.BatchV1().Jobs(namespace.String()).Delete(ctx, name, metav1.DeleteOptions{
PropagationPolicy: &foreground,
}); err != nil {
return err
}
glog.V(3).Infof("delete %s completed", name)
return nil
}