start with the new generated things

Author: BrandonRoehl

dump.go

@@ -327,15 +327,30 @@ func (table *table) Init() (err error) {
 	table.values = make([]interface{}, len(tt))
 	for i, tp := range tt {
 		st := tp.ScanType()
-		if tp.DatabaseTypeName() == "BLOB" {
+		dt := tp.DatabaseTypeName()
+
+		// SHOW FIELDS FROM jamfsoftware.computer_installed_certificates;
+		// if tp.Name() == "cert_blob_lookup_hash" {
+		// 	fmt.Println("why though =", st, ":", dt)
+		// }
+
+		if dt == "BLOB" || dt == "BINARY" {
 			t = reflect.TypeOf(sql.RawBytes{})
-		} else if st != nil && (st.Kind() == reflect.Int ||
+		} else if dt == "VARCHAR" || dt == "TEXT" || dt == "DECIMAL" {
+			t = reflect.TypeOf(sql.NullString{})
+		} else if (st != nil && (st.Kind() == reflect.Int ||
 			st.Kind() == reflect.Int8 ||
 			st.Kind() == reflect.Int16 ||
 			st.Kind() == reflect.Int32 ||
-			st.Kind() == reflect.Int64) {
+			st.Kind() == reflect.Int64)) ||
+			dt == "BIGINT" || dt == "TINYINT" || dt == "INT" {
 			t = reflect.TypeOf(sql.NullInt64{})
+		} else if (st != nil && (st.Kind() == reflect.Float32 ||
+			st.Kind() == reflect.Float64)) ||
+			tp.DatabaseTypeName() == "DOUBLE" {
+			t = reflect.TypeOf(sql.NullFloat64{})
 		} else {
+			// unknown datatype
 			t = reflect.TypeOf(sql.NullString{})
 		}
 		table.values[i] = reflect.New(t).Interface()

go.mod

@@ -2,6 +2,7 @@ module github.com/jamf/go-mysqldump
 
 require (
 	github.com/DATA-DOG/go-sqlmock v1.3.0
+	github.com/go-sql-driver/mysql v1.5.0
 	github.com/stretchr/testify v1.4.0
 )
 

go.sum

@@ -2,6 +2,8 @@ github.com/DATA-DOG/go-sqlmock v1.3.0 h1:ljjRxlddjfChBJdFKJs5LuCwCWPLaC1UZLwAo3P
 github.com/DATA-DOG/go-sqlmock v1.3.0/go.mod h1:f/Ixk793poVmq4qj/V1dPUg2JEAKC73Q5eFN3EC/SaM=
 github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/go-sql-driver/mysql v1.5.0 h1:ozyZYNQW3x3HtqT1jira07DN2PArx2v7/mN66gGcHOs=
+github.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=

random_test.go

@@ -0,0 +1,52 @@
+package mysqldump_test
+
+import (
+	"database/sql"
+	"fmt"
+	"os"
+	"testing"
+
+	"github.com/go-sql-driver/mysql"
+	"github.com/jamf/go-mysqldump"
+)
+
+var socketFiles = []string{
+	"/tmp/mysql.sock",
+	"/var/run/mysqld/mysqld.sock",
+	"/var/lib/mysql/mysql.sock",
+}
+
+func findSocketFile() string {
+	// Search under known defaults
+	for _, socket := range socketFiles {
+		if _, err := os.Stat(socket); err == nil {
+			fmt.Println("default mysqld socket found:", socket)
+			return socket
+		}
+	}
+	return socketFiles[0]
+}
+
+func TestDumpMyLocal(t *testing.T) {
+	config := mysql.NewConfig()
+	config.MaxAllowedPacket = 0
+	config.DBName = "jamfsoftware"
+	config.User = "jamfsoftware"
+	config.Passwd = "jamfsw03"
+	config.Net = "unix"
+	config.Addr = findSocketFile()
+
+	db, err := sql.Open("mysql", config.FormatDSN())
+	if err != nil {
+		t.Error(err)
+	}
+
+	dump := &mysqldump.Data{
+		Out:        os.Stdout,
+		Connection: db,
+	}
+
+	if err := dump.Dump(); err != nil {
+		t.Error(err)
+	}
+}