Add peer fields to replace inlined non proto-message Kubernetes objects

Author: liggitt

cluster-autoscaler/cloudprovider/externalgrpc/examples/external-grpc-cloud-provider-service/wrapper/wrapper.go

@@ -24,6 +24,7 @@ import (
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
 	"google.golang.org/protobuf/types/known/anypb"
+
 	apiv1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/autoscaler/cluster-autoscaler/cloudprovider"
@@ -148,9 +149,38 @@ func (w *Wrapper) PricingPodPrice(_ context.Context, req *protos.PricingPodPrice
 		}
 		return nil, err
 	}
-	reqPod := req.GetPod()
-	reqStartTime := req.GetStartTime()
-	reqEndTime := req.GetEndTime()
+
+	var reqPod *apiv1.Pod
+	if podBytes := req.GetPodBytes(); podBytes != nil {
+		// decode from opaque bytes into pod if set
+		pod := &apiv1.Pod{}
+		if err := pod.Unmarshal(podBytes); err != nil {
+			return nil, err
+		}
+		reqPod = pod
+	} else {
+		// otherwise fallback to reading inlined pod
+		reqPod = req.GetPod()
+	}
+
+	var reqStartTime *metav1.Time
+	if startTimestamp := req.GetStartTimestamp(); startTimestamp != nil {
+		// read standard protobuf timestamp if set
+		reqStartTime = &metav1.Time{Time: startTimestamp.AsTime()}
+	} else {
+		// otherwise fallback to reading metav1.Time
+		reqStartTime = req.GetStartTime()
+	}
+
+	var reqEndTime *metav1.Time
+	if endTimestamp := req.GetEndTimestamp(); endTimestamp != nil {
+		// read standard protobuf timestamp if set
+		reqEndTime = &metav1.Time{Time: endTimestamp.AsTime()}
+	} else {
+		// otherwise fallback to reading metav1.Time
+		reqEndTime = req.GetEndTime()
+	}
+
 	if reqPod == nil || reqStartTime == nil || reqEndTime == nil {
 		return nil, fmt.Errorf("request fields were nil")
 	}

cluster-autoscaler/cloudprovider/externalgrpc/externalgrpc_cloud_provider.go

@@ -30,6 +30,8 @@ import (
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/credentials"
 	"google.golang.org/grpc/status"
+	"google.golang.org/protobuf/types/known/timestamppb"
+
 	apiv1 "k8s.io/api/core/v1"
 	"k8s.io/apimachinery/pkg/api/resource"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
@@ -162,6 +164,9 @@ func (m *pricingModel) NodePrice(node *apiv1.Node, startTime time.Time, endTime
 		Node:      externalGrpcNode(node),
 		StartTime: &start,
 		EndTime:   &end,
+
+		StartTimestamp: timestamppb.New(startTime),
+		EndTimestamp:   timestamppb.New(endTime),
 	})
 	if err != nil {
 		st, ok := status.FromError(err)
@@ -182,10 +187,20 @@ func (m *pricingModel) PodPrice(pod *apiv1.Pod, startTime time.Time, endTime tim
 	klog.V(5).Infof("Performing gRPC call PricingPodPrice for pod %v", pod.Name)
 	start := metav1.NewTime(startTime)
 	end := metav1.NewTime(endTime)
+
+	podBytes, err := pod.Marshal()
+	if err != nil {
+		return 0, err
+	}
+
 	res, err := m.client.PricingPodPrice(ctx, &protos.PricingPodPriceRequest{
 		Pod:       pod,
 		StartTime: &start,
 		EndTime:   &end,
+
+		PodBytes:       podBytes,
+		StartTimestamp: timestamppb.New(startTime),
+		EndTimestamp:   timestamppb.New(endTime),
 	})
 	if err != nil {
 		st, ok := status.FromError(err)