From 59bdf56c51ed72e26a3a622264eafef71e2700d8 Mon Sep 17 00:00:00 2001
From: proxy666-dev <53252627+proxy666-dev@users.noreply.github.com>
Date: Sun, 31 May 2020 17:49:13 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=97=A5=E5=BF=97=E6=A8=A1?=
 =?UTF-8?q?=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 log.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/log.go b/log.go
index d4ad519..5981435 100644
--- a/log.go
+++ b/log.go
@@ -3,8 +3,31 @@ package gost
 import (
 	"fmt"
 	"log"
+	"strings"
+	"os"
 )
 
+//设置日志模式
+func SetLogMode(LogMode string) {
+	if LogMode != "" {
+	var split []string = strings.Split(LogMode, ":")
+	if split[0] == "file" {
+    logFile, err := os.OpenFile(split[1], os.O_RDWR | os.O_CREATE | os.O_APPEND, 0770)
+    if err != nil {
+        panic(err.Error())
+    } else {
+        log.SetOutput(logFile)
+    }
+    //defer logFile.Close()
+	}
+	if split[0] == "shell" {
+	if split[1] == "stdout" {
+	log.SetOutput(os.Stdout)
+	}
+	}
+	}
+}
+
 func init() {
 	log.SetFlags(log.LstdFlags | log.Lshortfile)
 }