1 // Copyright 2016 The Go Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style 3 // license that can be found in the LICENSE file. 4 5 package bio 6 7 import ( 8 "io" 9 "log" 10 ) 11 12 // MustClose closes Closer c and calls log.Fatal if it returns a non-nil error. 13 func MustClose(c io.Closer) { 14 if err := c.Close(); err != nil { 15 log.Fatal(err) 16 } 17 } 18 19 // MustWriter returns a Writer that wraps the provided Writer, 20 // except that it calls log.Fatal instead of returning a non-nil error. 21 func MustWriter(w io.Writer) io.Writer { 22 return mustWriter{w} 23 } 24 25 type mustWriter struct { 26 w io.Writer 27 } 28 29 func (w mustWriter) Write(b []byte) (int, error) { 30 n, err := w.w.Write(b) 31 if err != nil { 32 log.Fatal(err) 33 } 34 return n, nil 35 } 36 37 func (w mustWriter) WriteString(s string) (int, error) { 38 n, err := io.WriteString(w.w, s) 39 if err != nil { 40 log.Fatal(err) 41 } 42 return n, nil 43 } 44
