Set installation_source in NodeInfo

Author: bernd

src/main/java/org/graylog/plugins/usagestatistics/collectors/NodeCollector.java

@@ -78,6 +78,7 @@ public class NodeCollector {
     private final ClusterConfigService clusterConfigService;
     private final Set<PluginMetaData> plugins;
     private final long reportIntervalMs;
+    private final String installationSource;
 
     @Inject
     public NodeCollector(NodeId nodeId,
@@ -87,7 +88,8 @@ public NodeCollector(NodeId nodeId,
                          InputService inputService,
                          ClusterConfigService clusterConfigService,
                          Set<PluginMetaData> plugins,
-                         @Named("usage_statistics_report_interval") Duration reportInterval) {
+                         @Named("usage_statistics_report_interval") Duration reportInterval,
+                         @Named("installation_source") String installationSource) {
         this.nodeId = checkNotNull(nodeId);
         this.serverStatus = checkNotNull(serverStatus);
         this.metricRegistry = checkNotNull(metricRegistry);
@@ -96,6 +98,7 @@ public NodeCollector(NodeId nodeId,
         this.clusterConfigService = checkNotNull(clusterConfigService);
         this.plugins = checkNotNull(plugins);
         this.reportIntervalMs = checkNotNull(reportInterval).toMilliseconds();
+        this.installationSource = checkNotNull(installationSource);
     }
 
     public NodeDataSet getNodeDataSet() {
@@ -120,7 +123,8 @@ private NodeInfo buildNodeInfo() {
                 NodeRole.fromCapabilities(serverStatus),
                 Version.CURRENT_CLASSPATH.toString(),
                 buildPluginInfo(plugins),
-                buildOsInfo()
+                buildOsInfo(),
+                installationSource
         );
     }
 

src/main/java/org/graylog/plugins/usagestatistics/dto/NodeInfo.java

@@ -27,8 +27,9 @@ public abstract class NodeInfo {
     public static NodeInfo create(NodeRole role,
                                   String serverVersion,
                                   Set<PluginInfo> plugins,
-                                  Os os) {
-        return new AutoValue_NodeInfo(role, serverVersion, plugins, os);
+                                  Os os,
+                                  String installationSource) {
+        return new AutoValue_NodeInfo(role, serverVersion, plugins, os, installationSource);
     }
 
     @JsonProperty
@@ -42,4 +43,7 @@ public static NodeInfo create(NodeRole role,
 
     @JsonProperty
     public abstract Os os();
+
+    @JsonProperty
+    public abstract String installationSource();
 }