// Copyright 2020 The Bazel Authors. All rights reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package build.buildfarm.common.metrics; import build.bazel.remote.execution.v2.ExecuteOperationMetadata; import build.bazel.remote.execution.v2.ExecuteResponse; import build.bazel.remote.execution.v2.RequestMetadata; import build.buildfarm.v1test.OperationRequestMetadata; import com.google.common.annotations.VisibleForTesting; import com.google.longrunning.Operation; import com.google.protobuf.InvalidProtocolBufferException; import com.google.protobuf.util.JsonFormat; import com.google.rpc.PreconditionFailure; import java.util.logging.Level; import java.util.logging.Logger; public abstract class AbstractMetricsPublisher implements MetricsPublisher { private static final Logger logger = Logger.getLogger(AbstractMetricsPublisher.class.getName()); private final String clusterId; public AbstractMetricsPublisher(String clusterId) { this.clusterId = clusterId; } public AbstractMetricsPublisher() { this(/* clusterId=*/ null); } @Override public void publishRequestMetadata(Operation operation, RequestMetadata requestMetadata) { throw new UnsupportedOperationException("Not Implemented."); } @Override public abstract void publishMetric(String metricName, Object metricValue); @VisibleForTesting protected OperationRequestMetadata populateRequestMetadata( Operation operation, RequestMetadata requestMetadata) { try { OperationRequestMetadata operationRequestMetadata = OperationRequestMetadata.newBuilder() .setRequestMetadata(requestMetadata) .setOperationName(operation.getName()) .setDone(operation.getDone()) .setClusterId(clusterId) .build(); if (operation.getDone() && operation.getResponse().is(ExecuteResponse.class)) { operationRequestMetadata = operationRequestMetadata .toBuilder() .setExecuteResponse(operation.getResponse().unpack(ExecuteResponse.class)) .build(); } if (operation.getMetadata().is(ExecuteOperationMetadata.class)) { operationRequestMetadata = operationRequestMetadata .toBuilder() .setExecuteOperationMetadata( operation.getMetadata().unpack(ExecuteOperationMetadata.class)) .build(); } return operationRequestMetadata; } catch (Exception e) { logger.log( Level.WARNING, String.format("Could not populate request metadata for %s.", operation.getName()), e); return null; } } protected static String formatRequestMetadataToJson( OperationRequestMetadata operationRequestMetadata) throws InvalidProtocolBufferException { JsonFormat.TypeRegistry typeRegistry = JsonFormat.TypeRegistry.newBuilder() .add(ExecuteResponse.getDescriptor()) .add(ExecuteOperationMetadata.getDescriptor()) .add(PreconditionFailure.getDescriptor()) .build(); String formattedRequestMetadata = JsonFormat.printer() .usingTypeRegistry(typeRegistry) .omittingInsignificantWhitespace() .print(operationRequestMetadata); logger.log(Level.FINE, "{}", formattedRequestMetadata); return formattedRequestMetadata; } }