mirror of
https://github.com/singchia/frontier.git
synced 2025-09-26 20:31:25 +08:00
78 lines
2.3 KiB
Go
78 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
|
|
"github.com/singchia/geminio"
|
|
"github.com/singchia/geminio/client"
|
|
)
|
|
|
|
// RPCer is edge and it's method oriented
|
|
type RPCer interface {
|
|
NewRequest(data []byte) geminio.Request
|
|
|
|
Call(ctx context.Context, edgeID uint64, method string, req geminio.Request) (geminio.Response, error)
|
|
CallAsync(ctx context.Context, edgeID uint64, method string, req geminio.Request, ch chan *geminio.Call) (*geminio.Call, error)
|
|
Register(ctx context.Context, method string, rpc geminio.RPC) error
|
|
}
|
|
|
|
// Messager is edge oriented
|
|
type Messager interface {
|
|
NewMessage(data []byte) geminio.Message
|
|
|
|
Publish(ctx context.Context, edgeID uint64, msg geminio.Message) error
|
|
PublishAsync(ctx context.Context, edgeID uint64, msg geminio.Message, ch chan *geminio.Publish) (*geminio.Publish, error)
|
|
Receive(ctx context.Context) (geminio.Message, error)
|
|
}
|
|
|
|
type RPCMessager interface {
|
|
RPCer
|
|
Messager
|
|
}
|
|
|
|
// Stream multiplexer
|
|
type Multiplexer interface {
|
|
// Open a stream to specific edgeID
|
|
OpenStream(ctx context.Context, edgeID uint64) (geminio.Stream, error)
|
|
AcceptStream() (geminio.Stream, error)
|
|
ListStreams() []geminio.Stream
|
|
}
|
|
|
|
// Controller functions
|
|
type GetEdgeID func(meta []byte) (uint64, error)
|
|
type EdgeOnline func(edgeID uint64, meta []byte, addr net.Addr) error
|
|
type EdgeOffline func(edgeID uint64, meta []byte, addr net.Addr) error
|
|
|
|
type ControlRegister interface {
|
|
RegisterGetEdgeID(ctx context.Context, getEdgeID GetEdgeID) error
|
|
RegisterEdgeOnline(ctx context.Context, edgeOnline EdgeOnline) error
|
|
RegisterEdgeOffline(ctx context.Context, edgeOffline EdgeOffline) error
|
|
}
|
|
|
|
// Service
|
|
type Service interface {
|
|
// Service can direct Message or RPC
|
|
RPCMessager
|
|
|
|
// Service can manage streams from or to a Edge
|
|
Multiplexer
|
|
|
|
// Service is a net.Listener, actually it's wrapper of Multiplexer
|
|
// The Accept is a wrapper for AcceptStream
|
|
// The Addr is a wrapper for LocalAddr
|
|
net.Listener
|
|
|
|
// Service can register some control functions that be called by frontier when edge updated
|
|
ControlRegister
|
|
|
|
Close() error
|
|
}
|
|
|
|
type Dialer func() (net.Conn, error)
|
|
|
|
// the service field specific the role for this Service, and then Edge can OpenStream to this service
|
|
func NewService(dialer Dialer, opts ...ServiceOption) (Service, error) {
|
|
return newServiceEnd(client.Dialer(dialer), opts...)
|
|
}
|