Skip to content

Commit e840b56

Browse files
committed
name change disableK8sCrdReconcile to startCrdReconcilers
1 parent d328dcb commit e840b56

File tree

5 files changed

+20
-17
lines changed

5 files changed

+20
-17
lines changed

cmd/epp/runner/runner.go

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -204,10 +204,13 @@ func (r *Runner) Run(ctx context.Context) error {
204204
setupLog.Error(err, "Failed to extract GKNN")
205205
return err
206206
}
207-
disableK8sCrdReconcile := *endpointSelector != ""
208-
controllerCfg := runserver.NewControllerConfig(disableK8sCrdReconcile)
209-
ds, err := setupDatastore(setupLog, ctx, epf, int32(opts.ModelServerMetricsPort), disableK8sCrdReconcile,
207+
208+
startCrdReconcilers := *endpointSelector == "" // If endpointSelector is empty, it means it's not in the standalone mode. Then we should start the inferencePool and other CRD Reconciler.
209+
controllerCfg := runserver.NewControllerConfig(startCrdReconcilers)
210+
ds, err := setupDatastore(setupLog, ctx, epf, int32(opts.ModelServerMetricsPort), startCrdReconcilers,
210211
opts.PoolName, opts.PoolNamespace, opts.EndpointSelector, opts.EndpointTargetPorts)
212+
213+
ds, err := setupDatastore(setupLog, ctx, epf, int32(*modelServerMetricsPort), startCrdReconcilers, *poolName, *poolNamespace, *endpointSelector, *endpointTargetPorts)
211214
if err != nil {
212215
setupLog.Error(err, "Failed to setup datastore")
213216
return err
@@ -368,8 +371,8 @@ func (r *Runner) Run(ctx context.Context) error {
368371
}
369372

370373
func setupDatastore(setupLog logr.Logger, ctx context.Context, epFactory datalayer.EndpointFactory, modelServerMetricsPort int32,
371-
disableK8sCrdReconcile bool, namespace, name, endpointSelector string, endpointTargetPorts []int) (datastore.Datastore, error) {
372-
if !disableK8sCrdReconcile {
374+
startCrdReconcilers bool, namespace, name, endpointSelector string, endpointTargetPorts []int) (datastore.Datastore, error) {
375+
if startCrdReconcilers {
373376
return datastore.NewDatastore(ctx, epFactory, modelServerMetricsPort), nil
374377
} else {
375378
endpointPool := datalayer.NewEndpointPool(namespace, name)

pkg/epp/server/controller_config.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,19 +25,19 @@ import (
2525
)
2626

2727
type ControllerConfig struct {
28-
disableK8sCrdReconcile bool
28+
startCrdReconcilers bool
2929
hasInferenceObjective bool
3030
hasInferenceModelRewrites bool
3131
}
3232

33-
func NewControllerConfig(disableK8sCrdReconcile bool) ControllerConfig {
33+
func NewControllerConfig(startCrdReconcilers bool) ControllerConfig {
3434
return ControllerConfig{
35-
disableK8sCrdReconcile: disableK8sCrdReconcile,
35+
startCrdReconcilers: startCrdReconcilers,
3636
}
3737
}
3838

3939
func (cc *ControllerConfig) PopulateControllerConfig(cfg *rest.Config) error {
40-
if cc.disableK8sCrdReconcile {
40+
if !cc.startCrdReconcilers {
4141
return nil
4242
}
4343
dc, err := discovery.NewDiscoveryClientForConfig(cfg)

pkg/epp/server/controller_config_test.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,13 +27,13 @@ import (
2727

2828
func TestNewControllerConfig(t *testing.T) {
2929
c := NewControllerConfig(true)
30-
if !c.disableK8sCrdReconcile {
31-
t.Error("expected disableK8sCrdReconcile to be true")
30+
if !c.startCrdReconcilers {
31+
t.Error("expected startCrdReconcilers to be true")
3232
}
3333

3434
c = NewControllerConfig(false)
35-
if c.disableK8sCrdReconcile {
36-
t.Error("expected disableK8sCrdReconcile to be false")
35+
if c.startCrdReconcilers {
36+
t.Error("expected startCrdReconcilers to be false")
3737
}
3838
}
3939

@@ -96,7 +96,7 @@ func TestPopulateWithDiscovery(t *testing.T) {
9696
}
9797

9898
func TestPopulateControllerConfig_Disable(t *testing.T) {
99-
c := NewControllerConfig(true)
99+
c := NewControllerConfig(false)
100100
err := c.PopulateControllerConfig(nil)
101101
if err != nil {
102102
t.Errorf("expected nil error, got %v", err)

pkg/epp/server/controller_manager.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ func defaultManagerOptions(cfg ControllerConfig, gknn common.GKNN, metricsServer
5959
},
6060
Metrics: metricsServerOptions,
6161
}
62-
if !cfg.disableK8sCrdReconcile {
62+
if cfg.startCrdReconcilers {
6363
if cfg.hasInferenceObjective {
6464
opt.Cache.ByObject[&v1alpha2.InferenceObjective{}] = cache.ByObject{Namespaces: map[string]cache.Config{
6565
gknn.Namespace: {},

pkg/epp/server/runserver.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ func NewDefaultExtProcServerRunner() *ExtProcServerRunner {
8484
return &ExtProcServerRunner{
8585
GrpcPort: opts.GRPCPort,
8686
GKNN: gknn,
87-
ControllerCfg: ControllerConfig{false, true, true},
87+
ControllerCfg: ControllerConfig{true, true, true},
8888
SecureServing: opts.SecureServing,
8989
HealthChecking: opts.HealthChecking,
9090
RefreshPrometheusMetricsInterval: opts.RefreshPrometheusMetricsInterval,
@@ -96,7 +96,7 @@ func NewDefaultExtProcServerRunner() *ExtProcServerRunner {
9696
// SetupWithManager sets up the runner with the given manager.
9797
func (r *ExtProcServerRunner) SetupWithManager(ctx context.Context, mgr ctrl.Manager) error {
9898
// Create the controllers and register them with the manager
99-
if !r.ControllerCfg.disableK8sCrdReconcile {
99+
if r.ControllerCfg.startCrdReconcilers {
100100
if err := (&controller.InferencePoolReconciler{
101101
Datastore: r.Datastore,
102102
Reader: mgr.GetClient(),

0 commit comments

Comments
 (0)