0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package quickdb
|
|
|
|
import (
|
|
"crypto/tls"
|
|
|
|
"github.com/riza-io/grpc-go/credentials/basic"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/config"
|
|
pb "github.com/sqlc-dev/sqlc/internal/quickdb/v1"
|
|
"github.com/sqlc-dev/sqlc/internal/rpc"
|
|
)
|
|
|
|
const defaultHostname = "grpc.sqlc.dev"
|
|
|
|
func NewClientFromConfig(cloudConfig config.Cloud) (pb.QuickClient, error) {
|
|
projectID := cloudConfig.Project
|
|
return NewClient(projectID, cloudConfig.AuthToken, WithHost(cloudConfig.Hostname))
|
|
}
|
|
|
|
type options struct {
|
|
hostname string
|
|
}
|
|
|
|
type Option func(*options)
|
|
|
|
func WithHost(host string) Option {
|
|
return func(o *options) {
|
|
o.hostname = host
|
|
}
|
|
}
|
|
|
|
func NewClient(project, token string, opts ...Option) (pb.QuickClient, error) {
|
|
var o options
|
|
for _, apply := range opts {
|
|
apply(&o)
|
|
}
|
|
|
|
dialOpts := []grpc.DialOption{
|
|
grpc.WithTransportCredentials(credentials.NewTLS(&tls.Config{})),
|
|
grpc.WithPerRPCCredentials(basic.NewPerRPCCredentials(project, token)),
|
|
grpc.WithUnaryInterceptor(rpc.UnaryInterceptor),
|
|
}
|
|
|
|
hostname := o.hostname
|
|
if hostname == "" {
|
|
hostname = defaultHostname
|
|
}
|
|
|
|
conn, err := grpc.Dial(hostname+":443", dialOpts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return pb.NewQuickClient(conn), nil
|
|
}
|