Skip to content

Commit 5a5ed7b

Browse files
committed
filter using server-side namespace filter
1 parent dc95ba5 commit 5a5ed7b

File tree

1 file changed

+25
-1
lines changed

1 file changed

+25
-1
lines changed

cmd/epp/main.go

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,12 +30,15 @@ import (
3030
"go.uber.org/zap/zapcore"
3131
"google.golang.org/grpc"
3232
healthPb "google.golang.org/grpc/health/grpc_health_v1"
33+
corev1 "k8s.io/api/core/v1"
3334
"k8s.io/apimachinery/pkg/runtime"
3435
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
3536
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
3637
"k8s.io/client-go/rest"
3738
"k8s.io/component-base/metrics/legacyregistry"
3839
ctrl "sigs.k8s.io/controller-runtime"
40+
"sigs.k8s.io/controller-runtime/pkg/cache"
41+
"sigs.k8s.io/controller-runtime/pkg/client"
3942
"sigs.k8s.io/controller-runtime/pkg/log/zap"
4043
"sigs.k8s.io/controller-runtime/pkg/manager"
4144
"sigs.k8s.io/controller-runtime/pkg/metrics/filters"
@@ -140,7 +143,28 @@ func run() error {
140143
return err
141144
}
142145

143-
mgr, err := ctrl.NewManager(cfg, ctrl.Options{Scheme: scheme})
146+
mgr, err := ctrl.NewManager(cfg, ctrl.Options{
147+
Scheme: scheme,
148+
Cache: cache.Options{
149+
ByObject: map[client.Object]cache.ByObject{
150+
&corev1.Pod{}: {
151+
Namespaces: map[string]cache.Config{
152+
*poolNamespace: {},
153+
},
154+
},
155+
&v1alpha2.InferencePool{}: {
156+
Namespaces: map[string]cache.Config{
157+
*poolNamespace: {},
158+
},
159+
},
160+
&v1alpha2.InferenceModel{}: {
161+
Namespaces: map[string]cache.Config{
162+
*poolNamespace: {},
163+
},
164+
},
165+
},
166+
},
167+
})
144168
if err != nil {
145169
setupLog.Error(err, "Failed to create controller manager", "config", cfg)
146170
return err

0 commit comments

Comments
 (0)