1/*
2 * Copyright (C) 2013 Google Inc.
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.google.caliper.runner;
18
19import com.google.caliper.bridge.AbstractLogMessageVisitor;
20import com.google.caliper.bridge.FailureLogMessage;
21import com.google.caliper.bridge.VmOptionLogMessage;
22import com.google.caliper.bridge.VmPropertiesLogMessage;
23import com.google.caliper.model.VmSpec;
24import com.google.caliper.platform.Platform;
25import com.google.common.base.Optional;
26import com.google.common.collect.ImmutableMap;
27import com.google.common.collect.Maps;
28
29import javax.inject.Inject;
30
31/** An {@link AbstractLogMessageVisitor} that collects data about JVM properties and options. */
32@TrialScoped
33final class VmDataCollectingVisitor extends AbstractLogMessageVisitor {
34  private final ImmutableMap.Builder<String, String> vmOptionsBuilder = ImmutableMap.builder();
35  private final Platform platform;
36  private Optional<ImmutableMap<String, String>> vmProperties = Optional.absent();
37
38  @Inject VmDataCollectingVisitor(Platform platform) {
39    this.platform = platform;
40  }
41
42  /**
43   * Returns a {@link VmSpec} based on the data gathered by this visitor.
44   *
45   * @throws IllegalStateException if not all the data has been gathered.
46   */
47  VmSpec vmSpec() {
48    ImmutableMap<String, String> options = vmOptionsBuilder.build();
49    platform.checkVmProperties(options);
50    return new VmSpec.Builder()
51        .addAllProperties(vmProperties.get())
52        .addAllOptions(options)
53        .build();
54  }
55
56  @Override
57  public void visit(FailureLogMessage logMessage) {
58    throw new ProxyWorkerException(logMessage.stackTrace());
59  }
60
61  @Override
62  public void visit(VmOptionLogMessage logMessage) {
63    vmOptionsBuilder.put(logMessage.name(), logMessage.value());
64  }
65
66  @Override
67  public void visit(VmPropertiesLogMessage logMessage) {
68    vmProperties = Optional.of(ImmutableMap.copyOf(
69        Maps.filterKeys(logMessage.properties(), platform.vmPropertiesToRetain())));
70  }
71}
72