add format code for sqlbuilder

Author: free1139

sqlbuilder.go

@@ -44,16 +44,23 @@ func (b *SqlBuilder) SetIndent(indent string) *SqlBuilder {
 }
 
 func (b *SqlBuilder) Add(key string, args ...interface{}) *SqlBuilder {
-	if len(key) > 0 {
-		b.fromBuff.WriteString(b.indent)
-		b.fromBuff.WriteString(key)
+	if len(key) == 0 {
+		return b
 	}
+
+	b.fromBuff.WriteString(b.indent)
+	b.fromBuff.WriteString(key)
 	if len(args) > 0 {
 		b.args = append(b.args, args...)
 	}
 	return b
 }
 
+// recursive Add,  only format the code when coding
+func (b *SqlBuilder) AddTab(key string, args ...interface{}) *SqlBuilder {
+	return b.Add(key, args...)
+}
+
 func (b *SqlBuilder) AddIf(ok bool, key string, args ...interface{}) *SqlBuilder {
 	if !ok {
 		return b

sqlbuilder_test.go

@@ -9,20 +9,21 @@ func TestSqlBuilder(t *testing.T) {
 	bd := NewSqlBuilder(DRV_NAME_POSTGRES)
 	bd.Select("count(*)")
 	bd.Add("FROM")
-	bd.Add("tmp tb1")
-	bd.Add("INNER JOIN tmp1 tb2 ON tb2.id=tb2.tmp_id")
+	bd.AddTab("tmp tb1")
+	bd.AddTab("INNER JOIN tmp1 tb2 ON tb2.id=tb2.tmp_id")
 	bd.Add("WHERE")
-	bd.Add("1=1")
+	bd.AddTab("1=1")
 	bd.AddIf(true, "AND (1=?)", 0)
 	bd.AddIf(true, "OR (tb1 IN ("+bd.In([]interface{}{1, 2})+"))")
 	bd.Add("GROUP BY tb1.id")
 	bd.Add("HAVING count(*)>?", 1)
 	fmt.Println(bd)
 
-	bd1 := bd.Copy().Select("tb1.id", "count(*)")
+	bd1 := bd.Copy()
 	bd1.Add("ORDER BY tb1.id DESC")
 	bd1.Add("OFFSET ?", 1)
 	bd1.Add("LIMIT ?", 1)
+	bd1.Select("tb1.id", "count(*)")
 	fmt.Println(bd1)
 
 	bd2 := NewSqlBuilder(DRV_NAME_POSTGRES)