-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
externalhost.go
49 lines (39 loc) · 1.29 KB
/
externalhost.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
// Unless explicitly stated otherwise all files in this repository are licensed
// under the Apache License Version 2.0.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2016-2020 Datadog, Inc.
package externalhost
import "sync"
// hostname -> ExternalTags
type externalHost map[string]ExternalTags
var (
// externalHostCache maps source_type -> externalHost
externalHostCache = make(map[string]externalHost)
cacheMutex = &sync.Mutex{}
)
// SetExternalTags adds external tags for a specific host and source type
// to the cache.
func SetExternalTags(hostname, sourceType string, tags []string) {
cacheMutex.Lock()
defer cacheMutex.Unlock()
_, found := externalHostCache[sourceType]
if !found {
externalHostCache[sourceType] = make(externalHost)
}
externalHostCache[sourceType][hostname] = ExternalTags{sourceType: tags}
}
// GetPayload fills and return the external host tags metadata payload
func GetPayload() *Payload {
cacheMutex.Lock()
defer cacheMutex.Unlock()
payload := Payload{}
for _, extHost := range externalHostCache {
for hostname, tags := range extHost {
ht := hostTags{hostname, tags}
payload = append(payload, ht)
}
}
// clear the cache
externalHostCache = make(map[string]externalHost)
return &payload
}