Extend DatabaseSessionStart posthog event (gravitational#28931)

shaban00 · Jul 14, 2023 · d4805cc · d4805cc
1 parent 6b26d57
commit d4805cc
Show file tree

Hide file tree

Showing 10 changed files with 2,529 additions and 1,993 deletions.
diff --git a/api/proto/teleport/legacy/types/events/events.proto b/api/proto/teleport/legacy/types/events/events.proto
@@ -2431,6 +2431,10 @@ message DatabaseMetadata {
   string DatabaseGCPInstanceID = 10 [(gogoproto.jsontag) = "db_gcp_instance_id,omitempty"];
   // DatabaseRoles is a list of database roles for auto-provisioned users.
   repeated string DatabaseRoles = 11 [(gogoproto.jsontag) = "db_roles,omitempty"];
+  // DatabaseType is the database type.
+  string DatabaseType = 12 [(gogoproto.jsontag) = "db_type,omitempty"];
+  // DatabaseOrigin is the database origin source.
+  string DatabaseOrigin = 13 [(gogoproto.jsontag) = "db_origin,omitempty"];
 }
 
 // DatabaseCreate is emitted when a new database resource is created.

diff --git a/api/types/events/events.pb.go b/api/types/events/events.pb.go
diff --git a/gen/proto/go/prehog/v1alpha/teleport.pb.go b/gen/proto/go/prehog/v1alpha/teleport.pb.go
diff --git a/gen/proto/js/prehog/v1alpha/teleport_pb.d.ts b/gen/proto/js/prehog/v1alpha/teleport_pb.d.ts
diff --git a/gen/proto/js/prehog/v1alpha/teleport_pb.js b/gen/proto/js/prehog/v1alpha/teleport_pb.js
diff --git a/lib/srv/db/common/audit.go b/lib/srv/db/common/audit.go
@@ -103,6 +103,7 @@ func (a *audit) OnSessionStart(ctx context.Context, session *Session, sessionErr
 			Success: true,
 		},
 	}
+
 	// If the database session wasn't started successfully, emit
 	// a failure event with error details.
 	if sessionErr != nil {
@@ -212,5 +213,7 @@ func MakeDatabaseMetadata(session *Session) events.DatabaseMetadata {
 		DatabaseName:     session.DatabaseName,
 		DatabaseUser:     session.DatabaseUser,
 		DatabaseRoles:    session.DatabaseRoles,
+		DatabaseType:     session.Database.GetType(),
+		DatabaseOrigin:   session.Database.Origin(),
 	}
 }