// Code generated by sdkgen. DO NOT EDIT.

package websocket

import (
	"context"

	"google.golang.org/grpc"
)

// Websocket provides access to "websocket" component of Yandex.Cloud
type Websocket struct {
	getConn func(ctx context.Context) (*grpc.ClientConn, error)
}

// NewWebsocket creates instance of Websocket
func NewWebsocket(g func(ctx context.Context) (*grpc.ClientConn, error)) *Websocket {
	return &Websocket{g}
}

// Connection gets ConnectionService client
func (w *Websocket) Connection() *ConnectionServiceClient {
	return &ConnectionServiceClient{getConn: w.getConn}
}
