feat: add __rpc
This commit is contained in:
1316
__rpc/dubbo-rust/Cargo.lock
generated
Normal file
1316
__rpc/dubbo-rust/Cargo.lock
generated
Normal file
File diff suppressed because it is too large
Load Diff
30
__rpc/dubbo-rust/Cargo.toml
Normal file
30
__rpc/dubbo-rust/Cargo.toml
Normal file
@@ -0,0 +1,30 @@
|
||||
[package]
|
||||
name = "dubbo-rust"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
[[bin]]
|
||||
name = "greeter-server"
|
||||
path = "src/greeter/server.rs"
|
||||
|
||||
[[bin]]
|
||||
name = "greeter-client"
|
||||
path = "src/greeter/client.rs"
|
||||
|
||||
[dependencies]
|
||||
http = "0.2"
|
||||
http-body = "0.4.4"
|
||||
futures-util = {version = "0.3", default-features = false}
|
||||
tokio = { version = "1.0", features = [ "rt-multi-thread", "time", "fs", "macros", "net", "signal"] }
|
||||
prost-derive = {version = "0.10", optional = true}
|
||||
prost = "0.10.4"
|
||||
async-trait = "0.1.56"
|
||||
tokio-stream = "0.1"
|
||||
|
||||
dubbo = "0.2.0"
|
||||
dubbo-config = "0.2.0"
|
||||
|
||||
[build-dependencies]
|
||||
dubbo-build = "0.2.0"
|
||||
|
||||
|
||||
14
__rpc/dubbo-rust/README.md
Normal file
14
__rpc/dubbo-rust/README.md
Normal file
@@ -0,0 +1,14 @@
|
||||
|
||||
Install protoc on macOS:
|
||||
```
|
||||
PROTOC_ZIP=protoc-3.14.0-osx-x86_64.zip
|
||||
curl -OL https://github.com/protocolbuffers/protobuf/releases/download/v3.14.0/$PROTOC_ZIP
|
||||
sudo unzip -o $PROTOC_ZIP -d /usr/local bin/protoc
|
||||
sudo unzip -o $PROTOC_ZIP -d /usr/local 'include/*'
|
||||
rm -f $PROTOC_ZIP
|
||||
```
|
||||
|
||||
|
||||
from: http://google.github.io/proto-lens/installing-protoc.html
|
||||
|
||||
|
||||
9
__rpc/dubbo-rust/build.rs
Normal file
9
__rpc/dubbo-rust/build.rs
Normal file
@@ -0,0 +1,9 @@
|
||||
use std::path::PathBuf;
|
||||
|
||||
fn main() {
|
||||
let path = PathBuf::from("./src/greeter");
|
||||
dubbo_build::prost::configure()
|
||||
.output_dir(path)
|
||||
.compile(&["proto/greeter.proto"], &["proto/"])
|
||||
.unwrap();
|
||||
}
|
||||
19
__rpc/dubbo-rust/dubbo.yaml
Normal file
19
__rpc/dubbo-rust/dubbo.yaml
Normal file
@@ -0,0 +1,19 @@
|
||||
name: dubbo
|
||||
service:
|
||||
org.apache.dubbo.sample.tri.Greeter:
|
||||
version: 1.0.0
|
||||
group: test
|
||||
protocol: triple
|
||||
registry: ''
|
||||
serializer: json
|
||||
protocol_configs:
|
||||
triple:
|
||||
ip: 0.0.0.0
|
||||
port: '8888'
|
||||
name: triple
|
||||
protocols:
|
||||
triple:
|
||||
ip: 0.0.0.0
|
||||
port: '8888'
|
||||
name: triple
|
||||
|
||||
20
__rpc/dubbo-rust/proto/greeter.proto
Normal file
20
__rpc/dubbo-rust/proto/greeter.proto
Normal file
@@ -0,0 +1,20 @@
|
||||
syntax = "proto3";
|
||||
|
||||
option java_multiple_files = true;
|
||||
|
||||
package org.apache.dubbo.sample.tri;
|
||||
|
||||
// The request message containing the user's name.
|
||||
message GreeterRequest{
|
||||
string name = 1;
|
||||
}
|
||||
|
||||
// The response message containing the greetings
|
||||
message GreeterReply {
|
||||
string message = 1;
|
||||
}
|
||||
|
||||
service Greeter{
|
||||
// unary
|
||||
rpc greet(GreeterRequest) returns (GreeterReply);
|
||||
}
|
||||
23
__rpc/dubbo-rust/src/greeter/client.rs
Normal file
23
__rpc/dubbo-rust/src/greeter/client.rs
Normal file
@@ -0,0 +1,23 @@
|
||||
use dubbo::codegen::Request;
|
||||
use crate::sample::greeter_client::GreeterClient;
|
||||
use crate::sample::GreeterRequest;
|
||||
|
||||
mod sample;
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
let mut cli = GreeterClient::new().with_uri("http://127.0.0.1:8888".to_string());
|
||||
|
||||
println!("# unary call");
|
||||
let resp = cli
|
||||
.greet(Request::new(GreeterRequest {
|
||||
name: "message from client".to_string(),
|
||||
}))
|
||||
.await;
|
||||
let resp = match resp {
|
||||
Ok(resp) => resp,
|
||||
Err(err) => return println!("{:?}", err),
|
||||
};
|
||||
let (_parts, body) = resp.into_parts();
|
||||
println!("Response: {:?}", body);
|
||||
}
|
||||
169
__rpc/dubbo-rust/src/greeter/sample.rs
Normal file
169
__rpc/dubbo-rust/src/greeter/sample.rs
Normal file
@@ -0,0 +1,169 @@
|
||||
/// The request message containing the user's name.
|
||||
#[derive(Clone, PartialEq, ::prost::Message)]
|
||||
pub struct GreeterRequest {
|
||||
#[prost(string, tag="1")]
|
||||
pub name: ::prost::alloc::string::String,
|
||||
}
|
||||
/// The response message containing the greetings
|
||||
#[derive(Clone, PartialEq, ::prost::Message)]
|
||||
pub struct GreeterReply {
|
||||
#[prost(string, tag="1")]
|
||||
pub message: ::prost::alloc::string::String,
|
||||
}
|
||||
/// Generated client implementations.
|
||||
pub mod greeter_client {
|
||||
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
|
||||
use dubbo::codegen::*;
|
||||
#[derive(Debug, Clone, Default)]
|
||||
pub struct GreeterClient {
|
||||
inner: TripleClient,
|
||||
uri: String,
|
||||
}
|
||||
impl GreeterClient {
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
inner: TripleClient::new(),
|
||||
uri: "".to_string(),
|
||||
}
|
||||
}
|
||||
pub fn with_uri(mut self, uri: String) -> Self {
|
||||
self.uri = uri.clone();
|
||||
self.inner = self.inner.with_host(uri);
|
||||
self
|
||||
}
|
||||
/// unary
|
||||
pub async fn greet(
|
||||
&mut self,
|
||||
request: Request<super::GreeterRequest>,
|
||||
) -> Result<Response<super::GreeterReply>, dubbo::status::Status> {
|
||||
let codec = dubbo::codegen::ProstCodec::<
|
||||
super::GreeterRequest,
|
||||
super::GreeterReply,
|
||||
>::default();
|
||||
let path = http::uri::PathAndQuery::from_static(
|
||||
"/org.apache.dubbo.sample.tri.Greeter/greet",
|
||||
);
|
||||
self.inner.unary(request, codec, path).await
|
||||
}
|
||||
}
|
||||
}
|
||||
/// Generated server implementations.
|
||||
pub mod greeter_server {
|
||||
#![allow(unused_variables, dead_code, missing_docs, clippy::let_unit_value)]
|
||||
use dubbo::codegen::*;
|
||||
///Generated trait containing gRPC methods that should be implemented for use with GreeterServer.
|
||||
#[async_trait]
|
||||
pub trait Greeter: Send + Sync + 'static {
|
||||
/// unary
|
||||
async fn greet(
|
||||
&self,
|
||||
request: Request<super::GreeterRequest>,
|
||||
) -> Result<Response<super::GreeterReply>, dubbo::status::Status>;
|
||||
}
|
||||
#[derive(Debug)]
|
||||
pub struct GreeterServer<T: Greeter, I = TripleInvoker> {
|
||||
inner: _Inner<T>,
|
||||
invoker: Option<I>,
|
||||
}
|
||||
struct _Inner<T>(Arc<T>);
|
||||
impl<T: Greeter, I> GreeterServer<T, I> {
|
||||
pub fn new(inner: T) -> Self {
|
||||
Self {
|
||||
inner: _Inner(Arc::new(inner)),
|
||||
invoker: None,
|
||||
}
|
||||
}
|
||||
}
|
||||
impl<T, I, B> Service<http::Request<B>> for GreeterServer<T, I>
|
||||
where
|
||||
T: Greeter,
|
||||
B: Body + Send + 'static,
|
||||
B::Error: Into<StdError> + Send + 'static,
|
||||
I: Invoker + Send + 'static,
|
||||
{
|
||||
type Response = http::Response<BoxBody>;
|
||||
type Error = std::convert::Infallible;
|
||||
type Future = BoxFuture<Self::Response, Self::Error>;
|
||||
fn poll_ready(
|
||||
&mut self,
|
||||
_cx: &mut Context<'_>,
|
||||
) -> Poll<Result<(), Self::Error>> {
|
||||
Poll::Ready(Ok(()))
|
||||
}
|
||||
fn call(&mut self, req: http::Request<B>) -> Self::Future {
|
||||
let inner = self.inner.clone();
|
||||
match req.uri().path() {
|
||||
"/org.apache.dubbo.sample.tri.Greeter/greet" => {
|
||||
#[allow(non_camel_case_types)]
|
||||
struct greetServer<T: Greeter> {
|
||||
inner: _Inner<T>,
|
||||
}
|
||||
impl<T: Greeter> UnarySvc<super::GreeterRequest> for greetServer<T> {
|
||||
type Response = super::GreeterReply;
|
||||
type Future = BoxFuture<
|
||||
Response<Self::Response>,
|
||||
dubbo::status::Status,
|
||||
>;
|
||||
fn call(
|
||||
&mut self,
|
||||
request: Request<super::GreeterRequest>,
|
||||
) -> Self::Future {
|
||||
let inner = self.inner.0.clone();
|
||||
let fut = async move { inner.greet(request).await };
|
||||
Box::pin(fut)
|
||||
}
|
||||
}
|
||||
let fut = async move {
|
||||
let mut server = TripleServer::new(
|
||||
dubbo::codegen::ProstCodec::<
|
||||
super::GreeterReply,
|
||||
super::GreeterRequest,
|
||||
>::default(),
|
||||
);
|
||||
let res = server.unary(greetServer { inner }, req).await;
|
||||
Ok(res)
|
||||
};
|
||||
Box::pin(fut)
|
||||
}
|
||||
_ => {
|
||||
Box::pin(async move {
|
||||
Ok(
|
||||
http::Response::builder()
|
||||
.status(200)
|
||||
.header("grpc-status", "12")
|
||||
.header("content-type", "application/grpc")
|
||||
.body(empty_body())
|
||||
.unwrap(),
|
||||
)
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
impl<T: Greeter, I: Invoker + Send + 'static> Clone for GreeterServer<T, I> {
|
||||
fn clone(&self) -> Self {
|
||||
let inner = self.inner.clone();
|
||||
Self { inner, invoker: None }
|
||||
}
|
||||
}
|
||||
impl<T: Greeter> Clone for _Inner<T> {
|
||||
fn clone(&self) -> Self {
|
||||
Self(self.0.clone())
|
||||
}
|
||||
}
|
||||
impl<T: std::fmt::Debug> std::fmt::Debug for _Inner<T> {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
write!(f, "{:?}", self.0)
|
||||
}
|
||||
}
|
||||
pub fn register_server<T: Greeter>(server: T) {
|
||||
let s = GreeterServer::<_, TripleInvoker>::new(server);
|
||||
dubbo::protocol::triple::TRIPLE_SERVICES
|
||||
.write()
|
||||
.unwrap()
|
||||
.insert(
|
||||
"org.apache.dubbo.sample.tri.Greeter".to_string(),
|
||||
dubbo::utils::boxed_clone::BoxCloneService::new(s),
|
||||
);
|
||||
}
|
||||
}
|
||||
46
__rpc/dubbo-rust/src/greeter/server.rs
Normal file
46
__rpc/dubbo-rust/src/greeter/server.rs
Normal file
@@ -0,0 +1,46 @@
|
||||
use dubbo::codegen::{Request, Response};
|
||||
use dubbo::Dubbo;
|
||||
use dubbo_config::RootConfig;
|
||||
use crate::sample::greeter_server::{Greeter, register_server};
|
||||
use crate::sample::{GreeterReply, GreeterRequest};
|
||||
|
||||
mod sample;
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
register_server(GreeterServerImpl {
|
||||
name: "greeter".to_string(),
|
||||
});
|
||||
|
||||
Dubbo::new()
|
||||
.with_config({
|
||||
let r = RootConfig::new();
|
||||
match r.load() {
|
||||
Ok(config) => config,
|
||||
Err(_err) => panic!("err: {:?}", _err), // response was droped
|
||||
}
|
||||
})
|
||||
.start()
|
||||
.await;
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
#[derive(Default, Clone)]
|
||||
struct GreeterServerImpl {
|
||||
name: String,
|
||||
}
|
||||
|
||||
#[async_trait::async_trait]
|
||||
impl Greeter for GreeterServerImpl {
|
||||
async fn greet(
|
||||
&self,
|
||||
request: Request<GreeterRequest>,
|
||||
) -> Result<Response<GreeterReply>, dubbo::status::Status> {
|
||||
println!("GreeterServer::greet {:?}", request.metadata);
|
||||
|
||||
Ok(Response::new(GreeterReply {
|
||||
message: "hello, dubbo-rust".to_string(),
|
||||
}))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user