54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
|
package network
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/tls"
|
||
|
|
"github.com/lucas-clemente/quic-go"
|
||
|
|
|
||
|
|
"git.hpds.cc/Component/network/auth"
|
||
|
|
"git.hpds.cc/Component/network/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ClientOptions are the options for HPDS client.
|
||
|
|
type ClientOptions struct {
|
||
|
|
ObserveDataTags []byte
|
||
|
|
QuicConfig *quic.Config
|
||
|
|
TLSConfig *tls.Config
|
||
|
|
Credential *auth.Credential
|
||
|
|
Logger log.Logger
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithObserveDataTags sets data tag list for the client.
|
||
|
|
func WithObserveDataTags(tags ...byte) ClientOption {
|
||
|
|
return func(o *ClientOptions) {
|
||
|
|
o.ObserveDataTags = tags
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithCredential sets the client credential method (used by client).
|
||
|
|
func WithCredential(payload string) ClientOption {
|
||
|
|
return func(o *ClientOptions) {
|
||
|
|
o.Credential = auth.NewCredential(payload)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithClientTLSConfig sets tls config for the client.
|
||
|
|
func WithClientTLSConfig(tc *tls.Config) ClientOption {
|
||
|
|
return func(o *ClientOptions) {
|
||
|
|
o.TLSConfig = tc
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithClientQuicConfig sets quic config for the client.
|
||
|
|
func WithClientQuicConfig(qc *quic.Config) ClientOption {
|
||
|
|
return func(o *ClientOptions) {
|
||
|
|
o.QuicConfig = qc
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithLogger sets logger for the client.
|
||
|
|
func WithLogger(logger log.Logger) ClientOption {
|
||
|
|
return func(o *ClientOptions) {
|
||
|
|
o.Logger = logger
|
||
|
|
}
|
||
|
|
}
|