Gunakan semula klien pengelogan dalam pemintas kaedah pelayan grpc Golang

Kandungan soalan

Saya sedang membina pelayan grpc menggunakan go. Pada masa ini pelayan menyediakan tiga kaedah:

• Serahkan tugasan anda
• Batalkan kerja
• Dapatkan status pekerjaan

Saya menggunakan datadog untuk log beberapa metrik seperti kiraan permintaan, tempoh, dsb. Kaedah submitjob kelihatan seperti ini:

func (s *myserver) submitjob(ctx context.context, request *submitjobrequest) (*submitjobresponse, error) {
        s.dd_client.logrequestcount("submitjob")
        start_time := time.now()
        defer s.dd_client.logrequestduration("submitjob", time.since(start_time))

        resp := somefunc()

    return resp, nil
}

Kod pengelogan dalam tiga kaedah pelayan berbeza ini hampir sama. Jadi saya ingin tahu bagaimana untuk mengelakkan pertindihan ini.

Saya perasan golang grpc berkonsepkan pemintas. Pada asasnya saya boleh menentukan fungsi pemintas dan menggunakannya untuk melakukan pemprosesan pra/pasca panggilan kaedah pelayan.

Berikutan dokumentasi, saya menulis pemintas seperti berikut:

func unaryInterceptor(ctx context.Context,
    req interface{},
    info *grpc.UnaryServerInfo,
    handler grpc.UnaryHandler,
) (interface{}, error) {
    dd_client := NewDatadogClient()
    defer dd_client.Close()

        dd_client.LogRequestCount(info.FullMethod)

        start_time := time.Now()

    resp, err := handler(ctx, req)

        dd_client.LogRequestDuration(info.FullMethod, time.Since(start_time))

    return resp, err
}

Masalahnya ialah setiap kali pemintas dipanggil, ia mencipta dan memusnahkan pelanggan datadog baharu. Saya rasa ini tidak perlu. Tetapi oleh kerana unaryinterceptor hanyalah kaedah dan bukan kelas, saya tidak nampak cara untuk mencipta klien datadog sekali dan menggunakannya semula kemudian? Adakah terdapat cara untuk memenuhi keperluan saya?

Jawapan betul

Ya, jangan buat di tempat, tetapi alihkannya ke fail/pakej lain, mari panggil datadog.go

Anda perlu mengisytiharkannya sebagai singleton supaya ia hanya mempunyai 1 penunjuk dalam semua penggunaan seperti...

(Saya menggunakan jenis "datadogtype" untuk merujuk kepada jenis data yang dikembalikan oleh fungsi "newdatadogclient", sila nyatakan pakej dan versi untuk memberikan contoh kod yang lebih tepat)

//your package code....


var dataDog DataDogType

//access it through a wrapper, the wrapper will init the var the first time, 
//the remaining times it will be reused

func GetDataDog()(*DataDogType) {

if dataDog == nil {
    dataDog = NewDatadogClient()
}


return dataDog

}

Sekarang, bergantung pada kes penggunaan anda, jika anda mengembalikannya sebagai penunjuk dan menutupnya, anda akan menutup semua kegunaannya, jadi kembalikan penunjuk itu sendiri dan tinggalkan bahagian yang menutupnya, atau kembalikan nilainya, jadi ia adalah salinan daripada yang asli

BTW, bolehkah anda berkongsi versi datadog dan pakej khusus yang anda gunakan?

Atas ialah kandungan terperinci Gunakan semula klien pengelogan dalam pemintas kaedah pelayan grpc Golang. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!