WIP: Refactor

schema/constraint.go

@@ -0,0 +1,78 @@
+package schema
+
+import (
+	"errors"
+	"strings"
+)
+
+type Constraint struct {
+	Name       string
+	Type       string
+	Columns    []Column
+	RefTable   *Table
+	RefColumns []Column
+}
+
+func (c *Constraint) GetDDL() (string, error) {
+	switch c.Type {
+	case "pk":
+		return c.getPkDDL(), nil
+	case "fk":
+		return c.getFkDDL(), nil
+	}
+
+	return "", errors.New("Unimplemented constraing: " + c.Type)
+}
+
+func (c *Constraint) getPkDDL() string {
+	var ddl strings.Builder
+	ddl.WriteString("CONSTRAINT " + c.Name + " PRIMARY KEY(")
+	for i, col := range c.Columns {
+		if i != 0 {
+			ddl.WriteString(", ")
+		}
+		ddl.WriteString(col.Name)
+	}
+	ddl.WriteString(")")
+	return ddl.String()
+}
+
+func (c *Constraint) getFkDDL() string {
+	var ddl strings.Builder
+	ddl.WriteString("CONSTRAINT " + c.Name + " FOREIGN KEY(")
+
+	for i, col := range c.Columns {
+		if i != 0 {
+			ddl.WriteString(", ")
+		}
+		ddl.WriteString(col.Name)
+	}
+	ddl.WriteString(") REFERENCES " + c.RefTable.Name + "(")
+
+	for i, col := range c.RefColumns {
+		if i != 0 {
+			ddl.WriteString(", ")
+		}
+		ddl.WriteString(col.Name)
+	}
+
+	ddl.WriteString(")")
+
+	return ddl.String()
+}
+
+func (Constraint) GetSelectDML() (string, error) {
+	return "", errors.New("No DML for constraints")
+}
+
+func (Constraint) GetInsertDML() (string, error) {
+	return "", errors.New("No DML for constraints")
+}
+
+func (Constraint) GetUpdateDML() (string, error) {
+	return "", errors.New("No DML for constraints")
+}
+
+func (Constraint) GetDeleteDML() (string, error) {
+	return "", errors.New("No DML for constraints")
+}