Convers query results to CSV with format flag

api.go

@@ -68,5 +68,17 @@ func API_HandleQuery(query string, c *gin.Context) {
 		return
 	}
 
+	q := c.Request.URL.Query()
+
+	format := q["format"][0]
+	if format == "" {
+		format = "json"
+	}
+
+	if format == "csv" {
+		c.String(200, result.CSV())
+		return
+	}
+
 	c.JSON(200, result)
 }

client.go

@@ -1,6 +1,9 @@
 package main
 
 import (
+	"bytes"
+	"encoding/csv"
+	"fmt"
 	"github.com/jmoiron/sqlx"
 	"reflect"
 )
@@ -107,3 +110,30 @@ func (res *Result) Format() []map[string]interface{} {
 
 	return items
 }
+
+func (res *Result) CSV() string {
+	buff := &bytes.Buffer{}
+	writer := csv.NewWriter(buff)
+
+	for _, row := range res.Rows {
+		record := make([]string, len(res.Columns))
+
+		for i, item := range row {
+			if item != nil {
+				record[i] = fmt.Sprintf("%v", item)
+			} else {
+				record[i] = ""
+			}
+		}
+
+		err := writer.Write(record)
+
+		if err != nil {
+			fmt.Println(err)
+			break
+		}
+	}
+
+	writer.Flush()
+	return buff.String()
+}