You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1009 B
49 lines
1009 B
package grpc
|
|
|
|
import (
|
|
"github.com/go-kratos/kratos/v2/metadata"
|
|
"github.com/go-kratos/kratos/v2/transport"
|
|
)
|
|
|
|
var (
|
|
_ transport.Transporter = &Transport{}
|
|
)
|
|
|
|
// Transport is a gRPC transport.
|
|
type Transport struct {
|
|
endpoint string
|
|
operation string
|
|
metadata metadata.Metadata
|
|
}
|
|
|
|
// Kind returns the transport kind.
|
|
func (tr *Transport) Kind() string {
|
|
return "grpc"
|
|
}
|
|
|
|
// Endpoint returns the transport endpoint.
|
|
func (tr *Transport) Endpoint() string {
|
|
return tr.endpoint
|
|
}
|
|
|
|
// Operation returns the transport operation.
|
|
func (tr *Transport) Operation() string {
|
|
return tr.operation
|
|
}
|
|
|
|
// SetOperation sets the transport operation.
|
|
func (tr *Transport) SetOperation(operation string) {
|
|
tr.operation = operation
|
|
}
|
|
|
|
// Metadata returns the transport metadata.
|
|
func (tr *Transport) Metadata() metadata.Metadata {
|
|
return tr.metadata
|
|
}
|
|
|
|
// WithMetadata with a metadata into transport md.
|
|
func (tr *Transport) WithMetadata(md metadata.Metadata) {
|
|
for k, v := range md {
|
|
tr.metadata.Set(k, v)
|
|
}
|
|
}
|
|
|