From ab746d068e6b193ed3db170b79d489f5ab782069 Mon Sep 17 00:00:00 2001
From: Julius Volz <julius.volz@gmail.com>
Date: Fri, 11 Jan 2013 03:17:58 +0100
Subject: [PATCH] Set correct Content-Type header based on output format.

---
 api/query.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/api/query.go b/api/query.go
index 6d5ff33d59..4bfa3e044c 100644
--- a/api/query.go
+++ b/api/query.go
@@ -1,6 +1,7 @@
 package api
 
 import (
+        "code.google.com/p/gorest"
 	"github.com/matttproud/prometheus/rules"
 	"github.com/matttproud/prometheus/rules/ast"
         "time"
@@ -13,9 +14,15 @@ func (serv MetricsService) Query(Expr string, Json string, Start string, End str
 
         timestamp := time.Now()
 
-        format := ast.TEXT
+        rb := serv.ResponseBuilder()
+        var format ast.OutputFormat
         if Json != "" {
                 format = ast.JSON
+                rb.SetContentType(gorest.Application_Json)
+        } else {
+                format = ast.TEXT
+                rb.SetContentType(gorest.Text_Plain)
         }
+
         return ast.EvalToString(exprNode, &timestamp, format)
 }