You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
5 years ago
|
package main
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/json"
|
||
|
"fmt"
|
||
|
"net/http"
|
||
|
"net/url"
|
||
|
"strings"
|
||
|
)
|
||
|
|
||
|
func removeNode(client *http.Client, id string, argv *argT, timer bool) error {
|
||
|
u := url.URL{
|
||
|
Scheme: argv.Protocol,
|
||
|
Host: fmt.Sprintf("%s:%d", argv.Host, argv.Port),
|
||
|
Path: fmt.Sprintf("%sremove", argv.Prefix),
|
||
|
}
|
||
|
urlStr := u.String()
|
||
|
|
||
|
b, err := json.Marshal(map[string]string{
|
||
|
"id": id,
|
||
|
})
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
nRedirect := 0
|
||
|
for {
|
||
|
req, err := http.NewRequest("DELETE", urlStr, bytes.NewReader(b))
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if argv.Credentials != "" {
|
||
|
creds := strings.Split(argv.Credentials, ":")
|
||
|
if len(creds) != 2 {
|
||
|
return fmt.Errorf("invalid Basic Auth credentials format")
|
||
|
}
|
||
|
req.SetBasicAuth(creds[0], creds[1])
|
||
|
}
|
||
|
|
||
|
resp, err := client.Do(req)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if resp.StatusCode == http.StatusUnauthorized {
|
||
|
return fmt.Errorf("unauthorized")
|
||
|
}
|
||
|
|
||
|
if resp.StatusCode == http.StatusMovedPermanently {
|
||
|
nRedirect++
|
||
|
if nRedirect > maxRedirect {
|
||
|
return fmt.Errorf("maximum leader redirect limit exceeded")
|
||
|
}
|
||
|
urlStr = resp.Header["Location"][0]
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
if resp.StatusCode != http.StatusOK {
|
||
|
return fmt.Errorf("server responded with: %s", resp.Status)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
}
|