anubis/internal/dns/cache.go at bea369644e1f7edcbcd287f951102e540452df8d · btomaev/anubis · GitHub

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package dns

import (
	"log/slog"
	"time"

	"github.com/TecharoHQ/anubis/lib/store"

	_ "github.com/TecharoHQ/anubis/lib/store/all"
)

type DnsCache struct {
	forward    store.JSON[[]string]
	reverse    store.JSON[[]string]
	forwardTTL time.Duration
	reverseTTL time.Duration
}

func NewDNSCache(forwardTTL int, reverseTTL int, backend store.Interface) *DnsCache {
	return &DnsCache{
		forward: store.JSON[[]string]{
			Underlying: backend,
			Prefix:     "forwardDNS",
		},
		reverse: store.JSON[[]string]{
			Underlying: backend,
			Prefix:     "reverseDNS",
		},
		forwardTTL: time.Duration(forwardTTL) * time.Second,
		reverseTTL: time.Duration(reverseTTL) * time.Second,
	}
}

func (d *Dns) getCachedForward(host string) ([]string, bool) {
	if d.cache == nil {
		return nil, false
	}
	if cached, err := d.cache.forward.Get(d.ctx, host); err == nil {
		slog.Debug("DNS: forward cache hit", "name", host, "ips", cached)
		return cached, true
	}
	slog.Debug("DNS: forward cache miss", "name", host)
	return nil, false
}

func (d *Dns) getCachedReverse(addr string) ([]string, bool) {
	if d.cache == nil {
		return nil, false
	}
	if cached, err := d.cache.reverse.Get(d.ctx, addr); err == nil {
		slog.Debug("DNS: reverse cache hit", "addr", addr, "names", cached)
		return cached, true
	}
	slog.Debug("DNS: reverse cache miss", "addr", addr)
	return nil, false
}

func (d *Dns) forwardCachePut(host string, entries []string) {
	if d.cache == nil {
		return
	}
	d.cache.forward.Set(d.ctx, host, entries, d.cache.forwardTTL)
}

func (d *Dns) reverseCachePut(addr string, entries []string) {
	if d.cache == nil {
		return
	}
	d.cache.reverse.Set(d.ctx, addr, entries, d.cache.reverseTTL)
}