Use map to lookup status code

Co-authored-by: Aubrey Dean <[email protected]>
getsentry · Jan 21, 2021 · 93183e7 · 93183e7
1 parent 456ebca
commit 93183e7
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/grpc/options.go b/grpc/options.go
@@ -50,14 +50,11 @@ func ReportAlways(err error) bool {
 
 // ReportOnCodes returns true if error code matches on of the given codes.
 func ReportOnCodes(cc ...codes.Code) ReportOn {
+	cm := make(map[codes.Code]bool)
+	for _, c := range cc {
+		cm[c] = true
+	}
 	return func(err error) bool {
-		c := status.Code(err)
-		for i := range cc {
-			if c == cc[i] {
-				return true
-			}
-		}
-
-		return false
+		return cm[status.Code(err)]
 	}
 }