0
  • 聊天消息
  • 系統(tǒng)消息
  • 評(píng)論與回復(fù)
登錄后你可以
  • 下載海量資料
  • 學(xué)習(xí)在線課程
  • 觀看技術(shù)視頻
  • 寫文章/發(fā)帖/加入社區(qū)
會(huì)員中心
創(chuàng)作中心

完善資料讓更多小伙伴認(rèn)識(shí)你,還能領(lǐng)取20積分哦,立即完善>

3天內(nèi)不再提示

GRPC的基礎(chǔ)使用方法

科技綠洲 ? 來源:TinyZ ? 作者:TinyZ ? 2023-09-19 16:08 ? 次閱讀

gRPC 是 Google 開源的高性能、通用的 RPC 框架,它采用了基于 HTTP/2 協(xié)議的二進(jìn)制傳輸協(xié)議,支持多種語言,包括 Rust。Rust 語言 GRPC 模塊是一個(gè)用于 Rust 語言的 gRPC 客戶端和服務(wù)器實(shí)現(xiàn),它提供了一個(gè)簡單易用的 API,可以方便地創(chuàng)建和使用 gRPC 服務(wù)。

基礎(chǔ)用法

創(chuàng)建 gRPC 服務(wù)器

在 Rust 語言 GRPC 模塊中,可以使用ServerBuilder結(jié)構(gòu)體來創(chuàng)建 gRPC 服務(wù)器。下面是一個(gè)簡單的示例:

use grpc::{Server, ServerBuilder};

fn main() {
    let mut server = ServerBuilder::new_plain();
    server.http.set_port(50051);
    server.add_service(proto::greeter_server::GreeterServer::new_service_def(GreeterImpl {}));
    let server = server.build().unwrap();
    server.start();
    server.wait();
}

struct GreeterImpl {}

impl proto::greeter_server::Greeter for GreeterImpl {
    fn say_hello(&self, _m: grpc::RequestOptions, req: proto::HelloRequest) - > grpc::SingleResponse< proto::HelloReply > {
        let mut r = proto::HelloReply::new();
        r.set_message(format!("Hello, {}!", req.get_name()));
        grpc::SingleResponse::completed(r)
    }
}

這個(gè)示例中,我們創(chuàng)建了一個(gè)ServerBuilder對(duì)象,并通過http字段設(shè)置了服務(wù)器的端口號(hào)。然后我們使用add_service方法將我們實(shí)現(xiàn)的Greeter服務(wù)添加到服務(wù)器中。最后,我們通過build方法構(gòu)建了服務(wù)器,并通過start方法啟動(dòng)了服務(wù)器。服務(wù)器啟動(dòng)后,我們通過wait方法等待客戶端連接。

創(chuàng)建 gRPC 客戶端

在 Rust 語言 GRPC 模塊中,可以使用Client結(jié)構(gòu)體來創(chuàng)建 gRPC 客戶端。下面是一個(gè)簡單的示例:

use grpc::{ChannelBuilder, Client};

fn main() {
    let ch = ChannelBuilder::new_plain();
    let client = Client::new(ch);
    let mut req = proto::HelloRequest::new();
    req.set_name("world".to_string());
    let resp = client.say_hello(grpc::RequestOptions::new(), req);
    println!("{}", resp.wait().unwrap().get_message());
}

這個(gè)示例中,我們創(chuàng)建了一個(gè)ChannelBuilder對(duì)象,并使用Client結(jié)構(gòu)體創(chuàng)建了一個(gè) gRPC 客戶端。然后我們創(chuàng)建了一個(gè)HelloRequest對(duì)象,并設(shè)置了它的name字段。最后,我們使用say_hello方法向服務(wù)器發(fā)送請(qǐng)求,并通過wait方法等待響應(yīng)。響應(yīng)對(duì)象是一個(gè)SingleResponse對(duì)象,我們通過unwrap方法獲取了它的值,并打印了它的message字段。

使用流式 RPC

在 Rust 語言 GRPC 模塊中,可以使用流式 RPC 來傳輸流數(shù)據(jù)。下面是一個(gè)簡單的示例:

use grpc::{Client, ClientStreamingSink, Server, ServerBuilder, ServerStreamingSink, WriteFlags};

