Remove uncessary branching

mpchadwick · Nov 4, 2020 · 4d3b07a · 4d3b07a
1 parent 1e234aa
commit 4d3b07a
Showing 1 changed file with 23 additions and 40 deletions.
diff --git a/src/eav.go b/src/eav.go
@@ -32,47 +32,30 @@ func (eav Eav) processInsert (s string) {
 	stmt, _ := sqlparser.Parse(s)
 	insert := stmt.(*sqlparser.Insert)
 
-	table := insert.Table.Name.String()
-	if table == "eav_entity_type" {
-		var entityTypeId string
-		rows := insert.Rows.(sqlparser.Values)
-		for _, vt := range rows {
-			for i, e := range vt {
-				column := currentTable[i]
-				switch v := e.(type) {
-				case *sqlparser.SQLVal:
-					if column == "entity_type_id" {
-						entityTypeId = string(v.Val)
-					}
-					if column == "entity_type_code" {
-						eav.entityMap[string(v.Val)] = entityTypeId
-					}
+	var entityTypeId string
+	var attributeId string
+
+	rows := insert.Rows.(sqlparser.Values)
+	for _, vt := range rows {
+		for i, e := range vt {
+			column := currentTable[i]
+			switch v := e.(type) {
+			case *sqlparser.SQLVal:
+				if column == "attribute_id" {
+					attributeId = string(v.Val)
 				}
-			}
-		}
-	}
-	if table == "eav_attribute" {
-		var attributeId string
-		var entityTypeId string
-		rows := insert.Rows.(sqlparser.Values)
-		for _, vt := range rows {
-			for i, e := range vt {
-				column := currentTable[i]
-				switch v := e.(type) {
-				case *sqlparser.SQLVal:
-					if column == "attribute_id" {
-						attributeId = string(v.Val)
-					}
-					if column == "entity_type_id" {
-						entityTypeId = string(v.Val)
-					}
-					if column == "attribute_code" {
-						for _, eavConfig := range eav.Config.Eav {
-							if eav.entityMap[eavConfig.Name] == entityTypeId {
-								for eavK, eavV := range eavConfig.Attributes {
-									if eavK == string(v.Val) {
-										eavConfig.Attributes[attributeId] = eavV
-									}
+				if column == "entity_type_id" {
+					entityTypeId = string(v.Val)
+				}
+				if column == "entity_type_code" {
+					eav.entityMap[string(v.Val)] = entityTypeId
+				}
+				if column == "attribute_code" {
+					for _, eavConfig := range eav.Config.Eav {
+						if eav.entityMap[eavConfig.Name] == entityTypeId {
+							for eavK, eavV := range eavConfig.Attributes {
+								if eavK == string(v.Val) {
+									eavConfig.Attributes[attributeId] = eavV
 								}
 							}
 						}