Watch entities

1
package main
2
3
import (
4
	"context"
5
	"fmt"
6
	"net/http"
7
	"os"
8
	"time"
9
10
	"github.com/anduril/lattice-sdk-go/v2/client"
11
	"github.com/anduril/lattice-sdk-go/v2/option"
12
)
13
14
func main() {
15
	// Get environment variables
16
	latticeEndpoint := os.Getenv("LATTICE_ENDPOINT")
17
	environmentToken := os.Getenv("ENVIRONMENT_TOKEN")
18
19
	// Remove sandboxesToken from the following statements if you are not developing on Sandboxes.
20
	sandboxesToken := os.Getenv("SANDBOXES_TOKEN")
21
22
	// Check required environment variables
23
	if latticeEndpoint == "" || environmentToken == "" || sandboxesToken == "" {
24
		fmt.Println("Missing required environment variables")
25
		os.Exit(1)
26
	}
27
28
	// Initialize headers for sandbox authorization
29
	headers := http.Header{}
30
	headers.Add("Anduril-Sandbox-Authorization", fmt.Sprintf("Bearer %s", sandboxesToken))
31
32
	// Create the client
33
	LatticeClient := client.NewClient(
34
		option.WithToken(environmentToken),
35
		option.WithBaseURL(fmt.Sprintf("https://%s", latticeEndpoint)),
36
		option.WithHTTPHeader(headers),
37
	)
38
39
	// Continuously get the entity
40
	for {
41
		// Create context for the request
42
		ctx := context.Background()
43
44
		// Get the entity
45
		entity, err := LatticeClient.Entities.GetEntity(
46
			ctx,
47
			"<ENTITY_ID>",
48
		)
49
50
		// Handle errors
51
		if err != nil {
52
			fmt.Printf("Error: %v\n", err)
53
		} else {
54
			fmt.Printf("Asset name     | %s\n", *entity.Aliases.Name)
55
			fmt.Printf("Asset location | %f, %f\n", *entity.Location.Position.LatitudeDegrees, *entity.Location.Position.LongitudeDegrees)
56
		}
57
58
		// Wait before next request
59
		time.Sleep(5 * time.Second)
60
	}
61
}

1
package main
2
3
import (
4
	"context"
5
	"errors"
6
	"fmt"
7
	"io"
8
	"log"
9
	"net/http"
10
	"os"
11
	"time"
12
13
	Lattice "github.com/anduril/lattice-sdk-go/v2"
14
	"github.com/anduril/lattice-sdk-go/v2/client"
15
	"github.com/anduril/lattice-sdk-go/v2/option"
16
)
17
18
func main() {
19
	ctx, cancel := context.WithCancel(context.Background())
20
	defer cancel()
21
22
	latticeEndpoint := os.Getenv("LATTICE_ENDPOINT")
23
	environmentToken := os.Getenv("ENVIRONMENT_TOKEN")
24
	sandboxesToken := os.Getenv("SANDBOXES_TOKEN")
25
26
	if latticeEndpoint == "" || environmentToken == "" || sandboxesToken == "" {
27
		log.Fatal("Required environment variables not set.")
28
	}
29
30
	headers := http.Header{}
31
	headers.Add("Anduril-Sandbox-Authorization", fmt.Sprintf("Bearer %s", sandboxesToken))
32
33
	latticeClient := client.NewClient(
34
		option.WithToken(environmentToken),
35
		option.WithBaseURL(fmt.Sprintf("https://%s", latticeEndpoint)),
36
		option.WithHTTPHeader(headers),
37
	)
38
39
	// Create the entity stream.
40
	stream, err := latticeClient.Entities.StreamEntities(ctx, &Lattice.EntityStreamRequest{
41
		PreExistingOnly: Lattice.Bool(false),
42
	})
43
	if err != nil {
44
		log.Fatalf("Failed to create entity stream: %v", err)
45
	}
46
	defer stream.Close()
47
48
	for {
49
		select {
50
		case <-ctx.Done():
51
			log.Printf("Context canceled: %v", ctx.Err())
52
			return
53
		default:
54
			// Continue processing
55
		}
56
		message, err := stream.Recv()
57
58
		if errors.Is(err, io.EOF) {
59
			log.Println("Stream completed successfully.")
60
			return
61
		}
62
63
		if err != nil {
64
			log.Printf("Error receiving message: %v", err)
65
			continue
66
		}
67
68
		// Process the message based on its type
69
		switch message.Event {
70
		case "heartbeat":
71
			timestamp := *message.Heartbeat.Timestamp
72
			log.Printf("Heartbeat: %s", timestamp.Format(time.RFC3339))
73
		case "entity":
74
			log.Printf("Entity: %s", *message.Entity.Entity.EntityID)
75
		default:
76
			log.Printf("Unknown event type: %s", message.Event)
77
		}
78
	}
79
}

