1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/*
* Copyright (c) 2017 Cisco and/or its affiliates.
*
* 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 io.fd.honeycomb.benchmark.memory;
import com.google.common.collect.ImmutableSet;
import com.google.inject.Injector;
import com.google.inject.Module;
import io.fd.honeycomb.benchmark.memory.config.BindableCfgAttrsModule;
import io.fd.honeycomb.benchmark.memory.config.StaticHoneycombManagementModule;
import io.fd.honeycomb.benchmark.memory.write.NoopWritersModule;
import io.fd.honeycomb.infra.distro.Main;
import io.fd.honeycomb.infra.distro.cfgattrs.HoneycombConfiguration;
import io.fd.honeycomb.infra.distro.data.ConfigAndOperationalPipelineModule;
import io.fd.honeycomb.infra.distro.data.context.ContextPipelineModule;
import io.fd.honeycomb.infra.distro.initializer.InitializerPipelineModule;
import io.fd.honeycomb.infra.distro.netconf.NetconfModule;
import io.fd.honeycomb.infra.distro.netconf.NetconfReadersModule;
import io.fd.honeycomb.infra.distro.restconf.RestconfModule;
import io.fd.honeycomb.infra.distro.schema.SchemaModule;
import io.fd.honeycomb.infra.distro.schema.YangBindingProviderModule;
import io.fd.honeycomb.management.jmx.JMXBeanProvider;
import org.eclipse.jetty.server.Server;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.annotation.Nonnull;
import javax.management.openmbean.CompositeDataSupport;
import javax.management.remote.JMXConnector;
import javax.management.remote.JMXServiceURL;
import java.util.Set;
/**
* Measure memory consumption of config data storage
*/
public class MemoryFootprintBenchmark implements JMXBeanProvider, BenchmarkFilesProvider {
private static final Logger LOG = LoggerFactory.getLogger(MemoryFootprintBenchmark.class);
/**
* All modules from infra to load.Not static to not persist state
*/
private final Set<Module> BASE_MODULES = ImmutableSet.of(
new YangBindingProviderModule(),
new SchemaModule(),
new ConfigAndOperationalPipelineModule(),
new ContextPipelineModule(),
new InitializerPipelineModule(),
new NetconfModule(),
new NetconfReadersModule(),
new RestconfModule(),
// to enable jmx
new StaticHoneycombManagementModule(),
//adds noop writers
new NoopWritersModule());
// configuration class used to run benchmark, allows us to switch between honeycomb with data, or on rest
private final HoneycombConfiguration configuration;
// output file path
private final String outputPath;
public MemoryFootprintBenchmark(@Nonnull final HoneycombConfiguration configuration, @Nonnull final String outputPath) {
this.configuration = configuration;
this.outputPath = outputPath;
}
public void run() throws Exception {
// start honeycomb with configuration of BASE_MODULES + configuration class
final Injector injector = startHoneycomb();
// query memory beans with JMX and output results on output path
queryMemoryBeans(injector.getInstance(JMXServiceURL.class))
.forEach(memoryInfo -> outputBenchmarkResult(memoryInfo, outputPath, LOG));
// shutdowns server instance
injector.getInstance(Server.class).stop();
}
/**
* start honeycomb with basic modules + provided static configuration
*/
private Injector startHoneycomb() {
LOG.info("Starting embedded server with configuration {}", configuration);
return Main.init(ImmutableSet.<Module>builder()
.add(new BindableCfgAttrsModule(configuration))
.addAll(BASE_MODULES).build());
}
/**
* Queries heap and non-heap memory usage
*/
private Set<MemoryInfo> queryMemoryBeans(final JMXServiceURL url) {
LOG.info("Requesting memory bean on url {}", url);
try (final JMXConnector connector = getConnector(url)) {
MemoryInfo heapMemoryInfo = new MemoryInfo(
(CompositeDataSupport) getJMXAttribute(connector, MemoryInfo.MEMORY_MBEAN_TYPE,
MemoryInfo.HEAP_MEMORY), MemoryInfo.HEAP_MEMORY);
LOG.info("Heap memory usage {}", heapMemoryInfo);
MemoryInfo nonHeapMemoryInfo = new MemoryInfo(
(CompositeDataSupport) getJMXAttribute(connector, MemoryInfo.MEMORY_MBEAN_TYPE,
MemoryInfo.NON_HEAP_MEMORY), MemoryInfo.NON_HEAP_MEMORY);
LOG.info("NonHeap memory usage {}", nonHeapMemoryInfo);
return ImmutableSet.of(heapMemoryInfo, nonHeapMemoryInfo);
} catch (Exception e) {
throw new IllegalStateException("Unable to query memory beans", e);
}
}
}
|