not support to shrink logger buffer pool size

logger.go

@@ -189,18 +189,21 @@ var (
 )
 
 // SetPoolSize set pool size
-func SetPoolSize(size int) {
-	switch {
-	case int32(size) == poolSize:
+func SetPoolSize(s int) {
+	size := int32(s)
+
+	if size == poolSize {
 		return
-	case int32(size) > poolSize:
-		poolBuffers = make([][]byte, size)
-		poolFlag = make([]int32, size)
-		poolSize = int32(cap(poolBuffers))
-	case int32(size) < poolSize:
-		poolSize = int32(size)
-		// shrink pool not change the buffer size
 	}
+
+	if size < poolSize {
+		fmt.Println("not support to shrink logger buffer pool size")
+		return
+	}
+
+	poolBuffers = make([][]byte, size)
+	poolFlag = make([]int32, size)
+	poolSize = int32(cap(poolBuffers))
 }
 
 // WriteLog write log data

logger_test.go

@@ -49,6 +49,11 @@ func TestSetWriter(t *testing.T) {
 	}
 }
 
+func TestSetPoolSize(t *testing.T) {
+	SetPoolSize(512)
+	SetPoolSize(128)
+}
+
 func TestTimeFormat(t *testing.T) {
 	now := time.Now()
 	fmt.Println(now.Format("20060102 15:04:05.999"))

loggerprof/loggerprof.go

@@ -1,11 +1,12 @@
 package main
 
 import (
-	"github.com/vogo/logger"
 	"io/ioutil"
 	"log"
 	"net/http"
 	_ "net/http/pprof"
+
+	"github.com/vogo/logger"
 )
 
 func main() {