Add possibility to modify max allowed packet size

Greg0 · Jul 18, 2019 · 664e822 · 664e822
1 parent 5cd7a5e
commit 664e822
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 3 deletions.
diff --git a/go.mod b/go.mod
@@ -5,7 +5,7 @@ go 1.12
 require (
 	github.com/fatih/color v1.7.0
 	github.com/go-sql-driver/mysql v1.4.1
-	github.com/greg0/go-mysqldump v0.4.4
+	github.com/greg0/go-mysqldump v0.4.7
 	github.com/mattn/go-colorable v0.1.2 // indirect
 	gopkg.in/yaml.v2 v2.2.2
 )
diff --git a/go.sum b/go.sum
@@ -6,6 +6,12 @@ github.com/greg0/go-mysqldump v0.4.3 h1:022rVmevOvXCOycKENOtKO3u4gO37THch2Lzxh+E
 github.com/greg0/go-mysqldump v0.4.3/go.mod h1:mnGQCn4TO0r/+hujgI4HiJVukSEErNyNXtbexRbiTME=
 github.com/greg0/go-mysqldump v0.4.4 h1:oNOxeQvaGjZCugI3TgmYeg9im/65RKWVvW/YgkBD3Ek=
 github.com/greg0/go-mysqldump v0.4.4/go.mod h1:mnGQCn4TO0r/+hujgI4HiJVukSEErNyNXtbexRbiTME=
+github.com/greg0/go-mysqldump v0.4.5 h1:bLfDjvL2s8BSXESRFOE9qFelxxh4mXv6GYi22DZqarg=
+github.com/greg0/go-mysqldump v0.4.5/go.mod h1:mnGQCn4TO0r/+hujgI4HiJVukSEErNyNXtbexRbiTME=
+github.com/greg0/go-mysqldump v0.4.6 h1:Wp+otD4sBaLKhvv0rDRCDXH/o2dBcz+uxyVcMSyItag=
+github.com/greg0/go-mysqldump v0.4.6/go.mod h1:mnGQCn4TO0r/+hujgI4HiJVukSEErNyNXtbexRbiTME=
+github.com/greg0/go-mysqldump v0.4.7 h1:gEEXR25u4abFuHaSyQnDhy1y5+IgXWaiJGNnhYXHkSo=
+github.com/greg0/go-mysqldump v0.4.7/go.mod h1:mnGQCn4TO0r/+hujgI4HiJVukSEErNyNXtbexRbiTME=
 github.com/mattn/go-colorable v0.1.2 h1:/bC9yWikZXAL9uJdulbSfyVNIR3n3trXl+v8+1sx8mU=
 github.com/mattn/go-colorable v0.1.2/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
 github.com/mattn/go-isatty v0.0.8 h1:HLtExJ+uU2HOZ+wI0Tt5DtUDrx8yhUqDcp7fYERX4CE=

diff --git a/main.go b/main.go
@@ -54,6 +54,7 @@ type Options struct {
 	IgnoredTables       []string
 	StructureOnlyTables []string
 	OutputDirectory     string
+	PacketSize          int
 }
 
 func main() {
@@ -82,6 +83,7 @@ func main() {
 		dumpFilenameFormat,
 		options.StructureOnlyTables,
 		options.IgnoredTables,
+		options.PacketSize,
 	)
 	if err != nil {
 		fmt.Println("Error registering databse:", err)
@@ -131,6 +133,9 @@ func GetOptions() *Options {
 	var outputdir string
 	flag.StringVar(&outputdir, "output", "", "Dump output dir. Default is current working directory")
 
+	var packetSize int
+	flag.IntVar(&packetSize, "packetSize", 1048576, "Max allowed packet size. Default 1048576 (1MB)")
+
 	if len(os.Args) <= 1 || os.Args[1] == "--help" {
 		flag.PrintDefaults()
 		os.Exit(1)
@@ -179,7 +184,8 @@ func GetOptions() *Options {
 		database,
 		ignoredTablesArray,
 		structOnlyTablesArray,
-		outputdir)
+		outputdir,
+		packetSize)
 
 	stropts, _ := json.MarshalIndent(opts, "", "\t")
 	printMessage("Running with parameters", Info)
@@ -205,7 +211,7 @@ func FillArrayWithFileLines(filePath string) []string {
 	return array
 }
 
-func (o *Options) create(name string, address string, username string, password string, database string, ignoredTables []string, structOnlyTables []string, outputDir string) *Options {
+func (o *Options) create(name string, address string, username string, password string, database string, ignoredTables []string, structOnlyTables []string, outputDir string, packetSize int) *Options {
 
 	database = strings.Replace(database, " ", "", -1)
 	database = strings.Replace(database, " , ", ",", -1)
@@ -221,6 +227,7 @@ func (o *Options) create(name string, address string, username string, password
 		IgnoredTables:       ignoredTables,
 		StructureOnlyTables: structOnlyTables,
 		OutputDirectory:     outputDir,
+		PacketSize:          packetSize,
 	}
 }