Merge pull request #3 from wikylyu/main

feat: allow not to set logger
firefart · Jan 26, 2023 · c468788 · c468788
2 parents be1ab07 + e661729
commit c468788
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/logger.go b/logger.go
@@ -10,3 +10,15 @@ type Logger interface {
 	Warn(args ...interface{})
 	Warnf(format string, args ...interface{})
 }
+
+type NilLogger struct {
+}
+
+func (l *NilLogger) Debug(args ...interface{})                 {}
+func (l *NilLogger) Debugf(format string, args ...interface{}) {}
+func (l *NilLogger) Infof(format string, args ...interface{})  {}
+func (l *NilLogger) Info(args ...interface{})                  {}
+func (l *NilLogger) Error(args ...interface{})                 {}
+func (l *NilLogger) Errorf(format string, args ...interface{}) {}
+func (l *NilLogger) Warn(args ...interface{})                  {}
+func (l *NilLogger) Warnf(format string, args ...interface{})  {}
diff --git a/proxy.go b/proxy.go
@@ -2,7 +2,6 @@ package socks
 
 import (
 	"context"
-	"fmt"
 	"io"
 	"net"
 	"time"
@@ -30,7 +29,7 @@ type Proxy struct {
 // Start is the main function to start a proxy
 func (p *Proxy) Start() error {
 	if p.Log == nil {
-		return fmt.Errorf("please supply a logger")
+		p.Log = &NilLogger{} // allow not to set logger
 	}
 
 	listener, err := net.Listen("tcp", p.ServerAddr)