feat: set entity timestamp (at source) if not provided (#20)

Signed-off-by: Michal Fiedorowicz <[email protected]>

Author: mfiedorowicz

diode/client.go

@@ -12,12 +12,14 @@ import (
 	"regexp"
 	"runtime"
 	"strings"
+	"time"
 
 	"github.com/google/uuid"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
 	"google.golang.org/grpc/credentials/insecure"
 	"google.golang.org/grpc/metadata"
+	"google.golang.org/protobuf/types/known/timestamppb"
 
 	"github.com/netboxlabs/diode-sdk-go/diode/v1/diodepb"
 )
@@ -230,10 +232,7 @@ func (g *GRPCClient) Close() error {
 func (g *GRPCClient) Ingest(ctx context.Context, entities []Entity) (*diodepb.IngestResponse, error) {
 	stream := defaultStreamName
 
-	protoEntities := make([]*diodepb.Entity, 0)
-	for _, entity := range entities {
-		protoEntities = append(protoEntities, entity.ConvertToProtoEntity())
-	}
+	protoEntities := convertEntitiesToProto(entities)
 
 	req := &diodepb.IngestRequest{
 		Id:                 uuid.NewString(),
@@ -250,6 +249,17 @@ func (g *GRPCClient) Ingest(ctx context.Context, entities []Entity) (*diodepb.In
 	return g.client.Ingest(ctx, req)
 }
 
+// convertEntitiesToProto converts entities to proto entities
+func convertEntitiesToProto(entities []Entity) []*diodepb.Entity {
+	protoEntities := make([]*diodepb.Entity, 0)
+	for _, entity := range entities {
+		entityPb := entity.ConvertToProtoEntity()
+		entityPb.Timestamp = timestamppb.New(time.Now().UTC())
+		protoEntities = append(protoEntities, entityPb)
+	}
+	return protoEntities
+}
+
 // methodUnaryInterceptor returns a gRPC dial option with a unary interceptor
 //
 // It's used to intercept the client calls and modify the method details.

diode/client_test.go

@@ -404,3 +404,35 @@ func TestNewLogger(t *testing.T) {
 		})
 	}
 }
+
+func TestConvertEntitiesToProto(t *testing.T) {
+	tests := []struct {
+		desc     string
+		entities []Entity
+	}{
+		{
+			desc:     "empty entities",
+			entities: nil,
+		},
+		{
+			desc: "non-empty entities",
+			entities: []Entity{
+				&Device{Name: String("device-1")},
+				&Site{Name: String("site-1")},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.desc, func(t *testing.T) {
+			protoEntities := convertEntitiesToProto(tt.entities)
+			require.NotNil(t, protoEntities)
+			assert.Equal(t, len(tt.entities), len(protoEntities))
+			for _, entityPb := range protoEntities {
+				assert.NotNil(t, entityPb.Timestamp)
+				assert.NotZero(t, entityPb.Timestamp.Seconds)
+				assert.NotZero(t, entityPb.Timestamp.Nanos)
+			}
+		})
+	}
+}