move C header prologue and epilogue to constants

Signed-off-by: Denys Smirnov <denys@sourced.tech>
bblfsh · Jun 20, 2019 · 6ee88f8 · 6ee88f8
1 parent b8b90da
commit 6ee88f8
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/gen_header.go b/gen_header.go
@@ -25,18 +25,22 @@ func main() {
 	}
 }
 
-func run() error {
-	buf := new(bytes.Buffer)
-	buf.WriteString(`
+const (
+	headerPrologue = `
 #ifdef __cplusplus
 extern "C" {
 #endif
-`)
-	defer buf.WriteString(`
+`
+	headerEpilogue = `
 #ifdef __cplusplus
 }
 #endif
-`)
+`
+)
+
+func run() error {
+	buf := new(bytes.Buffer)
+	buf.WriteString(headerPrologue)
 	for _, fname := range []string{
 		"api.go",
 		"src_index.go",
@@ -72,6 +76,7 @@ extern "C" {
 			}
 		}
 	}
+	buf.WriteString(headerEpilogue)
 	return writeHeader(buf.Bytes())
 }