chore(client): support gRPC and gRPC web protocol

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu
2022-08-14 10:59:09 +08:00
committed by Jason Song
parent 0b885c5e5f
commit d359276fe1
5 changed files with 51 additions and 1 deletions

View File

@ -15,13 +15,19 @@ import (
)
// New returns a new runner client.
func New(endpoint, secret string, skipverify bool) *HTTPClient {
func New(endpoint, secret string, skipverify bool, opts ...Option) *HTTPClient {
client := &HTTPClient{
Endpoint: endpoint,
Secret: secret,
SkipVerify: skipverify,
}
// Loop through each option
for _, opt := range opts {
// Call the option giving the instantiated
opt.Apply(client)
}
client.Client = &http.Client{
Timeout: 5 * time.Second,
CheckRedirect: func(*http.Request, []*http.Request) error {
@ -57,6 +63,8 @@ type HTTPClient struct {
Endpoint string
Secret string
SkipVerify bool
opts []connect.ClientOption
}
// Ping sends a ping message to the server to test connectivity.
@ -64,6 +72,7 @@ func (p *HTTPClient) Ping(ctx context.Context, machine string) error {
client := v1connect.NewPingServiceClient(
p.Client,
p.Endpoint,
p.opts...,
)
req := connect.NewRequest(&v1.PingRequest{
Data: machine,