APIGOV-28984 Add CustomUnit client code - Init checkin

pkg/agent/agent.go

@@ -169,12 +169,12 @@ func InitializeWithAgentFeatures(centralCfg config.CentralConfig, agentFeaturesC
 	metricServicesConfigs := agentFeaturesCfg.GetMetricServicesConfigs()
 	// iterate over each metric service config
 	for _, config := range metricServicesConfigs {
-		ctx := context.WithValue(context.Background(), ctxLogger, logger)
+		ctx, ctxCancel := context.WithCancel(context.Background())
 		// Initialize custom units client
-		c := customunit.NewCustomUnitMetricReportingClient(ctx, config.URL)
-
-		c.MetricReporting()
+		factory := customunit.NewCustomMetricReportingClientFactory(config.URL, agent.cacheManager)
+		client, _ := factory(ctx, ctxCancel)
 
+		client.MetricReporting()
 	}
 	if !agent.isInitialized {
 		err = handleInitialization()

pkg/agent/handler/accessrequest.go

@@ -147,7 +147,20 @@ func (h *accessRequestHandler) onPending(ctx context.Context, ar *management.Acc
 			h.onError(ctx, ar, err)
 			return ar
 		}
-		errMessage := customunit.QuotaEnforcementInfo(metricServicesConfigs, ctx, quotaInfo)
+		errMessage := ""
+		for _, config := range metricServicesConfigs {
+			if config.MetricServiceEnabled() {
+				factory := customunit.NewQuotaEnforcementClientFactory(config.URL, quotaInfo)
+				client, _ := factory(ctx)
+				response, err := client.QuotaEnforcementInfo()
+				if err == nil {
+					// if error from QE and reject on fail, we return the error back to the central
+					if response.Error != "" && config.RejectOnFailEnabled() {
+						errMessage = errMessage + fmt.Sprintf("TODO: message: %s", err.Error())
+					}
+				}
+			}
+		}
 
 		if errMessage != "" {
 			status = prov.NewRequestStatusBuilder().