change type of migration type

Author: tmus

Migration.go

@@ -2,37 +2,27 @@ package exodus
 
 import "fmt"
 
-// Migration is a complete, raw SQL command that can be ran
-// against a database.
-type Migration struct {
-	SQL string
-}
-
-func (m Migration) String() string {
-	return string(m.SQL)
-}
+// Migration is a fully-formed SQL command that can be ran against
+// a database connection.
+type Migration string
 
 // MigrationInterface ...
 type MigrationInterface interface {
 	Up() Migration
 	Down() Migration
 }
 
-// Create generates a Migration that contains SQL to create
-// the schema against the table name provided.
+// Create generates an SQL command to create a table using the
+// schema provided.
 func Create(table string, schema Schema) Migration {
 	sql := parseColumns(loadColumnSQL(schema))
 
-	return Migration{
-		SQL: fmt.Sprintf("CREATE TABLE %s ( %s );", table, sql),
-	}
+	return Migration(fmt.Sprintf("CREATE TABLE %s ( %s );", table, sql))
 }
 
-// Drop generates an SQL command to drop a table.
+// Drop generates an SQL command to drop the given table.
 func Drop(table string) Migration {
-	return Migration{
-		SQL: fmt.Sprintf("DROP TABLE %s", table),
-	}
+	return Migration(fmt.Sprintf("DROP TABLE %s", table))
 }
 
 // loadColumnSQL iterates through the Columns defined in the

Migrator.go

@@ -135,7 +135,7 @@ func (m *Migrator) Run(migrations ...MigrationInterface) error {
 	batch := m.nextBatchNumber()
 
 	for _, migration := range migrations {
-		if _, err := m.DB.Exec(migration.Up().String()); err != nil {
+		if _, err := m.DB.Exec(string(migration.Up())); err != nil {
 			return err
 		}
 

column/UniqueSet.go

@@ -0,0 +1,39 @@
+package column
+
+import "fmt"
+
+type Constraint struct {
+	Name    string
+	Columns []string
+}
+
+// UniqueSet ...
+func UniqueSet(name string, columns ...string) *Constraint {
+	return &Constraint{
+		Name:    name,
+		Columns: columns,
+	}
+}
+
+func (us *Constraint) ToSQL() string {
+	sql := fmt.Sprintf(
+		"CONSTRAINT %s UNIQUE ( %s )",
+		us.Name,
+		us.getSet(),
+	)
+
+	fmt.Println(sql)
+
+	return sql
+}
+
+func (us *Constraint) getSet() (inline string) {
+	for i, column := range us.Columns {
+		inline = inline + column
+		if i < len(us.Columns)-1 {
+			inline = inline + ","
+		}
+	}
+
+	return
+}