diode-sdk-go/docs/examples/aggregate/main.go at dd23c63a2398b0c0ecc9cf608ef8201506f620f0 · netboxlabs/diode-sdk-go · 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
71
72
73
74
75
76
77
78
79
80
81
82
83
// Package main demonstrates ingesting Aggregate entities using the Diode SDK.
// This example includes three patterns: Minimal, Extended, and Explicit.
package main

import (
	"context"
	"log"

	"github.com/netboxlabs/diode-sdk-go/diode"
)

const (
	target     = "grpc://localhost:8080/diode"
	appName    = "aggregate-example"
	appVersion = "1.0.0"
)

func main() {
	client, err := diode.NewClient(
		target,
		appName,
		appVersion,
	)
	if err != nil {
		log.Fatalf("Failed to create client: %v", err)
	}

	// Choose one of the three patterns by uncommenting:
	aggregate := AggregateMinimal()
	// aggregate := AggregateExtended()
	// aggregate := AggregateExplicit()

	resp, err := client.Ingest(context.Background(), []diode.Entity{aggregate})
	if err != nil {
		log.Fatalf("Ingestion failed: %v", err)
	}
	if resp.Errors != nil {
		log.Printf("Errors: %v", resp.Errors)
	} else {
		log.Println("Aggregate ingested successfully")
	}
}

// AggregateMinimal Creates a Aggregate with only required fields.
func AggregateMinimal() *diode.Aggregate {
	return &diode.Aggregate{
		Prefix: diode.String("192.0.2.0/24"),
		Rir: &diode.RIR{
			Name: diode.String("Example Name"),
			Slug: diode.String("example-slug"),
		},
	}
}

// AggregateExtended Creates a Aggregate with common optional fields.
func AggregateExtended() *diode.Aggregate {
	return &diode.Aggregate{
		Prefix: diode.String("192.0.2.0/24"),
		Rir: &diode.RIR{
			Name: diode.String("Example Name"),
			Slug: diode.String("example-slug"),
		},
		Description: diode.String("Example description"),
	}
}

// AggregateExplicit Creates a Aggregate with fully nested objects and all common fields.
func AggregateExplicit() *diode.Aggregate {
	return &diode.Aggregate{
		Prefix: diode.String("192.0.2.0/24"),
		Rir: &diode.RIR{
			Name: diode.String("Example Name"),
			Slug: diode.String("example-slug"),
		},
		Description: diode.String("Example description"),
		Comments:    diode.String("Example comments"),
		Tenant: &diode.Tenant{
			Name: diode.String("Example Name"),
			Slug: diode.String("example-slug"),
		},
		Tags: []*diode.Tag{{Name: diode.String("production")}},
	}
}