Add optimize, refresh, and flush

Author: olivere

flush.go

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"log"
+)
+
+var cmdFlush = &Command{
+	Run:   runFlush,
+	Usage: "flush [index]",
+	Short: "flushes indices",
+	Long: `
+Flushes data to the index storage, frees memory and clears the
+internal transaction log. You often need to flush before updating
+ElasticSearch to a new version.
+
+Example:
+
+  $ es flush
+  $ es flush twitter
+`,
+	ApiUrl: "http://www.elasticsearch.org/guide/reference/api/admin-indices-flush.html",
+}
+
+func runFlush(cmd *Command, args []string) {
+	index := ""
+	if len(args) >= 1 {
+		index = args[0]
+	}
+
+	var response struct {
+		Ok     bool   `json:"ok,omitempty"`
+		Error  string `json:"error,omitempty"`
+		Status int    `json:"status,omitempty"`
+	}
+
+	if len(index) > 0 {
+		ESReq("POST", "/"+index+"/_flush").Do(&response)
+	} else {
+		ESReq("POST", "/_flush").Do(&response)
+	}
+
+	if len(response.Error) > 0 {
+		log.Fatalf("Error: %v (%v)\n", response.Error, response.Status)
+	}
+}

main.go

@@ -14,6 +14,8 @@ const (
 )
 
 var force bool
+var flush bool
+var refresh bool
 
 type Command struct {
 	Run  func(cmd *Command, args []string)
@@ -47,6 +49,9 @@ var commands = []*Command{
 	cmdDeleteIndex,
 	cmdStatus,
 	cmdStats,
+	cmdRefresh,
+	cmdOptimize,
+	cmdFlush,
 
 	cmdMapping,
 	cmdPutMapping,

optimize.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"log"
+	"net/url"
+)
+
+var cmdOptimize = &Command{
+	Run:   runOptimize,
+	Usage: "optimize [-r -f] [index]",
+	Short: "optimizes indices",
+	Long: `
+Optimize basically optimizes the index for faster search operations.
+
+Use -r, -r=true, -r=false to enable/disable refresh (enabled by default).
+Use -f, -f=true, -f=false to enable/disable flush (enabled by default).
+
+Example:
+
+  $ es optimize
+  $ es optimize -f=false twitter
+  $ es optimize twitter
+`,
+	ApiUrl: "http://www.elasticsearch.org/guide/reference/api/admin-indices-optimize.html",
+}
+
+func init() {
+	cmdOptimize.Flag.BoolVar(&flush, "f", true, "flush")
+	cmdOptimize.Flag.BoolVar(&refresh, "r", true, "refresh")
+}
+
+func runOptimize(cmd *Command, args []string) {
+	index := ""
+	if len(args) >= 1 {
+		index = args[len(args)-1]
+	}
+
+	values := url.Values{}
+	if flush {
+		values.Set("flush", "1")
+	} else {
+		values.Set("flush", "0")
+	}
+	if refresh {
+		values.Set("refresh", "1")
+	} else {
+		values.Set("refresh", "0")
+	}
+
+	var response struct {
+		Ok     bool   `json:"ok,omitempty"`
+		Error  string `json:"error,omitempty"`
+		Status int    `json:"status,omitempty"`
+	}
+
+	if len(index) > 0 {
+		ESReq("POST", "/"+index+"/_optimize?"+values.Encode()).Do(&response)
+	} else {
+		ESReq("POST", "/_optimize?"+values.Encode()).Do(&response)
+	}
+
+	if len(response.Error) > 0 {
+		log.Fatalf("Error: %v (%v)\n", response.Error, response.Status)
+	}
+}

refresh.go

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"log"
+)
+
+var cmdRefresh = &Command{
+	Run:   runRefresh,
+	Usage: "refresh [index]",
+	Short: "refreshes indices",
+	Long: `
+Refresh allows to explicitly refresh one or more index, 
+making all operations performed since the last refresh 
+available for search.
+
+Example:
+
+  $ es refresh
+  $ es refresh twitter
+`,
+	ApiUrl: "http://www.elasticsearch.org/guide/reference/api/admin-indices-refresh.html",
+}
+
+func runRefresh(cmd *Command, args []string) {
+	index := ""
+	if len(args) >= 1 {
+		index = args[0]
+	}
+
+	var response struct {
+		Ok     bool   `json:"ok,omitempty"`
+		Error  string `json:"error,omitempty"`
+		Status int    `json:"status,omitempty"`
+	}
+
+	if len(index) > 0 {
+		ESReq("POST", "/"+index+"/_refresh").Do(&response)
+	} else {
+		ESReq("POST", "/_refresh").Do(&response)
+	}
+
+	if len(response.Error) > 0 {
+		log.Fatalf("Error: %v (%v)\n", response.Error, response.Status)
+	}
+}