fn main() {
    let mut server = ServerBuilder::new_plain();
    server.http.set_port(50051);
    server.add_service(proto::streaming::create_greeter_server(GreeterImpl {}));
    let server = server.build().unwrap();
    server.start();

    let ch = ChannelBuilder::new_plain();
    let client = Client::new(ch);

    let reqs = vec![
        proto::HelloRequest::new(),
        proto::HelloRequest::new(),
        proto::HelloRequest::new(),
    ];

    let (mut tx, rx) = client.say_hello_stream(grpc::RequestOptions::new()).unwrap();
    for req in reqs {
        tx = tx.send((req, WriteFlags::default())).unwrap();
    }
    tx.close().unwrap();

    for resp in rx.wait() {
        println!("{}", resp.unwrap().get_message());
    }
}

struct GreeterImpl {}

impl proto::streaming::Greeter for GreeterImpl {
    fn say_hello_stream(&self, _m: grpc::RequestOptions, _stream: grpc::StreamingRequest< proto::HelloRequest >) - > grpc::StreamingResponse< proto::HelloReply > {
        let (tx, rx) = grpc::channel::mpsc::channel(0);
        std::thread::spawn(move || {
            for req in _stream.into_iter() {
                let mut r = proto::HelloReply::new();
                r.set_message(format!("Hello, {}!", req.get_name()));
                tx.send((r, WriteFlags::default())).unwrap();
            }
            tx.close().unwrap();
        });
        grpc::StreamingResponse::new(rx)
    }
}

這個(gè)示例中,我們創(chuàng)建了一個(gè)Greeter服務(wù),并實(shí)現(xiàn)了一個(gè)say_hello_stream方法,該方法接收一個(gè)StreamingRequest對(duì)象,并返回一個(gè)StreamingResponse對(duì)象。在該方法中,我們使用mpsc::channel方法創(chuàng)建了一個(gè)通道,用于傳輸流數(shù)據(jù)。然后我們使用std::thread::spawn方法創(chuàng)建了一個(gè)線程,該線程會(huì)將接收到的請(qǐng)求轉(zhuǎn)換成響應(yīng),并通過通道發(fā)送給客戶端。最后,我們使用StreamingResponse::new方法將通道包裝成一個(gè)StreamingResponse對(duì)象,并將其返回給客戶端。

在客戶端中,我們創(chuàng)建了一個(gè)say_hello_stream方法,并使用send方法向服務(wù)器發(fā)送請(qǐng)求。然后我們通過wait方法等待響應(yīng),并打印了響應(yīng)的message字段。

使用雙向流式 RPC

在 Rust 語言 GRPC 模塊中,可以使用雙向流式 RPC 來傳輸雙向流數(shù)據(jù)。下面是一個(gè)簡單的示例:

use grpc::{Client, ClientStreamingSink, Server, ServerBuilder, ServerStreamingSink, StreamingSink, WriteFlags};

fn main() {
    let mut server = ServerBuilder::new_plain();
    server.http.set_port(50051);
    server.add_service(proto::streaming::create_greeter_server(GreeterImpl {}));
    let server = server.build().unwrap();
    server.start();

    let ch = ChannelBuilder::new_plain();
    let client = Client::new(ch);

    let (mut tx, rx) = client.say_hello_bidi(grpc::RequestOptions::new()).unwrap();
    let reqs = vec![
        proto::HelloRequest::new(),
        proto::HelloRequest::new(),
        proto::HelloRequest::new(),
    ];
    std::thread::spawn(move || {
        for req in reqs {
            tx = tx.send((req, WriteFlags::default())).unwrap();
            let resp = rx.into_future().wait().unwrap().0;
            println!("{}", resp.unwrap().get_message());
        }
        tx.close().unwrap();
    });
}

struct GreeterImpl {}

impl proto::streaming::Greeter for GreeterImpl {
    fn say_hello_bidi(&self, _m: grpc::RequestOptions, stream: grpc::StreamingRequest< proto::HelloRequest >) - > grpc::StreamingResponse< proto::HelloReply > {
        let (tx, rx) = grpc::channel::mpsc::channel(0);
        std::thread::spawn(move || {
            for req in stream.into_iter() {
                let mut r = proto::HelloReply::new();
                r.set_message(format!("Hello, {}!", req.get_name()));
                tx.send((r, WriteFlags::default())).unwrap();
            }
            tx.close().unwrap();
        });
        grpc::StreamingResponse::new(rx)
    }
}

