closes gin-gonic#514, code from bobbo@b4f0b50

javierprovecho · javierprovecho · commit 007bd5124afe · 2016-04-14T23:47:49.000+02:00
diff --git a/context.go b/context.go
@@ -426,6 +426,11 @@ func (c *Context) XML(code int, obj interface{}) {
 	c.Render(code, render.XML{Data: obj})
 }
 
+// YAML serializes the given struct as YAML into the response body.
+func (c *Context) YAML(code int, obj interface{}) {
+	c.Render(code, render.YAML{Data: obj})
+}
+
 // String writes the given string into the response body.
 func (c *Context) String(code int, format string, values ...interface{}) {
 	c.Status(code)
diff --git a/context_test.go b/context_test.go
@@ -433,6 +433,17 @@ func TestContextRenderFile(t *testing.T) {
 	assert.Equal(t, w.HeaderMap.Get("Content-Type"), "text/plain; charset=utf-8")
 }
 
+// TestContextRenderYAML tests that the response is serialized as YAML
+// and Content-Type is set to application/x-yaml
+func TestContextRenderYAML(t *testing.T) {
+	c, w, _ := CreateTestContext()
+	c.YAML(201, H{"foo": "bar"})
+
+	assert.Equal(t, w.Code, 201)
+	assert.Equal(t, w.Body.String(), "foo: bar\n")
+	assert.Equal(t, w.HeaderMap.Get("Content-Type"), "application/x-yaml; charset=utf-8")
+}
+
 func TestContextHeaders(t *testing.T) {
 	c, _, _ := CreateTestContext()
 	c.Header("Content-Type", "text/plain")
diff --git a/render/render.go b/render/render.go
@@ -20,6 +20,7 @@ var (
 	_ Render     = HTML{}
 	_ HTMLRender = HTMLDebug{}
 	_ HTMLRender = HTMLProduction{}
+	_ Render     = YAML{}
 )
 
 func writeContentType(w http.ResponseWriter, value []string) {
diff --git a/render/yaml.go b/render/yaml.go
@@ -0,0 +1,29 @@
+// Copyright 2014 Manu Martinez-Almeida.  All rights reserved.
+// Use of this source code is governed by a MIT style
+// license that can be found in the LICENSE file.
+
+package render
+
+import (
+	"net/http"
+
+	"gopkg.in/yaml.v2"
+)
+
+type YAML struct {
+	Data interface{}
+}
+
+var yamlContentType = []string{"application/x-yaml; charset=utf-8"}
+
+func (r YAML) Render(w http.ResponseWriter) error {
+	writeContentType(w, yamlContentType)
+
+	bytes, err := yaml.Marshal(r.Data)
+	if err != nil {
+		return err
+	}
+
+	w.Write(bytes)
+	return nil
+}

Original file line number	Diff line number	Diff line change
`@@ -20,6 +20,7 @@ var (`
`20`	`20`	`_ Render = HTML{}`
`21`	`21`	`_ HTMLRender = HTMLDebug{}`
`22`	`22`	`_ HTMLRender = HTMLProduction{}`
	`23`	`+ _ Render = YAML{}`
`23`	`24`	`)`
`24`	`25`
`25`	`26`	`func writeContentType(w http.ResponseWriter, value []string) {`