| // Copyright 2016 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 com.google.devtools.build.lib.buildeventstream.transports; |
| |
| import static com.google.common.base.Preconditions.checkNotNull; |
| |
| import com.google.devtools.build.lib.buildeventstream.ArtifactGroupNamer; |
| import com.google.devtools.build.lib.buildeventstream.BuildEvent; |
| import com.google.devtools.build.lib.buildeventstream.BuildEventConverters; |
| import com.google.devtools.build.lib.buildeventstream.BuildEventStreamProtos; |
| import com.google.devtools.build.lib.buildeventstream.BuildEventTransport; |
| import com.google.devtools.build.lib.buildeventstream.PathConverter; |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.util.concurrent.Future; |
| import java.util.logging.Level; |
| import java.util.logging.Logger; |
| |
| /** |
| * A simple {@link BuildEventTransport} that writes a varint delimited binary representation of |
| * {@link BuildEvent} protocol buffers to a file. |
| */ |
| public final class BinaryFormatFileTransport extends FileTransport { |
| |
| private static final Logger logger = Logger.getLogger(BinaryFormatFileTransport.class.getName()); |
| |
| private static final int MAX_VARINT_BYTES = 9; |
| private final PathConverter pathConverter; |
| |
| BinaryFormatFileTransport(String path, PathConverter pathConverter) { |
| super(path); |
| this.pathConverter = pathConverter; |
| } |
| |
| @Override |
| public String name() { |
| return this.getClass().getSimpleName(); |
| } |
| |
| @Override |
| public synchronized void sendBuildEvent(BuildEvent event, final ArtifactGroupNamer namer) { |
| BuildEventConverters converters = |
| new BuildEventConverters() { |
| @Override |
| public PathConverter pathConverter() { |
| return pathConverter; |
| } |
| @Override |
| public ArtifactGroupNamer artifactGroupNamer() { |
| return namer; |
| } |
| }; |
| checkNotNull(event); |
| BuildEventStreamProtos.BuildEvent protoEvent = event.asStreamProto(converters); |
| |
| int maxSerializedSize = MAX_VARINT_BYTES + protoEvent.getSerializedSize(); |
| ByteArrayOutputStream out = new ByteArrayOutputStream(maxSerializedSize); |
| |
| try { |
| protoEvent.writeDelimitedTo(out); |
| writeData(out.toByteArray()); |
| } catch (IOException e) { |
| logger.log(Level.SEVERE, e.getMessage(), e); |
| @SuppressWarnings({"unused", "nullness"}) |
| Future<?> possiblyIgnoredError = close(); |
| } |
| } |
| } |