From 270848509bcfa0ecc72006d6325011bcb3096026 Mon Sep 17 00:00:00 2001 From: Brad Fitzpatrick Date: Mon, 12 May 2014 20:26:27 -0700 Subject: [PATCH] mime/multipart: add NewReader example Fixes #7888 LGTM=adg R=adg CC=golang-codereviews https://golang.org/cl/100420043 --- src/pkg/mime/multipart/example_test.go | 53 ++++++++++++++++++++++++++ src/pkg/mime/multipart/multipart.go | 10 +++-- 2 files changed, 60 insertions(+), 3 deletions(-) create mode 100644 src/pkg/mime/multipart/example_test.go diff --git a/src/pkg/mime/multipart/example_test.go b/src/pkg/mime/multipart/example_test.go new file mode 100644 index 0000000000..26135b7858 --- /dev/null +++ b/src/pkg/mime/multipart/example_test.go @@ -0,0 +1,53 @@ +// Copyright 2014 The Go Authors. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +package multipart_test + +import ( + "fmt" + "io" + "io/ioutil" + "log" + "mime" + "mime/multipart" + "net/mail" + "strings" +) + +func ExampleNewReader() { + msg := &mail.Message{ + Header: map[string][]string{ + "Content-Type": []string{"multipart/mixed; boundary=foo"}, + }, + Body: strings.NewReader( + "--foo\r\nFoo: one\r\n\r\nA section\r\n" + + "--foo\r\nFoo: two\r\n\r\nAnd another\r\n" + + "--foo--\r\n"), + } + mediaType, params, err := mime.ParseMediaType(msg.Header.Get("Content-Type")) + if err != nil { + log.Fatal(err) + } + if strings.HasPrefix(mediaType, "multipart/") { + mr := multipart.NewReader(msg.Body, params["boundary"]) + for { + p, err := mr.NextPart() + if err == io.EOF { + return + } + if err != nil { + log.Fatal(err) + } + slurp, err := ioutil.ReadAll(p) + if err != nil { + log.Fatal(err) + } + fmt.Printf("Part %q: %q\n", p.Header.Get("Foo"), slurp) + } + } + + // Output: + // Part "one": "A section" + // Part "two": "And another" +} diff --git a/src/pkg/mime/multipart/multipart.go b/src/pkg/mime/multipart/multipart.go index 2b4f5b433e..7382efab96 100644 --- a/src/pkg/mime/multipart/multipart.go +++ b/src/pkg/mime/multipart/multipart.go @@ -81,12 +81,16 @@ func (p *Part) parseContentDisposition() { } } -// NewReader creates a new multipart Reader reading from reader using the +// NewReader creates a new multipart Reader reading from r using the // given MIME boundary. -func NewReader(reader io.Reader, boundary string) *Reader { +// +// The boundary is usually obtained from the "boundary" parameter of +// the message's "Content-Type" header. Use mime.ParseMediaType to +// parse such headers. +func NewReader(r io.Reader, boundary string) *Reader { b := []byte("\r\n--" + boundary + "--") return &Reader{ - bufReader: bufio.NewReader(reader), + bufReader: bufio.NewReader(r), nl: b[:2], nlDashBoundary: b[:len(b)-2], -- 2.50.0