From c07727548d39e039ddee6aa4db3439b719ca44bc Mon Sep 17 00:00:00 2001
From: Garrison Davis <garrison.davis@featurebase.com>
Date: Mon, 23 Jan 2023 09:59:14 -0700
Subject: [PATCH] Stop using string keys in contexts

---
 server/grpc.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/server/grpc.go b/server/grpc.go
index 8090e60b3..50e50c570 100644
--- a/server/grpc.go
+++ b/server/grpc.go
@@ -1637,7 +1637,10 @@ func NewGRPCServer(opts ...grpcServerOption) (*grpcServer, error) {
 
 // LogQuery logs requests
 func LogQuery(ctx context.Context, method string, req interface{}, logger logger.Logger) {
-	uinfo, _ := ctx.Value("userinfo").(*authn.UserInfo)
+	uinfo, ok := authn.GetUserInfo(ctx)
+	if !ok {
+		uinfo = &authn.UserInfo{}
+	}
 	md, _ := metadata.FromIncomingContext(ctx)
 	p, ok := peer.FromContext(ctx)
 	ip := ""