帶入gRPC：對(duì) RPC 方法做自定義認(rèn)證

wh469012917 發(fā)布于2019-07-01 10:49 / 2164人閱讀

摘要：帶入對(duì)方法做自定義認(rèn)證原文地址帶入對(duì)方法做自定義認(rèn)證項(xiàng)目地址前言在前面的章節(jié)中，我們介紹了兩種證書(shū)算一種可全局認(rèn)證的方法證書(shū)認(rèn)證基于的證書(shū)認(rèn)證而在實(shí)際需求中，常常會(huì)對(duì)某些模塊的方法做特殊認(rèn)證或校驗(yàn)。

帶入gRPC：對(duì) RPC 方法做自定義認(rèn)證

原文地址：帶入gRPC：對(duì) RPC 方法做自定義認(rèn)證
項(xiàng)目地址：https://github.com/EDDYCJY/go...

前言

在前面的章節(jié)中，我們介紹了兩種（證書(shū)算一種）可全局認(rèn)證的方法：

TLS 證書(shū)認(rèn)證

基于 CA 的 TLS 證書(shū)認(rèn)證

Unary and Stream interceptor

而在實(shí)際需求中，常常會(huì)對(duì)某些模塊的 RPC 方法做特殊認(rèn)證或校驗(yàn)。今天將會(huì)講解、實(shí)現(xiàn)這塊的功能點(diǎn)

課前知識(shí)

type PerRPCCredentials interface {
    GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error)
    RequireTransportSecurity() bool
}

在 gRPC 中默認(rèn)定義了 PerRPCCredentials，它就是本章節(jié)的主角，是 gRPC 默認(rèn)提供用于自定義認(rèn)證的接口，它的作用是將所需的安全認(rèn)證信息添加到每個(gè) RPC 方法的上下文中。其包含 2 個(gè)方法：

GetRequestMetadata：獲取當(dāng)前請(qǐng)求認(rèn)證所需的元數(shù)據(jù)（metadata）

RequireTransportSecurity：是否需要基于 TLS 認(rèn)證進(jìn)行安全傳輸

目錄結(jié)構(gòu)

新建 simple_token_server/server.go 和 simple_token_client/client.go，目錄結(jié)構(gòu)如下：

go-grpc-example
├── client
│?? ├── simple_client
│?? ├── simple_http_client
│?? ├── simple_token_client
│?? └── stream_client
├── conf
├── pkg
├── proto
├── server
│?? ├── simple_http_server
│?? ├── simple_server
│?? ├── simple_token_server
│?? └── stream_server
└── vendor

gRPC Client

package main

import (
    "context"
    "log"

    "google.golang.org/grpc"

    "github.com/EDDYCJY/go-grpc-example/pkg/gtls"
    pb "github.com/EDDYCJY/go-grpc-example/proto"
)

const PORT = "9004"

type Auth struct {
    AppKey    string
    AppSecret string
}

func (a *Auth) GetRequestMetadata(ctx context.Context, uri ...string) (map[string]string, error) {
    return map[string]string{"app_key": a.AppKey, "app_secret": a.AppSecret}, nil
}

func (a *Auth) RequireTransportSecurity() bool {
    return true
}

func main() {
    tlsClient := gtls.Client{
        ServerName: "go-grpc-example",
        CertFile:   "../../conf/server/server.pem",
    }
    c, err := tlsClient.GetTLSCredentials()
    if err != nil {
        log.Fatalf("tlsClient.GetTLSCredentials err: %v", err)
    }

    auth := Auth{
        AppKey:    "eddycjy",
        AppSecret: "20181005",
    }
    conn, err := grpc.Dial(":"+PORT, grpc.WithTransportCredentials(c), grpc.WithPerRPCCredentials(&auth))
    ...
}

在 Client 端，重點(diǎn)實(shí)現(xiàn) type PerRPCCredentials interface 所需的方法，關(guān)注兩點(diǎn)即可：

struct Auth：GetRequestMetadata、RequireTransportSecurity

grpc.WithPerRPCCredentials

Server

package main

import (
    "context"
    "log"
    "net"

    "google.golang.org/grpc"
    "google.golang.org/grpc/codes"
    "google.golang.org/grpc/metadata"
    "google.golang.org/grpc/status"

    "github.com/EDDYCJY/go-grpc-example/pkg/gtls"
    pb "github.com/EDDYCJY/go-grpc-example/proto"
)

type SearchService struct {
    auth *Auth
}

func (s *SearchService) Search(ctx context.Context, r *pb.SearchRequest) (*pb.SearchResponse, error) {
    if err := s.auth.Check(ctx); err != nil {
        return nil, err
    }
    return &pb.SearchResponse{Response: r.GetRequest() + " Token Server"}, nil
}

const PORT = "9004"

func main() {
    ...
}

type Auth struct {
    appKey    string
    appSecret string
}

