Merge pull request #1 from k1LoW/add-driver-interface

Add driver interface

db/db.go

@@ -1,13 +1,18 @@
 package db
 
 import (
+	"database/sql"
 	"fmt"
 	"github.com/k1LoW/tbls/drivers/postgres"
 	"github.com/k1LoW/tbls/schema"
 	"github.com/xo/dburl"
 	"strings"
 )
 
+type Driver interface {
+	Analize(*sql.DB, *schema.Schema) error
+}
+
 func Analyze(urlstr string) (*schema.Schema, error) {
 	s := &schema.Schema{}
 	u, err := dburl.Parse(urlstr)
@@ -22,14 +27,18 @@ func Analyze(urlstr string) (*schema.Schema, error) {
 		return s, err
 	}
 	defer db.Close()
+
+	var driver Driver
+
 	switch u.Driver {
 	case "postgres":
-		err := postgres.Analize(db, s)
-		if err != nil {
-			return s, err
-		}
+		driver = new(postgres.Postgres)
 	default:
 		return s, fmt.Errorf("Error: %s", "unsupported driver")
 	}
+	err = driver.Analize(db, s)
+	if err != nil {
+		return s, err
+	}
 	return s, nil
 }

drivers/postgres/postgres.go

@@ -8,9 +8,12 @@ import (
 	"strings"
 )
 
-var ReFK = regexp.MustCompile(`FOREIGN KEY \((.+)\) REFERENCES (.+)\((.+)\)`)
+var reFK = regexp.MustCompile(`FOREIGN KEY \((.+)\) REFERENCES (.+)\((.+)\)`)
 
-func Analize(db *sql.DB, s *schema.Schema) error {
+type Postgres struct{}
+
+// Analize PostgreSQL database schema
+func (p *Postgres) Analize(db *sql.DB, s *schema.Schema) error {
 
 	// tables
 	tableRows, err := db.Query(`
@@ -198,7 +201,7 @@ WHERE table_name = $1`, tableName)
 
 	// Relations
 	for _, r := range relations {
-		result := ReFK.FindAllStringSubmatch(r.Def, -1)
+		result := reFK.FindAllStringSubmatch(r.Def, -1)
 		strColumns := strings.Split(result[0][1], ", ")
 		strParentTable := result[0][2]
 		strParentColumns := strings.Split(result[0][3], ", ")