update backend service in compute tcp proxy

gemmahou · Nov 23, 2024 · 799ec31 · 799ec31
1 parent fdb12cb
commit 799ec31
Show file tree

Hide file tree

Showing 21 changed files with 2,785 additions and 155 deletions.
diff --git a/mockgcp/mockcompute/globaltargettcpproxyv1.go b/mockgcp/mockcompute/globaltargettcpproxyv1.go
@@ -63,6 +63,7 @@ func (s *GlobalTargetTcpProxyV1) Insert(ctx context.Context, req *pb.InsertTarge
 	obj.CreationTimestamp = PtrTo(s.nowString())
 	obj.Id = &id
 	obj.Kind = PtrTo("compute#targetTcpProxy")
+
 	if obj.ProxyHeader == nil {
 		obj.ProxyHeader = PtrTo("NONE")
 	}
@@ -107,6 +108,35 @@ func (s *GlobalTargetTcpProxyV1) Delete(ctx context.Context, req *pb.DeleteTarge
 	})
 }
 
+func (s *GlobalTargetTcpProxyV1) SetBackendService(ctx context.Context, req *pb.SetBackendServiceTargetTcpProxyRequest) (*pb.Operation, error) {
+	reqName := "projects/" + req.GetProject() + "/global/targetTcpProxies/" + req.GetTargetTcpProxy()
+	name, err := s.parseTargetTcpProxyName(reqName)
+	if err != nil {
+		return nil, err
+	}
+
+	fqn := name.String()
+
+	obj := &pb.TargetTcpProxy{}
+	if err := s.storage.Get(ctx, fqn, obj); err != nil {
+		return nil, err
+	}
+
+	if err := s.storage.Update(ctx, fqn, obj); err != nil {
+		return nil, err
+	}
+
+	op := &pb.Operation{
+		TargetId:      obj.Id,
+		TargetLink:    obj.SelfLink,
+		OperationType: PtrTo("TargetTcpProxySetBackendService"),
+		User:          PtrTo("[email protected]"),
+	}
+	return s.startGlobalLRO(ctx, name.Project.ID, op, func() (proto.Message, error) {
+		return obj, nil
+	})
+}
+
 func (s *GlobalTargetTcpProxyV1) SetProxyHeader(ctx context.Context, req *pb.SetProxyHeaderTargetTcpProxyRequest) (*pb.Operation, error) {
 	reqName := "projects/" + req.GetProject() + "/global/targetTcpProxies/" + req.GetTargetTcpProxy()
 	name, err := s.parseTargetTcpProxyName(reqName)

diff --git a/pkg/controller/direct/compute/targettcpproxy/targettcpproxy_controller.go b/pkg/controller/direct/compute/targettcpproxy/targettcpproxy_controller.go
@@ -113,6 +113,14 @@ func (m *targetTCPProxyModel) AdapterForObject(ctx context.Context, reader clien
 	}
 	location := parent.Region
 
+	// Handle API/TF default values
+	if obj.Spec.ProxyBind != nil && *obj.Spec.ProxyBind == false {
+		obj.Spec.ProxyBind = nil
+	}
+	if obj.Spec.ProxyHeader == nil {
+		obj.Spec.ProxyHeader = direct.PtrTo("NONE")
+	}
+
 	// Get GCP client
 	if location == "global" {
 		gcpClient, err := m.client(ctx)
@@ -256,42 +264,43 @@ func (a *targetTCPProxyAdapter) Update(ctx context.Context, updateOp *directbase
 	targetTCPProxy.Name = direct.LazyPtr(tokens[len(tokens)-1])
 
 	// Regional API does not support Update
-	if !reflect.DeepEqual(targetTCPProxy.Service, a.actual.Service) && region == "global" {
-		setBackendServiceReq := &computepb.SetBackendServiceTargetTcpProxyRequest{
+	if !reflect.DeepEqual(targetTCPProxy.ProxyHeader, a.actual.ProxyHeader) && region == "global" {
+		setProxyHeaderReq := &computepb.SetProxyHeaderTargetTcpProxyRequest{
 			Project: parent.ProjectID,
-			TargetTcpProxiesSetBackendServiceRequestResource: &computepb.TargetTcpProxiesSetBackendServiceRequest{Service: targetTCPProxy.Service},
+			TargetTcpProxiesSetProxyHeaderRequestResource: &computepb.TargetTcpProxiesSetProxyHeaderRequest{ProxyHeader: targetTCPProxy.ProxyHeader},
 			TargetTcpProxy: tokens[len(tokens)-1],
 		}
-		op, err = a.targetTcpProxiesClient.SetBackendService(ctx, setBackendServiceReq)
+		op, err = a.targetTcpProxiesClient.SetProxyHeader(ctx, setProxyHeaderReq)
 		if err != nil {
-			return fmt.Errorf("updating ComputeTargetTCPProxy backend service %s: %w", a.id.External, err)
+			return fmt.Errorf("updating ComputeTargetTCPProxy proxy header %s: %w", a.id.External, err)
 		}
 		if !op.Done() {
 			err = op.Wait(ctx)
 			if err != nil {
-				return fmt.Errorf("waiting ComputeTargetTCPProxy backend service %s update failed: %w", a.id.External, err)
+				return fmt.Errorf("waiting ComputeTargetTCPProxy proxy header %s update failed: %w", a.id.External, err)
 			}
 		}
-		log.V(2).Info("successfully updated ComputeTargetTCPProxy backend service", "name", a.id.External)
-
+		log.V(2).Info("successfully updated ComputeTargetTCPProxy proxy header", "name", a.id.External)
 	}
-	if !reflect.DeepEqual(targetTCPProxy.ProxyHeader, a.actual.ProxyHeader) && region == "global" {
-		setProxyHeaderReq := &computepb.SetProxyHeaderTargetTcpProxyRequest{
+
+	if !reflect.DeepEqual(targetTCPProxy.Service, a.actual.Service) && region == "global" {
+		setBackendServiceReq := &computepb.SetBackendServiceTargetTcpProxyRequest{
 			Project: parent.ProjectID,
-			TargetTcpProxiesSetProxyHeaderRequestResource: &computepb.TargetTcpProxiesSetProxyHeaderRequest{ProxyHeader: targetTCPProxy.ProxyHeader},
+			TargetTcpProxiesSetBackendServiceRequestResource: &computepb.TargetTcpProxiesSetBackendServiceRequest{Service: targetTCPProxy.Service},
 			TargetTcpProxy: tokens[len(tokens)-1],
 		}
-		op, err = a.targetTcpProxiesClient.SetProxyHeader(ctx, setProxyHeaderReq)
+		op, err = a.targetTcpProxiesClient.SetBackendService(ctx, setBackendServiceReq)
 		if err != nil {
-			return fmt.Errorf("updating ComputeTargetTCPProxy proxy header %s: %w", a.id.External, err)
+			return fmt.Errorf("updating ComputeTargetTCPProxy backend service %s: %w", a.id.External, err)
 		}
 		if !op.Done() {
 			err = op.Wait(ctx)
 			if err != nil {
-				return fmt.Errorf("waiting ComputeTargetTCPProxy proxy header %s update failed: %w", a.id.External, err)
+				return fmt.Errorf("waiting ComputeTargetTCPProxy backend service %s update failed: %w", a.id.External, err)
 			}
 		}
-		log.V(2).Info("successfully updated ComputeTargetTCPProxy proxy header", "name", a.id.External)
+		log.V(2).Info("successfully updated ComputeTargetTCPProxy backend service", "name", a.id.External)
+
 	}
 
 	// Get the updated resource