Skip to content

K
koko
Commit a33b8909 authored by Eric's avatar Eric
Browse files
Options

[Update] fix lock bug

parent a1b2ff4b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
pkg/handler/wrappersession.go
View file @ a33b8909
...@@ -66,6 +66,9 @@ func (w *WrapperSession) Close() error { ...@@ -66,6 +66,9 @@ func (w *WrapperSession) Close() error {
return nil return nil
default: default:
} }
w.mux.Lock()
defer w.mux.Unlock()
_ = w.outReader.Close()
err := w.inWriter.Close() err := w.inWriter.Close()
w.initReadPip() w.initReadPip()
return err return err
...@@ -76,8 +79,6 @@ func (w *WrapperSession) Write(p []byte) (int, error) { ...@@ -76,8 +79,6 @@ func (w *WrapperSession) Write(p []byte) (int, error) {
} }
func (w *WrapperSession) initReadPip() { func (w *WrapperSession) initReadPip() {
w.mux.Lock()
defer w.mux.Unlock()
w.outReader, w.inWriter = io.Pipe() w.outReader, w.inWriter = io.Pipe()
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment