mirror of https://github.com/k3s-io/k3s
37 lines
747 B
Go
37 lines
747 B
Go
|
package dialer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"crypto/tls"
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
|
||
|
|
"github.com/canonical/go-dqlite/client"
|
||
|
|
"github.com/rancher/k3s/pkg/dqlite/pipe"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewHTTPDialer(advertiseAddress, bindAddress string, tls *tls.Config) (client.DialFunc, error) {
|
||
|
|
d := &dialer{
|
||
|
|
advertiseAddress: advertiseAddress,
|
||
|
|
bindAddress: bindAddress,
|
||
|
|
tls: tls,
|
||
|
|
}
|
||
|
|
|
||
|
|
return d.dial, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type dialer struct {
|
||
|
|
advertiseAddress string
|
||
|
|
bindAddress string
|
||
|
|
tls *tls.Config
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *dialer) dial(ctx context.Context, address string) (net.Conn, error) {
|
||
|
|
if address == d.advertiseAddress {
|
||
|
|
return net.Dial("unix", d.bindAddress)
|
||
|
|
}
|
||
|
|
|
||
|
|
url := fmt.Sprintf("https://%s/db/connect", address)
|
||
|
|
return pipe.ToHTTP(ctx, url, d.tls)
|
||
|
|
}
|