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
1.0 KiB
49 lines
1.0 KiB
package transport
|
|
|
|
import (
|
|
"context"
|
|
|
|
// init encoding
|
|
_ "github.com/go-kratos/kratos/v2/encoding/json"
|
|
_ "github.com/go-kratos/kratos/v2/encoding/proto"
|
|
_ "github.com/go-kratos/kratos/v2/encoding/xml"
|
|
_ "github.com/go-kratos/kratos/v2/encoding/yaml"
|
|
)
|
|
|
|
// Registry is registry endpoint.
|
|
type Registry interface {
|
|
Endpoint() (string, error)
|
|
}
|
|
|
|
// Server is transport server.
|
|
type Server interface {
|
|
Start() error
|
|
Stop() error
|
|
}
|
|
|
|
// Transport is transport context value.
|
|
type Transport struct {
|
|
Kind Kind
|
|
}
|
|
|
|
// Kind defines the type of Transport
|
|
type Kind string
|
|
|
|
// Defines a set of transport kind
|
|
const (
|
|
KindGRPC Kind = "gRPC"
|
|
KindHTTP Kind = "HTTP"
|
|
)
|
|
|
|
type transportKey struct{}
|
|
|
|
// NewContext returns a new Context that carries value.
|
|
func NewContext(ctx context.Context, tr Transport) context.Context {
|
|
return context.WithValue(ctx, transportKey{}, tr)
|
|
}
|
|
|
|
// FromContext returns the Transport value stored in ctx, if any.
|
|
func FromContext(ctx context.Context) (tr Transport, ok bool) {
|
|
tr, ok = ctx.Value(transportKey{}).(Transport)
|
|
return
|
|
}
|
|
|