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) }