enable to skip cluster status check, use index instead of create for …

…bulk operations
medcl · Sep 5, 2019 · c4580cb · c4580cb
1 parent 7dc95a9
commit c4580cb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/bulk.go b/bulk.go
@@ -121,7 +121,7 @@ func (c *Migrator) NewBulkWorker(docCount *int, pb *pb.ProgressBar, wg *sync.Wai
 
 		// encode the doc and and the _source field for a bulk request
 			post := map[string]Document{
-				"create": doc,
+				"index": doc,
 			}
 			if err = docEnc.Encode(post); err != nil {
 				log.Error(err)

diff --git a/main.go b/main.go
@@ -11,7 +11,7 @@ import (
 	"bufio"
 	log "github.com/cihub/seelog"
 	goflags "github.com/jessevdk/go-flags"
-	pb "gopkg.in/cheggaaa/pb.v1"
+	"gopkg.in/cheggaaa/pb.v1"
 	"os"
 	"io"
 	"io/ioutil"
@@ -485,8 +485,12 @@ func (c *Migrator) ClusterVersion(host string, auth *Auth,proxy string) (*Cluste
 }
 
 func (c *Migrator) ClusterReady(api ESAPI) (*ClusterHealth, bool) {
-
 	health := api.ClusterHealth()
+
+	if !c.Config.WaitForGreen{
+		return health,true
+	}
+
 	if health.Status == "red" {
 		return health, false
 	}