added: main.go implementation for the main driver

Author: kmacmcfarlane

.idea/workspace.xml

@@ -4,11 +4,12 @@ all: |build_deps test_deps build test
 
 build_deps:
 	go get github.com/lib/pq
+	go get github.com/iancoleman/strcase
 
 test_deps:
 	go get -u github.com/onsi/ginkgo/ginkgo
 	go get -u github.com/onsi/gomega/...
-	@go get github.com/vektra/mockery/.../
+	go get github.com/vektra/mockery/.../
 
 build:
 	go build -o ./bin/csv-2-postgresql ./csv-2-postgresql

Makefile

@@ -4,8 +4,13 @@ import (
 	"database/sql"
 	"flag"
 	"fmt"
+	"github.com/iancoleman/strcase"
+	"github.com/kmacmcfarlane/csv-2-postgresql-go/csv"
 	"github.com/kmacmcfarlane/csv-2-postgresql-go/postgresql"
+	"io"
 	"os"
+	"path/filepath"
+	"strings"
 )
 
 func main() {
@@ -19,7 +24,14 @@ func main() {
 	}
 
 	inputFile := args[0]
-	databaseName := args[1]
+
+
+	var databaseName string
+	if len(args) == 2 {
+		databaseName = args[1]
+	} else {
+		databaseName = "test"
+	}
 
 	// Connect to DB
 	connectionString := fmt.Sprintf("user=postgres dbname=%s sslmode=verify-full", databaseName)
@@ -34,10 +46,62 @@ func main() {
 
 	sqlWrapper := postgresql.NewSQLWrapper(db)
 
+	dbWriter := postgresql.NewWriter(sqlWrapper)
+
 	// Parse schema of input
+	file, err := os.Open(inputFile)
+
+	if nil != err {
+		panic(fmt.Sprintf("error opening file for reading: %s", err.Error()))
+	}
+
+	var parser Parser = csv.NewParser(file)
+
+	schema, err := parser.ParseSchema()
+
+	if nil != err {
+		panic(fmt.Sprintf("error parsing csv file: %s", err.Error()))
+	}
+
+	// Create table
+	_, filenameAlone := filepath.Split(file.Name())
+	tableName := strcase.ToSnake(filenameAlone)
+
+	err = dbWriter.CreateTable(tableName, schema)
+
+	if nil != err {
+		panic(fmt.Sprintf("error creating db table: %s", err.Error()))
+	}
+
+	fmt.Printf("created table '%s' in database %s", tableName, databaseName)
 
 	// Insert data to database
-	
+	for {
+
+		record, err := parser.Read()
+
+		if nil != err {
+
+			if io.EOF == err {
+				break
+			}
+
+			fmt.Printf("error parsing record: %s", strings.Join(record, ", "))
+			fmt.Printf("error: %s", err.Error())
+		}
+
+		err = dbWriter.Insert(record, schema, tableName)
+
+		if nil != err {
+
+			fmt.Printf("error inserting record: %s", strings.Join(record, ", "))
+			fmt.Printf("error: %s", err.Error())
+		}
+	}
+
+	println("import ")
+
+	os.Exit(0)
 }
 
 func usage(){

bin/csv-2-postgresql

@@ -74,9 +74,7 @@ func (w Writer) Insert(values []string, schema schema.Schema, tableName string)
 		}
 	}
 
-	statementTemplate := `INSERT INTO "%s" VALUES (%s);`
-
-	statement := fmt.Sprintf(statementTemplate, tableName, sb.String())
+	statement := fmt.Sprintf(`INSERT INTO "%s" VALUES (%s);`, tableName, sb.String())
 
 	_, err = w.db.Exec(statement, values)