crichardson/locust-operator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

asD
Some checks failed
Tests / Run on Ubuntu (push) Failing after 55s
Details
E2E Tests / Run on Ubuntu (push) Failing after 1m15s
Details
Lint / Run on Ubuntu (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Chris Richardson
2025-05-13 08:01:53 -04:00
parent b03514b861
commit 60f4f65407
4 changed files with 120 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
internal/controller/build.go Normal file
View File

@@ -0,0 +1,16 @@
package controller
import (
"context"
locustCluster "git.lilpenguins.com/crichardson/locust-operator/api/v1alpha1"
"github.com/go-logr/logr"
)
func (r *LocustClusterReconciler) CreateLeaderJob(log logr.Logger, locustCluster *locustCluster.LocustCluster, ctx context.Context) error {
return nil
}
func (r *LocustClusterReconciler) CreateLeaderService(log logr.Logger, locustCluster *locustCluster.LocustCluster,
ctx context.Context) error {
return nil
}

56
internal/controller/common.go
View File

@@ -1,6 +1,56 @@
package controller
import "sigs.k8s.io/controller-runtime/pkg/client"
import (
"fmt"
"sigs.k8s.io/controller-runtime/pkg/client"
"time"
)
const (
MaxConcurrentReconciles = 20
NamespaceDefault = "pie-coordination-dev"
TestEnvCheck = "TEST_ENV_CHECK"
KubernetesClusterEnv = "KUBERNETES_CLUSTER"
LocustOperatorVersionDesc = "Locust Operator Version"
LeaderContainer = "locustio/locust:2.37.1"
LeaderTestFile = "/usr/local/locust/dummy.py"
LeaderPortNameWeb = "locust-web"
LeaderPortProtocol = "TCP"
LeaderPortWeb = 8089
LeaderPortNameLocust1 = "listener1"
LeaderPortListener1 = 5557
LeaderPortNameLocust2 = "listener2"
LeaderPortListener2 = 5558
LeaderMaxCPU = "0.5"
LeaderMaxMemory = "1Gi"
LeaderMinCPU = "0.25"
LeaderMinMemory = "512Mi"
LeaderBuildParamConfig = "/usr/local/etc/locust_build_params"
MosaicContainer = "docker.apple.com/telemetry/mosaic-agent:2.19.1"
MosaicPortNameHttp = "mosaic-http"
MosaicPortHttp = 8080
MosaicPortProtocol = "TCP"
MosaicVolumeConfigPath = "/usr/local/mosaic/conf.d"
MosaicEndpointSuffix = "PROMETHEUS"
GrafanaEndpoint = "playground-fuji.grafana.telemetry.g.apple.com"
WorkerMaxCPU = "2"
WorkerMinCPU = "1"
WorkerMaxMemory = "3Gi"
WorkerMinMemory = "1Gi"
WorkerMaxStorage = "20Gi"
WorkerMinStorage = "10Gi"
SDRIdentName = "identity-certs"
CertsMount = "/certs/"
TimeoutDefault = time.Second * 20
TimeoutInterval = time.Millisecond * 250
SUCCESS bool = true
FAILURE bool = false
KeyMosaic = "mosaic"
KeyLeader = "leader"
KeyWorker = "worker"
KeyOperator = "operator"
KeyDB = "locust-db"
)
func BuildObjectKey(namespace string, name string) client.ObjectKey {
return client.ObjectKey{
@@ -8,3 +58,7 @@ func BuildObjectKey(namespace string, name string) client.ObjectKey {
Name: name,
}
}
func BuildName(name string, suffix string) string {
return fmt.Sprintf("%s-%s", name, suffix)
}

23
internal/controller/locustcluster_controller.go
View File

@@ -48,24 +48,31 @@ type LocustClusterReconciler struct {
// For more details, check Reconcile and its Result here:
// - https://pkg.go.dev/sigs.k8s.io/controller-runtime@v0.20.4/pkg/reconcile
func (r *LocustClusterReconciler) Reconcile(ctx context.Context, req ctrl.Request) (ctrl.Result, error) {
_ = logf.FromContext(ctx)
log := logf.FromContext(ctx).WithValues()
logf.Log.Info("Looking up LocustCluster", "LocustClusterName", req.Name,
"Namespace", req.Namespace)
log.Info("Looking up LocustCluster")
// logf.Log.Info("Looking up LocustCluster", "LocustClusterName", req.Name,
// "Namespace", req.Namespace)
// locustCluster, err := GetLocustCluster(r.Client, ctx, req.Namespace, req.Name)
_, err := GetLocustCluster(r.Client, ctx, req.Namespace, req.Name)
locustCluster, err := GetLocustCluster(r.Client, ctx, req.Namespace, req.Name)
if err != nil {
if apierrors.IsNotFound(err) {
logf.Log.Info("LocustCluster not found", "LocustClusterName", req.Name,
"Namespace", req.Namespace)
log.Info("LocustCluster not found")
// logf.Log.Info("LocustCluster not found", "LocustClusterName", req.Name,
// "Namespace", req.Namespace)
return ctrl.Result{}, nil
}
logf.Log.Error(err, "Failed to lookup LocustCluster", "LocustClusterName", req.Name,
"Namespace", req.Namespace)
log.Error(err, "LocustCluster not found")
// logf.Log.Error(err, "Failed to lookup LocustCluster", "LocustClusterName", req.Name,
// "Namespace", req.Namespace)
return ctrl.Result{Requeue: true}, err
}
podList, err := r.IsLeaderUp(log, locustCluster, ctx)
r.CreateUpdateLeader(log, locustCluster, ctx, podList)
return ctrl.Result{}, nil
}

35
internal/controller/sync.go
View File

@@ -2,8 +2,9 @@ package controller
import (
"context"
locustCluster "git.lilpenguins.com/crichardson/locust-operator/api/v1alpha1"
"github.com/go-logr/logr"
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
"sigs.k8s.io/controller-runtime/pkg/client"
)
@@ -16,3 +17,35 @@ func GetLocustCluster(client client.Client, ctx context.Context, namespace strin
}
return cluster, nil
}
func (r *LocustClusterReconciler) IsLeaderUp(log logr.Logger, locustCluster *locustCluster.LocustCluster, ctx context.Context) (*unstructured.UnstructuredList, error) {
podkind := &unstructured.UnstructuredList{}
podkind.SetKind("pod")
podkind.SetAPIVersion("v1")
searchLabels := map[string]string{
"job-name": BuildName(locustCluster.GetName(), KeyLeader),
}
err := r.List(ctx, podkind, client.InNamespace(locustCluster.Namespace), client.MatchingLabels(searchLabels))
if err != nil {
log.Info("Leader not found")
return nil, err
}
return podkind, nil
}
func (r *LocustClusterReconciler) CreateUpdateLeader(log logr.Logger, locustCluster *locustCluster.LocustCluster, ctx context.Context,
podList *unstructured.UnstructuredList) error {
var err error
if len(podList.Items) == 0 {
if err = r.CreateLeaderJob(log, locustCluster, ctx); err != nil {
log.Error(err, "Failed to create leader")
return err
}
if err = r.CreateLeaderService(log, locustCluster, ctx); err != nil {
log.Error(err, "Failed to create leader service")
return err
}
}
return err
}