Refactor logrwc to wrappers

wrappers/logrwc.go

@@ -0,0 +1,46 @@
+package wrappers
+
+import (
+	"io"
+
+	"go.uber.org/zap"
+)
+
+// LogReadWriteCloser adds additional logging for reading/writing. All
+// logging is performed for debug mode only.
+type LogReadWriteCloser struct {
+	conn   io.ReadWriteCloser
+	logger *zap.SugaredLogger
+	sockid string
+	name   string
+}
+
+// Read reads from connection
+func (l *LogReadWriteCloser) Read(p []byte) (n int, err error) {
+	n, err = l.conn.Read(p)
+	l.logger.Debugw("Finish reading", "name", l.name, "socketid", l.sockid, "nbytes", n, "error", err)
+	return
+}
+
+// Write writes into connection.
+func (l *LogReadWriteCloser) Write(p []byte) (n int, err error) {
+	n, err = l.conn.Write(p)
+	l.logger.Debugw("Finish writing", "name", l.name, "socketid", l.sockid, "nbytes", n, "error", err)
+	return
+}
+
+// Close closes underlying connection.
+func (l *LogReadWriteCloser) Close() error {
+	err := l.conn.Close()
+	l.logger.Debugw("Finish closing socket", "name", l.name, "socketid", l.sockid, "error", err)
+	return err
+}
+
+func NewLogRWC(conn io.ReadWriteCloser, logger *zap.SugaredLogger, sockid string, name string) io.ReadWriteCloser {
+	return &LogReadWriteCloser{
+		conn:   conn,
+		logger: logger,
+		sockid: sockid,
+		name:   name,
+	}
+}