Merge pull request #45 from knz/20190411-align

Preserve full alignment in data responses
maddyblue · Apr 11, 2019 · c900015 · c900015
2 parents 1ef4b4b + 703f5a7
commit c900015
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -144,7 +144,7 @@ func fmtsql(cfg tree.PrettyCfg, stmts []string) (string, error) {
 					break
 				}
 				// Remove trailing whitespace but keep up to 2 newlines.
-				prettied.WriteString(strings.TrimSpace(found))
+				prettied.WriteString(strings.TrimRightFunc(found, unicode.IsSpace))
 				newlines := strings.Count(found, "\n")
 				if newlines > 2 {
 					newlines = 2
@@ -177,7 +177,7 @@ func fmtsql(cfg tree.PrettyCfg, stmts []string) (string, error) {
 		}
 	}
 
-	return strings.TrimSpace(prettied.String()), nil
+	return strings.TrimRightFunc(prettied.String(), unicode.IsSpace), nil
 }
 
 func serveHTTP(spec Specification) {