Add featrure Union

Author: bonkzero404

jql.go

@@ -10,19 +10,29 @@ import (
 )
 
 type Json2Sql struct {
-	sqlJson *SQLJson
+	sqlJson            *SQLJson
+	sqlJsonSelectUnion *[]SQLJson
 }
 
-func NewJson2Sql(jsonData json.RawMessage) (*Json2Sql, error) {
+func NewJson2Sql(jsonData json.RawMessage, isUnion ...bool) (*Json2Sql, error) {
+	if isUnion != nil && isUnion[0] {
+		var sqlJsonUnion []SQLJson
+
+		err := json.Unmarshal([]byte(jsonData), &sqlJsonUnion)
+		if err != nil {
+			return nil, fmt.Errorf("error: %s", err)
+		}
+		return &Json2Sql{sqlJsonSelectUnion: &sqlJsonUnion}, nil
+	}
+
 	var sqlJson *SQLJson
 
 	err := json.Unmarshal(jsonData, &sqlJson)
 
 	if err != nil {
 		return nil, fmt.Errorf("error: %s", err)
 	}
-
-	return &Json2Sql{sqlJson}, nil
+	return &Json2Sql{sqlJson: sqlJson}, nil
 }
 
 func cleanSpaces(input string) string {
@@ -405,3 +415,34 @@ func (jql *Json2Sql) Generate() (string, []interface{}, error) {
 
 	return newQuery, values, nil
 }
+
+func (jql *Json2Sql) buildRawUnion() string {
+	var sql string
+	var sqlUnion []string
+
+	if jql.sqlJsonSelectUnion != nil {
+		for _, v := range *jql.sqlJsonSelectUnion {
+			jql.sqlJson = &v
+			strBuild := jql.rawBuild()
+			sqlUnion = append(sqlUnion, strBuild)
+		}
+	}
+
+	jql.sqlJson = nil
+	sql = strings.Join(sqlUnion, " UNION ")
+
+	return sql
+}
+
+func (jql *Json2Sql) BuildUnion() string {
+	sqlCleanValue := jql.rawValueExtractor(jql.buildRawUnion())
+
+	return cleanSpaces(sqlCleanValue)
+}
+
+func (jql *Json2Sql) GenerateUnion() (string, []interface{}, error) {
+	sql := jql.buildRawUnion()
+	newQuery, values := jql.MaskedQueryValue(sql)
+
+	return newQuery, values, nil
+}

jql_test.go

@@ -19,6 +19,11 @@ func TestConstructor_Fail(t *testing.T) {
 	assert.NotNil(t, err)
 }
 
+func TestConstructor_Fail_Union(t *testing.T) {
+	var _, err = NewJson2Sql([]byte(`[{"table":"test"`), true)
+	assert.NotNil(t, err)
+}
+
 func TestRawJson_OK(t *testing.T) {
 	var expected = "hello"
 
@@ -1019,3 +1024,96 @@ func TestGenerateJsonToSql(t *testing.T) {
 	assert.Equal(t, strExpectation, sql)
 	assert.Equal(t, []interface{}{float64(1), "foo", true, float64(100), float64(1), "2020-01-01", "2023-01-01", "2", float64(10)}, filter)
 }
+
+func TestBuildRawUnion(t *testing.T) {
+	jsonData := `
+		[
+			{
+				"table": "table_1",
+				"selectFields": [
+					"a",
+					"b"
+				],
+				"conditions": [
+					{
+						"datatype": "number",
+						"clause": "a",
+						"operator": "=",
+						"value": 1
+					}
+				],
+				"limit": 1
+			},
+			{
+				"table": "table_2",
+				"selectFields": [
+					"a",
+					"b"
+				],
+				"conditions": [
+					{
+						"datatype": "number",
+						"clause": "a",
+						"operator": "=",
+						"value": 1
+					}
+				],
+				"limit": 1
+			}
+		]
+	`
+
+	jql, _ := NewJson2Sql([]byte(jsonData), true)
+	sql := jql.BuildUnion()
+
+	strExpectation := "SELECT a, b FROM table_1 WHERE a = 1 LIMIT 1 UNION SELECT a, b FROM table_2 WHERE a = 1 LIMIT 1"
+
+	assert.Equal(t, strExpectation, sql)
+}
+
+func TestGenerateUnion(t *testing.T) {
+	jsonData := `
+		[
+			{
+				"table": "table_1",
+				"selectFields": [
+					"a",
+					"b"
+				],
+				"conditions": [
+					{
+						"datatype": "number",
+						"clause": "a",
+						"operator": "=",
+						"value": 1
+					}
+				],
+				"limit": 1
+			},
+			{
+				"table": "table_2",
+				"selectFields": [
+					"a",
+					"b"
+				],
+				"conditions": [
+					{
+						"datatype": "number",
+						"clause": "a",
+						"operator": "=",
+						"value": 1
+					}
+				],
+				"limit": 1
+			}
+		]
+	`
+
+	jql, _ := NewJson2Sql([]byte(jsonData), true)
+	sql, filter, _ := jql.GenerateUnion()
+
+	strExpectation := "SELECT a, b FROM table_1 WHERE a = ? LIMIT 1 UNION SELECT a, b FROM table_2 WHERE a = ? LIMIT 1"
+
+	assert.Equal(t, strExpectation, sql)
+	assert.Equal(t, []interface{}{float64(1), float64(1)}, filter)
+}