Skip to content

Commit 7d54bc5

Browse files
committed
Parse optional category from default labels in extractor_info.json
1 parent 883d9bd commit 7d54bc5

File tree

2 files changed

+18
-6
lines changed

2 files changed

+18
-6
lines changed

app/api/Extractions.scala

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -477,11 +477,17 @@ class Extractions @Inject()(
477477
extractors.updateExtractorInfo(info) match {
478478
case Some(u) => {
479479
// Create/assign any default labels for this extractor
480-
u.defaultLabels.foreach(labelName => {
480+
u.defaultLabels.foreach(labelStr => {
481+
val segments = labelStr.split("/")
482+
val (labelName, labelCategory) = if (segments.length > 1) {
483+
(segments(1), segments(0))
484+
} else {
485+
(segments(0), "Other")
486+
}
481487
extractors.getExtractorsLabel(labelName) match {
482488
case None => {
483-
// Label does not exist - create then assign it
484-
val createdLabel = extractors.createExtractorsLabel(labelName, Some("Default"), List[String](u.name))
489+
// Label does not exist - create and assign it
490+
val createdLabel = extractors.createExtractorsLabel(labelName, Some(labelCategory), List[String](u.name))
485491
}
486492
case Some(lbl) => {
487493
// Label already exists, assign it

app/services/RabbitmqPlugin.scala

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1256,11 +1256,17 @@ class ExtractorsHeartbeats(channel: Channel, queue: String) extends Actor {
12561256
case None => {}
12571257
case Some(eInfo) => {
12581258
// Create (if needed) and assign default labels
1259-
eInfo.defaultLabels.foreach(labelName => {
1259+
eInfo.defaultLabels.foreach(labelStr => {
1260+
val segments = labelStr.split("/")
1261+
val (labelName, labelCategory) = if (segments.length > 1) {
1262+
(segments(1), segments(0))
1263+
} else {
1264+
(segments(0), "Other")
1265+
}
12601266
extractorsService.getExtractorsLabel(labelName) match {
12611267
case None => {
1262-
// Label does not exist - create then assign it
1263-
val createdLabel = extractorsService.createExtractorsLabel(labelName, Some("Default"), List[String](eInfo.name))
1268+
// Label does not exist - create and assign it
1269+
val createdLabel = extractorsService.createExtractorsLabel(labelName, Some(labelCategory), List[String](eInfo.name))
12641270
}
12651271
case Some(lbl) => {
12661272
// Label already exists, assign it

0 commit comments

Comments
 (0)