Merge pull request #8 from Cervest/log-principal

Log principal
Cervest · Sep 9, 2022 · 28ec75f · 28ec75f
2 parents 2826bf1 + bf18b0b
commit 28ec75f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/api.go b/src/api.go
@@ -10,6 +10,7 @@ import (
 
 type session struct {
 	Catalog   string
+	Principal string
 	User      string
 	UserAgent string
 }
@@ -32,6 +33,7 @@ type QueryInfo struct {
 	Catalog        string
 	CreateTime     time.Time
 	DurationMs     int64
+	Principal      string
 	QueryId        string
 	QuerySizeBytes int
 	QueryType      string
@@ -82,6 +84,7 @@ func FetchQueryInfo() ([]QueryInfo, error) {
 				Catalog:        q.Session.Catalog,
 				CreateTime:     q.QueryStats.CreateTime,
 				DurationMs:     q.QueryStats.EndTime.Sub(q.QueryStats.CreateTime).Milliseconds(),
+				Principal:      q.Session.Principal,
 				QueryId:        q.QueryId,
 				QuerySizeBytes: len(q.Query),
 				QueryType:      q.QueryType,

diff --git a/src/main.go b/src/main.go
@@ -28,6 +28,7 @@ func metricsHandler(w http.ResponseWriter, r *http.Request) {
 				Str("catalog", q.Catalog).
 				Int64("duration_ms", q.DurationMs).
 				Str("id", q.QueryId).
+				Str("principal", q.Principal).
 				Time("query_time", q.CreateTime).
 				Int("size_bytes", q.QuerySizeBytes).
 				Str("state", q.State).