From 5061bfb5e8d061e6304624f64bc44c1b089a74e6 Mon Sep 17 00:00:00 2001 From: qydysky <32743305+qydysky@users.noreply.github.com> Date: Sun, 16 Oct 2022 02:45:21 +0800 Subject: [PATCH] fix --- websocket/Recoder.go | 7 +++---- websocket/Server_test.go | 5 ++--- 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/websocket/Recoder.go b/websocket/Recoder.go index ce0ad34..0faf427 100644 --- a/websocket/Recoder.go +++ b/websocket/Recoder.go @@ -20,16 +20,15 @@ var ( type Recorder struct { Server *Server - FilePath string onlyOnce funcCtrl.SkipFunc stopflag *signal.Signal } -func (t *Recorder) Start() error { +func (t *Recorder) Start(filePath string) error { if t.Server == nil { return ErrSerIsNil } - if t.FilePath == "" { + if filePath == "" { return ErrFileNoSet } if t.onlyOnce.NeedSkip() { @@ -37,7 +36,7 @@ func (t *Recorder) Start() error { } go func() { - f := file.New(t.FilePath, 0, false) + f := file.New(filePath, 0, false) defer f.Close() var startTimeStamp time.Time diff --git a/websocket/Server_test.go b/websocket/Server_test.go index 749e5ff..ccf3689 100644 --- a/websocket/Server_test.go +++ b/websocket/Server_test.go @@ -18,10 +18,9 @@ func Test_Server(t *testing.T) { ws_mq := s.Interface() recoder := &Recorder{ - Server: s, - FilePath: "l.csv", + Server: s, } - recoder.Start() + recoder.Start("l.csv") defer recoder.Stop() ws_mq.Pull_tag(map[string]func(interface{}) bool{ -- 2.39.2