這個(gè)示例中,我們創(chuàng)建了一個(gè)Greeter服務(wù),并實(shí)現(xiàn)了一個(gè)say_hello_bidi方法,該方法接收一個(gè)StreamingRequest對(duì)象,并返回一個(gè)StreamingResponse對(duì)象。在該方法中,我們使用mpsc::channel方法創(chuàng)建了一個(gè)通道,用于傳輸流數(shù)據(jù)。然后我們使用std::thread::spawn方法創(chuàng)建了一個(gè)線程,該線程會(huì)將接收到的請(qǐng)求轉(zhuǎn)換成響應(yīng),并通過通道發(fā)送給客戶端。最后,我們使用StreamingResponse::new方法將通道包裝成一個(gè)StreamingResponse對(duì)象,并將其返回給客戶端。

在客戶端中,我們使用say_hello_bidi方法向服務(wù)器發(fā)送請(qǐng)求,并通過into_future方法獲取響應(yīng)。然后我們通過println方法打印了響應(yīng)的message字段。

進(jìn)階用法

使用 tokio

在 Rust 語言 GRPC 模塊中,可以使用 tokio 來實(shí)現(xiàn)異步 RPC。下面是一個(gè)簡單的示例:

use grpc::{Client, ClientStreamingSink, Server, ServerBuilder, ServerStreamingSink, StreamingSink, WriteFlags};

#[tokio::main]
async fn main() {
    let mut server = ServerBuilder::new_plain();
    server.http.set_port(50051);
    server.add_service(proto::greeter_server::GreeterServer::new_service_def(GreeterImpl {}));
    let server = server.build().unwrap();
    server.start();

    let ch = ChannelBuilder::new_plain();
    let client = Client::new(ch);

    let mut req = proto::HelloRequest::new();
    req.set_name("world".to_string());
    let resp = client.say_hello_async(grpc::RequestOptions::new(), req).await.unwrap();
    println!("{}", resp.get_message());
}

struct GreeterImpl {}

impl proto::greeter_server::Greeter for GreeterImpl {
    fn say_hello(&self, _m: grpc::RequestOptions, req: proto::HelloRequest) - > grpc::SingleResponse< proto::HelloReply > {
        let mut r = proto::HelloReply::new();
        r.set_message(format!("Hello, {}!", req.get_name()));
        grpc::SingleResponse::completed(r)
    }
}

這個(gè)示例中,我們使用tokio::main宏來創(chuàng)建異步運(yùn)行時(shí)。在服務(wù)器和客戶端中,我們使用async關(guān)鍵字來定義異步函數(shù)。在客戶端中,我們使用await關(guān)鍵字來等待異步響應(yīng)。

tokio 使用流式 RPC

下面是一個(gè)使用 tokio 和流式 RPC 的示例:

use grpc::{Client, ClientStreamingSink, Server, ServerBuilder, ServerStreamingSink, StreamingSink, WriteFlags};
use tokio::sync::mpsc;

#[tokio::main]
async fn main() {
    let mut server = ServerBuilder::new_plain();
    server.http.set_port(50051);
    server.add_service(proto::streaming::create_greeter_server(GreeterImpl {}));
    let server = server.build().unwrap();
    server.start();

    let ch = ChannelBuilder::new_plain();
    let client = Client::new(ch);

    let (mut tx, rx) = mpsc::channel(10);
    let mut stream = client.say_hello_streaming(grpc::RequestOptions::new()).unwrap();
    tokio::spawn(async move {
        while let Some(req) = rx.recv().await {
            stream.send((req, WriteFlags::default())).unwrap();
        }
        stream.close().unwrap();
    });

    let reqs = vec![
        proto::HelloRequest::new(),
        proto::HelloRequest::new(),
        proto::HelloRequest::new(),
    ];
    for req in reqs {
        tx.send(req).await.unwrap();
    }

    for resp in stream.into_stream().await {
        println!("{}", resp.unwrap().get_message());
    }
}

struct GreeterImpl {}

impl proto::streaming::Greeter for GreeterImpl {
    fn say_hello_streaming(&self, _m: grpc::RequestOptions, _stream: grpc::StreamingRequest< proto::HelloRequest >) - > grpc::StreamingResponse< proto::HelloReply > {
        let (tx, rx) = grpc::channel::mpsc::channel(0);
        tokio::spawn(async move {
            for req in _stream.into_async_iter().await {
                let mut r = proto::HelloReply::new();
                r.set_message(format!("Hello, {}!", req.get_name()));
                tx.send((r, WriteFlags::default())).unwrap();
            }
            tx.close().unwrap();
        });
        grpc::StreamingResponse::new(rx)
    }
}

