writer.go

   1 /*
   2 netstring -- netstring format serialization library
   3 Copyright (C) 2015 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 package netstring
  20
  21 import (
  22         "bufio"
  23         "io"
  24         "strconv"
  25 )
  26
  27 type Writer struct {
  28         writer *bufio.Writer
  29         prefix []byte
  30         err    error
  31 }
  32
  33 func NewWriter(w io.Writer) *Writer {
  34         return &Writer{
  35                 writer: bufio.NewWriter(w),
  36                 prefix: make([]byte, 0, MaxPrefixSize),
  37         }
  38 }
  39
  40 // Write serialized representation of data to the underlying writer.
  41 // It returns number of serialized data bytes written.
  42 func (self *Writer) Write(data []byte) (bytesWritten int, err error) {
  43         self.prefix = strconv.AppendUint(self.prefix[:0], uint64(len(data)), 10)
  44         self.prefix = append(self.prefix, ':')
  45         if _, self.err = self.writer.Write(self.prefix); self.err != nil {
  46                 return 0, self.err
  47         }
  48         if _, self.err = self.writer.Write(data); self.err != nil {
  49                 return 0, self.err
  50         }
  51         if _, self.err = self.writer.Write([]byte{','}); self.err != nil {
  52                 return 0, self.err
  53         }
  54         if self.err = self.writer.Flush(); self.err != nil {
  55                 return 0, self.err
  56         }
  57         return len(self.prefix) + len(data) + 1, nil
  58 }