|
10 | 10 | import com.navercorp.pinpoint.metric.collector.CollectorTypeParser;
|
11 | 11 | import com.navercorp.pinpoint.metric.collector.MetricCollectorApp;
|
12 | 12 | import com.navercorp.pinpoint.metric.collector.TypeSet;
|
| 13 | +import com.navercorp.pinpoint.uristat.collector.UriStatCollectorConfig; |
13 | 14 | import org.springframework.boot.Banner;
|
14 | 15 | import org.springframework.boot.SpringBootConfiguration;
|
15 | 16 | import org.springframework.boot.WebApplicationType;
|
@@ -43,20 +44,20 @@ public static void main(String[] args) {
|
43 | 44 |
|
44 | 45 | if (types.hasType(CollectorType.BASIC)) {
|
45 | 46 | logger.info(String.format("Start %s collector", CollectorType.BASIC));
|
46 |
| - SpringApplicationBuilder collectorAppBuilder = createAppBuilder(builder, BasicCollectorApp.class, 15400); |
| 47 | + SpringApplicationBuilder collectorAppBuilder = createAppBuilder(builder, 15400, BasicCollectorApp.class, UriStatCollectorConfig.class); |
47 | 48 | collectorAppBuilder.build().run(args);
|
48 | 49 | }
|
49 | 50 |
|
50 | 51 | if (types.hasType(CollectorType.METRIC)) {
|
51 | 52 | logger.info(String.format("Start %s collector", CollectorType.METRIC));
|
52 |
| - SpringApplicationBuilder metricAppBuilder = createAppBuilder(builder, MetricCollectorApp.class, 15200); |
| 53 | + SpringApplicationBuilder metricAppBuilder = createAppBuilder(builder, 15200, MetricCollectorApp.class); |
53 | 54 | metricAppBuilder.listeners(new AdditionalProfileListener("metric"));
|
54 | 55 | metricAppBuilder.build().run(args);
|
55 | 56 | }
|
56 | 57 | }
|
57 | 58 |
|
58 | 59 |
|
59 |
| - private static SpringApplicationBuilder createAppBuilder(SpringApplicationBuilder builder, Class<?> appClass, int port) { |
| 60 | + private static SpringApplicationBuilder createAppBuilder(SpringApplicationBuilder builder, int port, Class<?>... appClass) { |
60 | 61 | SpringApplicationBuilder appBuilder = builder.child(appClass)
|
61 | 62 | .web(WebApplicationType.SERVLET)
|
62 | 63 | .bannerMode(Banner.Mode.OFF)
|
|
0 commit comments