grpc服務如何新增sentry監控(新增中介軟體)

雪山飛豬發表於2020-09-16

需求

sentry是一款非常好用的工具,可以方便追蹤線上的異常,在gin框架裡邊可以使用Use新增中件間,但是grpc服務在網上搜尋了一堆沒一個能用的,只能硬著頭皮看原始碼
終於搞定!

grpc服務有一個github.com/grpc-ecosystem/go-grpc-middleware的中介軟體外掛,可以新增多個外掛,使用如下,其實就是住裡邊追加中介軟體


分別是捕獲正常方法和流式方法的請求

我們看看中介軟體的定義

type UnaryServerInterceptor func(ctx context.Context, req interface{}, info *UnaryServerInfo, handler UnaryHandler) (resp interface{}, err error)

我們只需要寫一個類似的中介軟體就行了,分別是正常方法和流式方法,talk is cheap,上程式碼!

解決

新增中介軟體

package middleware

import (
	"errors"
	"fmt"
	"github.com/getsentry/sentry-go"
	"golang.org/x/net/context"
	"google.golang.org/grpc"
	"time"
)

func SentryUnaryServerInterceptor() grpc.UnaryServerInterceptor {
	return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (_ interface{}, err error) {
		defer func() {
			if r := recover(); r != nil {
				flushToSentry(r)
			}
		}()

		return handler(ctx, req)
	}
}

func SentryStreamServerInterceptor() grpc.StreamServerInterceptor {
	return func(srv interface{}, stream grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) (err error) {
		defer func() {
			if r := recover(); r != nil {
				flushToSentry(r)
			}
		}()
		return handler(srv, stream)
	}
}

func flushToSentry(v interface{}) {
	sentry.CaptureException(errors.New(fmt.Sprintf("%v", v)))
	sentry.Flush(time.Second * 5)
}

將上面的中介軟體註冊

再來找個程式碼試一試,這裡我顯式拋了一個異常chenqionghe test

呼叫後,發現sentry已經能正常收集到異常,如下

就是這麼簡單,喵~

相關文章