func (a *Auth) Check(ctx context.Context) error {
    md, ok := metadata.FromIncomingContext(ctx)
    if !ok {
        return status.Errorf(codes.Unauthenticated, "自定義認(rèn)證 Token 失敗")
    }

    var (
        appKey    string
        appSecret string
    )
    if value, ok := md["app_key"]; ok {
        appKey = value[0]
    }
    if value, ok := md["app_secret"]; ok {
        appSecret = value[0]
    }

    if appKey != a.GetAppKey() || appSecret != a.GetAppSecret() {
        return status.Errorf(codes.Unauthenticated, "自定義認(rèn)證 Token 無(wú)效")
    }

    return nil
}

func (a *Auth) GetAppKey() string {
    return "eddycjy"
}

func (a *Auth) GetAppSecret() string {
    return "20181005"
}

在 Server 端就更簡(jiǎn)單了，實(shí)際就是調(diào)用 metadata.FromIncomingContext 從上下文中獲取 metadata，再在不同的 RPC 方法中進(jìn)行認(rèn)證檢查

驗(yàn)證

重新啟動(dòng) server.go 和 client.go，得到以下結(jié)果：

$ go run client.go
2018/10/05 20:59:58 resp: gRPC Token Server

修改 client.go 的值，制造兩者不一致，得到無(wú)效結(jié)果：

$ go run client.go
2018/10/05 21:00:05 client.Search err: rpc error: code = Unauthenticated desc = invalid token
exit status 1

一個(gè)個(gè)加太麻煩

我相信你肯定會(huì)問(wèn)一個(gè)個(gè)加，也太麻煩了吧？有這個(gè)想法的你，應(yīng)當(dāng)把 type PerRPCCredentials interface 做成一個(gè)攔截器（interceptor）

總結(jié)

本章節(jié)比較簡(jiǎn)單，主要是針對(duì) RPC 方法的自定義認(rèn)證進(jìn)行了介紹，如果是想做全局的，建議是舉一反三從攔截器下手哦

GPU云服務(wù)器云服務(wù)器對(duì)人工智能定義 grpc 屬性定義方法自定義方法

文章版權(quán)歸作者所有，未經(jīng)允許請(qǐng)勿轉(zhuǎn)載,若此文章存在違規(guī)行為，您可以聯(lián)系管理員刪除。

轉(zhuǎn)載請(qǐng)注明本文地址：http://www.ezyhdfw.cn/yun/29522.html

發(fā)表評(píng)論

登陸后可評(píng)論

0條評(píng)論

wh469012917

男|高級(jí)講師

我要關(guān)注我要私信

TA的文章

tensorflow

閱讀 3094·2023-04-26 00:32
css清除浮動(dòng)clearfix：after

閱讀 570·2019-08-30 15:52
頁(yè)面設(shè)計(jì)筆記

閱讀 2172·2019-08-30 15:52
小程序輸入框文字上移問(wèn)題文字彈出輸入框

閱讀 3449·2019-08-30 15:44
使用GitHub（一）：添加SSHkey

閱讀 3340·2019-08-30 14:09
Fixed with absolute

閱讀 1477·2019-08-29 15:15
## CSS 的黑科技

閱讀 3454·2019-08-28 18:12
原生js實(shí)現(xiàn)each方法

閱讀 1165·2019-08-26 13:55

亚洲中字慕日产2020,大陆极品少妇内射AAAAAA,无码av大香线蕉伊人久久,久久精品国产亚洲av麻豆网站

資訊專(zhuān)欄INFORMATION COLUMN

上云采購(gòu)季！| 2核2G4M爆款云服務(wù)器低至59元/年，更有多臺(tái)、長(zhǎng)期優(yōu)惠，快來(lái)選購(gòu)！

帶入gRPC：對(duì) RPC 方法做自定義認(rèn)證

相關(guān)文章

**帶入gRPC：gRPC及相關(guān)介紹**

**帶入gRPC：gRPC及相關(guān)介紹**

**帶入gRPC：gRPC Streaming, Client and Server**

帶入gRPC：基于 CA 的 TLS 證書(shū)認(rèn)證

**帶入gRPC：分布式鏈路追蹤 gRPC-Opentracing-Zipkin**

發(fā)表評(píng)論

0條評(píng)論

wh469012917

男|高級(jí)講師

TA的文章

tensorflow

css清除浮動(dòng)clearfix：after

頁(yè)面設(shè)計(jì)筆記

小程序輸入框文字上移問(wèn)題文字彈出輸入框

使用GitHub（一）：添加SSHkey

Fixed with absolute

## CSS 的黑科技

原生js實(shí)現(xiàn)each方法

最新活動(dòng)

資訊專(zhuān)欄INFORMATION COLUMN

上云采購(gòu)季！| 2核2G4M爆款云服務(wù)器低至59元/年，更有多臺(tái)、長(zhǎng)期優(yōu)惠，快來(lái)選購(gòu)！

帶入gRPC：對(duì) RPC 方法做自定義認(rèn)證

相關(guān)文章

發(fā)表評(píng)論

0條評(píng)論

男|高級(jí)講師

TA的文章

最新活動(dòng)

上云采購(gòu)季！| 2核2G4M爆款云服務(wù)器低至59元/年，更有多臺(tái)、長(zhǎng)期優(yōu)惠，快來(lái)選購(gòu)！