logger.go 1.29 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
// Copyright 2009 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 iotest

import (
	"io";
	"log";
	"os";
)

type writeLogger struct {
14 15
	prefix	string;
	w	io.Writer;
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
}

func (l *writeLogger) Write(p []byte) (n int, err os.Error) {
	n, err = l.w.Write(p);
	if err != nil {
		log.Stdoutf("%s %x: %v", l.prefix, p[0:n], err);
	} else {
		log.Stdoutf("%s %x", l.prefix, p[0:n]);
	}
	return;
}

// NewWriteLogger returns a writer that behaves like w except
// that it logs (using log.Stdout) each write to standard output,
// printing the prefix and the hexadecimal data written.
func NewWriteLogger(prefix string, w io.Writer) io.Writer {
32
	return &writeLogger{prefix, w};
33 34 35
}

type readLogger struct {
36 37
	prefix	string;
	r	io.Reader;
38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
}

func (l *readLogger) Read(p []byte) (n int, err os.Error) {
	n, err = l.r.Read(p);
	if err != nil {
		log.Stdoutf("%s %x: %v", l.prefix, p[0:n], err);
	} else {
		log.Stdoutf("%s %x", l.prefix, p[0:n]);
	}
	return;
}

// NewReadLogger returns a writer that behaves like w except
// that it logs (using log.Stdout) each write to standard output,
// printing the prefix and the hexadecimal data written.
func NewReadLogger(prefix string, r io.Reader) io.Reader {
54
	return &readLogger{prefix, r};
55
}