1
package main
2
3
import (
4
	"context"
5
	"errors"
6
	"fmt"
7
	"io"
8
	"log"
9
	"net/http"
10
	"os"
11
	"time"
12
13
	Lattice "github.com/anduril/lattice-sdk-go/v2"
14
	"github.com/anduril/lattice-sdk-go/v2/client"
15
	"github.com/anduril/lattice-sdk-go/v2/option"
16
)
17
18
func main() {
19
	ctx, cancel := context.WithCancel(context.Background())
20
	defer cancel()
21
22
	latticeEndpoint := os.Getenv("LATTICE_ENDPOINT")
23
	environmentToken := os.Getenv("ENVIRONMENT_TOKEN")
24
	sandboxesToken := os.Getenv("SANDBOXES_TOKEN")
25
26
	if latticeEndpoint == "" || environmentToken == "" || sandboxesToken == "" {
27
		log.Fatal("Required environment variables not set.")
28
	}
29
30
	headers := http.Header{}
31
	headers.Add("Anduril-Sandbox-Authorization", fmt.Sprintf("Bearer %s", sandboxesToken))
32
33
	latticeClient := client.NewClient(
34
		option.WithToken(environmentToken),
35
		option.WithBaseURL(fmt.Sprintf("https://%s", latticeEndpoint)),
36
		option.WithHTTPHeader(headers),
37
	)
38
39
	// Create the entity stream.
40
	stream, err := latticeClient.Entities.StreamEntities(ctx, &Lattice.EntityStreamRequest{
41
		PreExistingOnly: Lattice.Bool(false),
42
		// Define a list of components to control which entities are fetched.
43
		// If set, Lattice streams only entities with the components you provide.
44
		ComponentsToInclude: []string{"aliases", "locationUncertainty"},
45
	})
46
	if err != nil {
47
		log.Fatalf("Failed to create entity stream: %v", err)
48
	}
49
	defer stream.Close()
50
51
	for {
52
		select {
53
		case <-ctx.Done():
54
			log.Printf("Context canceled: %v", ctx.Err())
55
			return
56
		default:
57
			// Continue processing
58
		}
59
		message, err := stream.Recv()
60
61
		// Handle stream completion
62
		if errors.Is(err, io.EOF) {
63
			log.Println("Stream completed successfully.")
64
			return
65
		}
66
67
		if err != nil {
68
			log.Printf("Error receiving message: %v", err)
69
			continue
70
		}
71
72
		// Process the event based whether it is a heartbeat or entity event.
73
		switch message.Event {
74
		case "heartbeat":
75
			timestamp := *message.Heartbeat.Timestamp
76
			log.Printf("Heartbeat: %s", timestamp.Format(time.RFC3339))
77
		case "entity":
78
			log.Printf("Entity: %s", *message.Entity.Entity.Aliases.Name)
79
		default:
80
			log.Printf("Unknown event type: %s", message.Event)
81
		}
82
	}
83
}