github.com/gagliardetto/golang-go@v0.0.0-20201020153340-53909ea70814/cmd/internal/bio/must.go (about)

     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  }