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
71
72
73
|
package writer
import (
"fmt"
"net/netip"
"github.com/maxmind/mmdbwriter"
"github.com/maxmind/mmdbwriter/mmdbtype"
"go4.org/netipx"
)
// WriteDeeplyNestedStructuresTestDB writes an mmdb file with deeply nested record value types.
func (w *Writer) WriteDeeplyNestedStructuresTestDB() error {
dbWriter, err := mmdbwriter.New(
mmdbwriter.Options{
DatabaseType: "MaxMind DB Nested Data Structures",
Description: map[string]string{
"en": "MaxMind DB Nested Data Structures Test database - contains deeply nested map/array structures",
},
DisableIPv4Aliasing: false,
IncludeReservedNetworks: true,
IPVersion: 6,
Languages: []string{"en"},
RecordSize: 24,
},
)
if err != nil {
return fmt.Errorf("creating mmdbwriter: %w", err)
}
addrs, err := parseIPSlice(ipSample)
if err != nil {
return fmt.Errorf("parsing ip addresses: %w", err)
}
if err := insertNestedStructure(dbWriter, addrs); err != nil {
return fmt.Errorf("inserting all types records: %w", err)
}
if err := w.write(dbWriter, "MaxMind-DB-test-nested.mmdb"); err != nil {
return fmt.Errorf("writing database: %w", err)
}
return nil
}
// insertNestedStructure inserts records with deeply nested structures.
func insertNestedStructure(w *mmdbwriter.Tree, ipAddresses []netip.Prefix) error {
nestedStruct := mmdbtype.Map{
"map1": mmdbtype.Map{
"map2": mmdbtype.Map{
"array": mmdbtype.Slice{
mmdbtype.Map{
"map3": mmdbtype.Map{
"a": mmdbtype.Uint32(1),
"b": mmdbtype.Uint32(2),
"c": mmdbtype.Uint32(3),
},
},
},
},
},
}
for _, addr := range ipAddresses {
err := w.Insert(
netipx.PrefixIPNet(addr),
nestedStruct,
)
if err != nil {
return fmt.Errorf("inserting ip: %w", err)
}
}
return nil
}
|
