From da87a8e2de54b23827a933839e528762ec54cd50 Mon Sep 17 00:00:00 2001
From: genofire <geno+dev@fireorbit.de>
Date: Sat, 20 Jul 2024 00:33:16 +0200
Subject: [PATCH] fix(database/influxdb*): add type to link

---
 database/influxdb/link.go      | 1 +
 database/influxdb/node_test.go | 1 +
 database/influxdb2/link.go     | 3 ++-
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/database/influxdb/link.go b/database/influxdb/link.go
index 37c2bd9..3ccbebc 100644
--- a/database/influxdb/link.go
+++ b/database/influxdb/link.go
@@ -14,6 +14,7 @@ func (conn *Connection) InsertLink(link *runtime.Link, t time.Time) {
 	tags.SetString("source.addr", link.SourceAddress)
 	tags.SetString("target.id", link.TargetID)
 	tags.SetString("target.addr", link.TargetAddress)
+	tags.SetString("type", link.Type.String())
 	if link.SourceHostname != "" {
 		tags.SetString("source.hostname", link.SourceHostname)
 	}
diff --git a/database/influxdb/node_test.go b/database/influxdb/node_test.go
index 1a3683a..5580ce4 100644
--- a/database/influxdb/node_test.go
+++ b/database/influxdb/node_test.go
@@ -187,6 +187,7 @@ func TestToInflux(t *testing.T) {
 		"source.addr": "a-interface-mac",
 		"target.id":   "foobar",
 		"target.addr": "BAFF1E5",
+		"type":        "vpn",
 	}, tags)
 	assert.EqualValues(80, fields["tq"])
 
diff --git a/database/influxdb2/link.go b/database/influxdb2/link.go
index 4dd36d0..a28f570 100644
--- a/database/influxdb2/link.go
+++ b/database/influxdb2/link.go
@@ -19,7 +19,8 @@ func (conn *Connection) InsertLink(link *runtime.Link, t time.Time) {
 		AddTag("source.id", link.SourceID).
 		AddTag("source.addr", link.SourceAddress).
 		AddTag("target.id", link.TargetID).
-		AddTag("target.addr", link.TargetAddress)
+		AddTag("target.addr", link.TargetAddress).
+		AddTag("type", link.Type.String())
 	if link.SourceHostname != "" {
 		p.AddTag("source.hostname", link.SourceHostname)
 	}