mirror of
https://github.com/buger/goreplay.git
synced 2024-04-21 12:32:02 +00:00
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
// Copyright 2013 Matthew Baird
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package elastigo
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
)
|
|
|
|
// The cluster health API allows to get a very simple status on the health of the cluster.
|
|
// see http://www.elasticsearch.org/guide/reference/api/admin-cluster-health.html
|
|
// information returned. Defaults to cluster.)
|
|
func (c *Conn) Reroute(dryRun bool, commands Commands) (ClusterHealthResponse, error) {
|
|
var url string
|
|
var retval ClusterHealthResponse
|
|
|
|
if len(commands.Commands) > 0 {
|
|
url = fmt.Sprintf("/_cluster/reroute%s&%s", dryRunOption(dryRun))
|
|
} else {
|
|
return retval, errors.New("Must pass at least one command")
|
|
}
|
|
m := map[string]interface{}{"commands": commands.Commands}
|
|
body, err := c.DoCommand("POST", url, m, nil)
|
|
if err != nil {
|
|
return retval, err
|
|
}
|
|
if err == nil {
|
|
// marshall into json
|
|
jsonErr := json.Unmarshal(body, &retval)
|
|
if jsonErr != nil {
|
|
return retval, jsonErr
|
|
}
|
|
}
|
|
return retval, err
|
|
}
|
|
|
|
func dryRunOption(isDryRun bool) string {
|
|
if isDryRun {
|
|
return "dry_run"
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// supported commands are
|
|
// move (index, shard, from_node, to_node)
|
|
// cancel (index, shard, node, allow_primary)
|
|
// allocate (index, shard, node, allow_primary)
|
|
|
|
type Commands struct {
|
|
Commands []interface{} `json:"commands"`
|
|
}
|
|
|
|
type MoveCommand struct {
|
|
Index string `json:"index"`
|
|
Shard string `json:"shard"`
|
|
FromNode string `json:"from_node"`
|
|
ToNode string `json:"to_node"`
|
|
}
|
|
|
|
type CancelCommand struct {
|
|
Index string `json:"index"`
|
|
Shard string `json:"shard"`
|
|
Node string `json:"node"`
|
|
AllowPrimary bool `json:"allow_primary,omitempty"`
|
|
}
|
|
type AllocateCommand struct {
|
|
Index string `json:"index"`
|
|
Shard string `json:"shard"`
|
|
Node string `json:"node"`
|
|
AllowPrimary bool `json:"allow_primary,omitempty"`
|
|
}
|