mirror of https://github.com/hashicorp/consul
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
|
// Copyright (c) HashiCorp, Inc.
|
||
|
|
// SPDX-License-Identifier: BUSL-1.1
|
||
|
|
|
||
|
|
package topoutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RequestRegisterService registers a service at the given node address
|
||
|
|
// using consul http request.
|
||
|
|
//
|
||
|
|
// The service definition must be a JSON string.
|
||
|
|
func RequestRegisterService(clusterHttpCli *http.Client, nodeAddress string, serviceDefinition string, token string) error {
|
||
|
|
var js json.RawMessage
|
||
|
|
if err := json.Unmarshal([]byte(serviceDefinition), &js); err != nil {
|
||
|
|
return fmt.Errorf("failed to unmarshal service definition: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
u, err := url.Parse(nodeAddress)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("failed to parse node address: %s", err)
|
||
|
|
}
|
||
|
|
u.Path = "/v1/agent/service/register"
|
||
|
|
|
||
|
|
req, err := http.NewRequest(http.MethodPut, u.String(), bytes.NewBuffer(js))
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("failed to create request: %s", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if token != "" {
|
||
|
|
req.Header.Set("X-Consul-Token", token)
|
||
|
|
}
|
||
|
|
|
||
|
|
resp, err := clusterHttpCli.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("failed to send request: %s", err)
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
|
||
|
|
if resp.StatusCode != http.StatusOK {
|
||
|
|
return fmt.Errorf("failed to register service: %s", resp.Status)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|