| // Copyright 2015 The Vanadium 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 lib |
| |
| // TypeWriter implements io.Writer but allows changing the underlying buffer. |
| // This is useful for merging discrete messages that are part of the same flow. |
| type TypeWriter struct { |
| writer ClientWriter |
| } |
| |
| func NewTypeWriter(w ClientWriter) *TypeWriter { |
| return &TypeWriter{writer: w} |
| } |
| |
| func (w *TypeWriter) Write(p []byte) (int, error) { |
| err := w.writer.Send(ResponseTypeMessage, p) |
| if err != nil { |
| return 0, err |
| } |
| return len(p), nil |
| } |