這個(gè)示例中,我們使用tokio::sync::mpsc庫來創(chuàng)建一個(gè)通道,用于傳輸流數(shù)據(jù)。在客戶端中,我們使用say_hello_streaming方法向服務(wù)器發(fā)送請(qǐng)求,并將請(qǐng)求通過通道發(fā)送給異步任務(wù)。在異步任務(wù)中,我們使用into_async_iter方法將請(qǐng)求流轉(zhuǎn)換成異步迭代器,并將響應(yīng)通過通道發(fā)送給客戶端。在客戶端中,我們使用into_stream方法將響應(yīng)流轉(zhuǎn)換成異步流,并等待響應(yīng)。

使用 TLS 加密

在 Rust 語言 GRPC 模塊中,可以使用 TLS 加密來保護(hù)通信安全。下面是一個(gè)簡單的示例:

use grpc::{ChannelBuilder, Client};
use rustls::{Certificate, PrivateKey, ServerConfig};
use std::fs::File;
use std::io::BufReader;

fn main() {
    let mut config = ServerConfig::new(rustls::NoClientAuth::new());
    let cert_file = &mut BufReader::new(File::open("server.crt").unwrap());
    let key_file = &mut BufReader::new(File::open("server.key").unwrap());
    let cert_chain = rustls::internal::pemfile::certs(cert_file).unwrap();
    let mut keys = rustls::internal::pemfile::rsa_private_keys(key_file).unwrap();
    config.set_single_cert(cert_chain, keys.remove(0)).unwrap();
    let mut server = grpc_tls::ServerBuilder::new_plain();
    server.http.set_port(50051);
    server.http.set_tls(config);
    server.add_service(proto::greeter_server::GreeterServer::new_service_def(GreeterImpl {}));
    let server = server.build().unwrap();
    server.start();

    let mut config = rustls::ClientConfig::new();
    let cert_file = &mut BufReader::new(File::open("client.crt").unwrap());
    let key_file = &mut BufReader::new(File::open("client.key").unwrap());
    let cert_chain = rustls::internal::pemfile::certs(cert_file).unwrap();
    let mut keys = rustls::internal::pemfile::rsa_private_keys(key_file).unwrap();
    config.set_single_client_cert(cert_chain, keys.remove(0));
    let ch = ChannelBuilder::new_tls().rustls_config(config);
    let client = Client::new(ch);
    let mut req = proto::HelloRequest::new();
    req.set_name("world".to_string());
    let resp = client.say_hello(grpc::RequestOptions::new(), req);
    println!("{}", resp.wait().unwrap().get_message());
}

struct GreeterImpl {}

impl proto::greeter_server::Greeter for GreeterImpl {
    fn say_hello(&self, _m: grpc::RequestOptions, req: proto::HelloRequest) - > grpc::SingleResponse< proto::HelloReply > {
        let mut r = proto::HelloReply::new();
        r.set_message(format!("Hello, {}!", req.get_name()));
        grpc::SingleResponse::completed(r)
    }
}

這個(gè)示例中,我們使用rustls庫來創(chuàng)建 TLS 配置,并使用grpc_tls::ServerBuilderChannelBuilder::new_tls方法來創(chuàng)建帶有 TLS 加密的服務(wù)器和客戶端。在服務(wù)器中,我們使用set_single_cert方法來設(shè)置服務(wù)器證書和私鑰。在客戶端中,我們使用set_single_client_cert方法來設(shè)置客戶端證書和私鑰。

總結(jié)

教程介紹了 GRPC 的基礎(chǔ)使用方法,并針對(duì) tokio 結(jié)合 GRPC 的進(jìn)階使用進(jìn)入入門級(jí)的探討。希望能幫助同學(xué)們掌握 Rust 語言 GRPC 的應(yīng)用。

聲明:本文內(nèi)容及配圖由入駐作者撰寫或者入駐合作網(wǎng)站授權(quán)轉(zhuǎn)載。文章觀點(diǎn)僅代表作者本人,不代表電子發(fā)燒友網(wǎng)立場(chǎng)。文章及其配圖僅供工程師學(xué)習(xí)之用,如有內(nèi)容侵權(quán)或者其他違規(guī)問題,請(qǐng)聯(lián)系本站處理。 舉報(bào)投訴
  • 服務(wù)器
    +關(guān)注

    關(guān)注

    12

    文章

    8700

    瀏覽量

    84528
  • RPC
    RPC
    +關(guān)注

    關(guān)注

    0

    文章

    110

    瀏覽量

    11475
  • 傳輸協(xié)議
    +關(guān)注

    關(guān)注

    0

    文章

    70

    瀏覽量

    11387
  • 開源
    +關(guān)注

    關(guān)注

    3

    文章

    3125

    瀏覽量

    42067
