mirror of
https://github.com/coder/coder.git
synced 2025-07-12 00:14:10 +00:00
37832413ba
Our internal drpc package name conflicts with the external one in usage. `drpc.*` == external `drpcsdk.*` == internal
25 lines
627 B
Go
25 lines
627 B
Go
package tailnet
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
|
|
"github.com/hashicorp/yamux"
|
|
"golang.org/x/xerrors"
|
|
|
|
"cdr.dev/slog"
|
|
"github.com/coder/coder/v2/codersdk/drpcsdk"
|
|
"github.com/coder/coder/v2/tailnet/proto"
|
|
)
|
|
|
|
func NewDRPCClient(conn net.Conn, logger slog.Logger) (proto.DRPCTailnetClient, error) {
|
|
config := yamux.DefaultConfig()
|
|
config.LogOutput = nil
|
|
config.Logger = slog.Stdlib(context.Background(), logger, slog.LevelInfo)
|
|
session, err := yamux.Client(conn, config)
|
|
if err != nil {
|
|
return nil, xerrors.Errorf("multiplex client: %w", err)
|
|
}
|
|
return proto.NewDRPCTailnetClient(drpcsdk.MultiplexedConn(session)), nil
|
|
}
|