@@ -30,6 +30,7 @@ import (
30
30
healthpb "google.golang.org/grpc/health/grpc_health_v1"
31
31
"google.golang.org/grpc/reflection"
32
32
"google.golang.org/grpc/status"
33
+ "google.golang.org/protobuf/types/known/emptypb"
33
34
34
35
"github.com/aws/amazon-vpc-cni-k8s/pkg/ipamd/datastore"
35
36
"github.com/aws/amazon-vpc-cni-k8s/pkg/networkutils"
@@ -320,6 +321,18 @@ func (s *server) DelNetwork(ctx context.Context, in *rpc.DelNetworkRequest) (*rp
320
321
return & rpc.DelNetworkReply {Success : err == nil , IPv4Addr : ipv4Addr , IPv6Addr : ipv6Addr , DeviceNumber : int32 (deviceNumber )}, err
321
322
}
322
323
324
+ func (s * server ) GetNetworkPolicyConfigs (ctx context.Context , e * emptypb.Empty ) (* rpc.NetworkPolicyAgentConfigReply , error ) {
325
+
326
+ log .Infof ("Received request for Network Policy Agent configs" )
327
+
328
+ resp := & rpc.NetworkPolicyAgentConfigReply {
329
+ NetworkPolicyMode : s .ipamContext .networkPolicyMode ,
330
+ }
331
+
332
+ log .Infof ("Send NetworkPolicyAgentConfigReply: NetworkPolicyMode: %v" , resp .NetworkPolicyMode )
333
+ return resp , nil
334
+ }
335
+
323
336
// RunRPCHandler handles request from gRPC
324
337
func (c * IPAMContext ) RunRPCHandler (version string ) error {
325
338
log .Infof ("Serving RPC Handler version %s on %s" , version , ipamdgRPCaddress )
@@ -330,6 +343,7 @@ func (c *IPAMContext) RunRPCHandler(version string) error {
330
343
}
331
344
grpcServer := grpc .NewServer ()
332
345
rpc .RegisterCNIBackendServer (grpcServer , & server {version : version , ipamContext : c })
346
+ rpc .RegisterConfigServerBackendServer (grpcServer , & server {version : version , ipamContext : c })
333
347
healthServer := health .NewServer ()
334
348
// If ipamd can talk to the API server and to the EC2 API, the pod is healthy.
335
349
// No need to ever change this to HealthCheckResponse_NOT_SERVING since it's a local service only
0 commit comments