feat: slow threshold customizable in zrpc (#1191)

* feat: slow threshold customizable in rest * feat: slow threshold customizable in rest * feat: slow threshold customizable in rest * feat: slow threshold customizable in zrpc
2021-11-01 15:04:38 +08:00
parent ebc90720ea
commit ba43214dae
9 changed files with 50 additions and 4 deletions
--- a/zrpc/internal/serverinterceptors/statinterceptor_test.go
+++ b/zrpc/internal/serverinterceptors/statinterceptor_test.go
@@ -13,6 +13,12 @@ import (
 	"google.golang.org/grpc/peer"
 )

+func TestSetSlowThreshold(t *testing.T) {
+	assert.Equal(t, defaultSlowThreshold, slowThreshold.Load())
+	SetSlowThreshold(time.Second)
+	assert.Equal(t, time.Second, slowThreshold.Load())
+}
+
 func TestUnaryStatInterceptor(t *testing.T) {
 	metrics := stat.NewMetrics("mock")
 	interceptor := UnaryStatInterceptor(metrics)