Have ProcessTable tell if it's EAV

mpchadwick · Apr 6, 2019 · bdc7c6c · bdc7c6c
1 parent 27693ba
commit bdc7c6c
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 8 deletions.
diff --git a/src/config.go b/src/config.go
@@ -44,14 +44,20 @@ func NewConfig(requested string) (*Config, error) {
 }
 
 
-func (c Config) ProcessTable(t string) bool {
+func (c Config) ProcessTable(t string) string {
 	for _, table := range c.Tables {
 		if (table.Name == t) {
-			return true
+			return "table"
 		}
 	}
 
-	return false
+	for _, e := range c.Eav {
+		if (e.Name == t) {
+			return "eav"
+		}
+	}
+
+	return ""
 }
 
 func (c Config) ProcessColumn(t string, col string) (bool, string) {

diff --git a/src/config_test.go b/src/config_test.go
@@ -24,12 +24,14 @@ func TestNewConfig(t *testing.T) {
 func TestProcessTable(t *testing.T) {
 	c, _ := NewConfig("magento2")
 
-	if !c.ProcessTable("admin_user") {
-		t.Error("Got false want true")
+	r1 := c.ProcessTable("admin_user")
+	if r1 != "table" {
+		t.Errorf("Got %s wanted table", r1)
 	}
 
-	if c.ProcessTable("catalog_product") {
-		t.Error("Got true want false")
+	r2 := c.ProcessTable("catalog_product")
+	if r2 != "" {
+		t.Errorf("Got %s wanted empty string", r2)
 	}
 }
 

diff --git a/src/processor.go b/src/processor.go
@@ -28,7 +28,9 @@ func (p LineProcessor) ProcessLine(s string) string {
 
 		table := stmt.Table.Name.String()
 
-		if !p.Config.ProcessTable(table) {
+		processor := p.Config.ProcessTable(table)
+
+		if processor == "" {
 			return s
 		}