收藏 人收藏

    評(píng)論

    相關(guān)推薦

    示波器的使用方法

    示波器的使用方法 1 示波器使用        本節(jié)介紹示波器的使用方法。示波器種類、型號(hào)很多,功能也不同。
    發(fā)表于 01-14 13:06 ?1.8w次閱讀

    Matlab使用方法和程序設(shè)計(jì)

    Matlab使用方法和程序設(shè)計(jì) 實(shí)驗(yàn)一 Matlab使用方法和程序設(shè)計(jì)一、
    發(fā)表于 10-17 00:18 ?5294次閱讀
    Matlab<b class='flag-5'>使用方法</b>和程序設(shè)計(jì)

    多路輸出的使用方法

    多路輸出的使用方法  
    發(fā)表于 10-27 14:20 ?1598次閱讀

    示波器的使用方法

    數(shù)字萬用表使用方法和示波器的使用方法詳解。
    發(fā)表于 03-14 10:38 ?32次下載

    AT指令使用方法

    AT指令使用方法
    發(fā)表于 07-21 14:18 ?23次下載

    xilinx 原語使用方法

    xilinx 原語使用方法
    發(fā)表于 10-17 08:57 ?11次下載
    xilinx 原語<b class='flag-5'>使用方法</b>

    CC debuger的使用方法

    CC debuger的使用方法
    發(fā)表于 10-18 10:07 ?5次下載

    ORCAD PSPICE 使用方法

    ORCAD PSPICE 使用方法
    發(fā)表于 10-18 14:52 ?39次下載
    ORCAD PSPICE <b class='flag-5'>使用方法</b>

    xilinx原語使用方法

    xilinx原語使用方法
    發(fā)表于 10-19 08:50 ?15次下載
    xilinx原語<b class='flag-5'>使用方法</b>

    git使用方法

    git使用方法
    發(fā)表于 10-24 09:45 ?19次下載
    git<b class='flag-5'>使用方法</b>

    示波器的使用方法(三):示波器的使用方法詳解

    示波器的使用方法并非很難,重點(diǎn)在于正確使用示波器的使用方法。往期文章中,小編對(duì)模擬示波器的使用方法和數(shù)字示波器的使用方法均有所介紹。為增進(jìn)大家對(duì)示波器的
    的頭像 發(fā)表于 12-24 20:37 ?2981次閱讀

    IP知識(shí)百科之什么是gRPC

    對(duì)網(wǎng)絡(luò)設(shè)備進(jìn)行配置和管理的一種方法gRPC vs REST REST(Representational State Transfer)表征狀態(tài)轉(zhuǎn)移,是一種軟件架構(gòu)風(fēng)格,同樣為管理和配置網(wǎng)絡(luò)設(shè)備提供
    的頭像 發(fā)表于 11-16 15:13 ?3084次閱讀

    gRPC-Nebula微服務(wù)框架

    ./oschina_soft/grpc-nebula.zip
    發(fā)表于 06-22 14:59 ?0次下載
    <b class='flag-5'>gRPC</b>-Nebula微服務(wù)框架

    gRPC-Web訪問gRPC服務(wù)的Web客戶端

    ./oschina_soft/grpc-web.zip
    發(fā)表于 06-22 09:25 ?0次下載
    <b class='flag-5'>gRPC</b>-Web訪問<b class='flag-5'>gRPC</b>服務(wù)的Web客戶端

    使用go語言實(shí)現(xiàn)一個(gè)grpc攔截器

    在開發(fā)grpc服務(wù)時(shí),我們經(jīng)常會(huì)遇到一些通用的需求,比如:日志、鏈路追蹤、鑒權(quán)等。這些需求可以通過grpc攔截器來實(shí)現(xiàn)。本文使用go語言來實(shí)現(xiàn)一個(gè) grpc一元模式(Unary)攔截器,上報(bào)鏈路追蹤信息。
    的頭像 發(fā)表于 12-18 10:13 ?528次閱讀
    使用go語言實(shí)現(xiàn)一個(gè)<b class='flag-5'>grpc</b>攔截器