Run all queries with CONCURRENTLY option; Remove concurrency option (redundant); go mod

Author: joaodlf

go.mod

@@ -0,0 +1,12 @@
+module github.com/joaodlf/pgpi
+
+go 1.13
+
+require (
+	github.com/fatih/color v1.6.0
+	github.com/lib/pq v0.0.0-20180201184707-88edab080323
+	github.com/mattn/go-colorable v0.0.9 // indirect
+	github.com/mattn/go-isatty v0.0.3 // indirect
+	golang.org/x/crypto v0.0.0-20180228161326-91a49db82a88
+	golang.org/x/sys v0.0.0-20180302081741-dd2ff4accc09 // indirect
+)

go.sum

@@ -0,0 +1,12 @@
+github.com/fatih/color v1.6.0 h1:66qjqZk8kalYAvDRtM1AdAJQI0tj4Wrue3Eq3B3pmFU=
+github.com/fatih/color v1.6.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=
+github.com/lib/pq v0.0.0-20180201184707-88edab080323 h1:Ou506ViB5uo2GloKFWIYi5hwRJn4AAOXuLVv8RMY9+4=
+github.com/lib/pq v0.0.0-20180201184707-88edab080323/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
+github.com/mattn/go-colorable v0.0.9 h1:UVL0vNpWh04HeJXV0KLcaT7r06gOH2l4OW6ddYRUIY4=
+github.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU=
+github.com/mattn/go-isatty v0.0.3 h1:ns/ykhmWi7G9O+8a448SecJU3nSMBXJfqQkl0upE1jI=
+github.com/mattn/go-isatty v0.0.3/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4=
+golang.org/x/crypto v0.0.0-20180228161326-91a49db82a88 h1:jLkAo/qlT9whgCLYC5GAJ9kcKrv3Wj8VCc4N+KJ4wpw=
+golang.org/x/crypto v0.0.0-20180228161326-91a49db82a88/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4=
+golang.org/x/sys v0.0.0-20180302081741-dd2ff4accc09 h1:wNPZbZUOH0tyqngVRXeF2iQm19+ssqyebJTCFBvxsow=
+golang.org/x/sys v0.0.0-20180302081741-dd2ff4accc09/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=

main.go

@@ -8,7 +8,6 @@ import (
 	"os"
 	"os/signal"
 	"strconv"
-	"sync"
 	"syscall"
 
 	"github.com/fatih/color"
@@ -56,20 +55,6 @@ func validateConfirmation(value string) (bool, error) {
 	return false, errors.New("invalid value supplied")
 }
 
-// createPartition executes the CREATE INDEX queries.
-func createPartition(db *sql.DB, query string, wg *sync.WaitGroup, concurrency bool) {
-	if concurrency == true {
-		defer wg.Done()
-	}
-
-	cyan.Println(fmt.Sprintf("Executing '%s'", query))
-	_, err := db.Exec(query)
-
-	if err != nil {
-		red.Println(err)
-	}
-}
-
 func main() {
 	scanner := bufio.NewScanner(os.Stdin)
 
@@ -150,7 +135,7 @@ func main() {
 	yellow.Println("Connnected!")
 
 tr:
-// TABLE NAME REGEX.
+	// TABLE NAME REGEX.
 	tableNameRegex := ""
 	for {
 		green.Print("Table name regex to apply index (E.g. tablename_.*_.*): ")
@@ -234,21 +219,10 @@ tr:
 		}
 	}
 
-	var concurrency bool
-	for {
-		green.Print("Execute queries concurrently? [y/n]: ")
-		scanner.Scan()
-		concurrency, err = validateConfirmation(scanner.Text())
-
-		if err == nil {
-			break
-		}
-	}
-
 	var queries []string
 	for _, table := range tables {
 		fullIndexName := fmt.Sprintf("%s_%s", table, indexName)
-		indexQuery := fmt.Sprintf(`CREATE %s INDEX %s ON %s (%s);`, uniqueIndexStr, fullIndexName, table, indexColumns)
+		indexQuery := fmt.Sprintf(`CREATE %s INDEX CONCURRENTLY %s ON %s (%s);`, uniqueIndexStr, fullIndexName, table, indexColumns)
 		queries = append(queries, indexQuery)
 		cyan.Println(indexQuery)
 	}
@@ -267,17 +241,14 @@ tr:
 	if !execute {
 		os.Exit(0)
 	}
-
-	var wg sync.WaitGroup
 	for _, query := range queries {
-		if concurrency {
-			wg.Add(1)
-			go createPartition(db, query, &wg, concurrency)
-		} else {
-			createPartition(db, query, &wg, concurrency)
+		cyan.Println(fmt.Sprintf("Executing '%s'", query))
+		_, err := db.Exec(query)
+
+		if err != nil {
+			red.Println(err)
 		}
 	}
 
-	wg.Wait()
 	green.Println("All queries executed!")
 }