85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
// Copyright (C) 2015 Audrius Butkevicius and Contributors (see the CONTRIBUTORS file).
|
|
|
|
package client
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/syncthing/syncthing/lib/relay/protocol"
|
|
"github.com/syncthing/syncthing/lib/svcutil"
|
|
"github.com/syncthing/syncthing/lib/sync"
|
|
|
|
"github.com/thejerf/suture/v4"
|
|
)
|
|
|
|
type relayClientFactory func(uri *url.URL, certs []tls.Certificate, invitations chan protocol.SessionInvitation, timeout time.Duration) RelayClient
|
|
|
|
var (
|
|
supportedSchemes = map[string]relayClientFactory{
|
|
"relay": newStaticClient,
|
|
"dynamic+http": newDynamicClient,
|
|
"dynamic+https": newDynamicClient,
|
|
}
|
|
)
|
|
|
|
type RelayClient interface {
|
|
suture.Service
|
|
Error() error
|
|
Latency() time.Duration
|
|
String() string
|
|
Invitations() chan protocol.SessionInvitation
|
|
URI() *url.URL
|
|
}
|
|
|
|
func NewClient(uri *url.URL, certs []tls.Certificate, invitations chan protocol.SessionInvitation, timeout time.Duration) (RelayClient, error) {
|
|
factory, ok := supportedSchemes[uri.Scheme]
|
|
if !ok {
|
|
return nil, fmt.Errorf("unsupported scheme: %s", uri.Scheme)
|
|
}
|
|
|
|
return factory(uri, certs, invitations, timeout), nil
|
|
}
|
|
|
|
type commonClient struct {
|
|
svcutil.ServiceWithError
|
|
|
|
invitations chan protocol.SessionInvitation
|
|
closeInvitationsOnFinish bool
|
|
mut sync.RWMutex
|
|
}
|
|
|
|
func newCommonClient(invitations chan protocol.SessionInvitation, serve func(context.Context) error, creator string) commonClient {
|
|
c := commonClient{
|
|
invitations: invitations,
|
|
mut: sync.NewRWMutex(),
|
|
}
|
|
newServe := func(ctx context.Context) error {
|
|
defer c.cleanup()
|
|
return serve(ctx)
|
|
}
|
|
c.ServiceWithError = svcutil.AsService(newServe, creator)
|
|
if c.invitations == nil {
|
|
c.closeInvitationsOnFinish = true
|
|
c.invitations = make(chan protocol.SessionInvitation)
|
|
}
|
|
return c
|
|
}
|
|
|
|
func (c *commonClient) cleanup() {
|
|
c.mut.Lock()
|
|
if c.closeInvitationsOnFinish {
|
|
close(c.invitations)
|
|
}
|
|
c.mut.Unlock()
|
|
}
|
|
|
|
func (c *commonClient) Invitations() chan protocol.SessionInvitation {
|
|
c.mut.RLock()
|
|
defer c.mut.RUnlock()
|
|
return c.invitations
|
|
}
|