From: qydysky <32743305+qydysky@users.noreply.github.com> Date: Sat, 15 Oct 2022 18:45:21 +0000 (+0800) Subject: fix X-Git-Tag: v0.10.1 X-Git-Url: http://127.0.0.1:8081/?a=commitdiff_plain;h=5061bfb5e8d061e6304624f64bc44c1b089a74e6;p=part%2F.git fix --- 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{