Просмотр исходного кода

Refactor logrwc to wrappers

tags/0.9
9seconds 8 лет назад
Родитель
Сommit
26020fcd0c
2 измененных файлов: 4 добавлений и 4 удалений
  1. 2
    2
      proxy/server.go
  2. 2
    2
      wrappers/logrwc.go

+ 2
- 2
proxy/server.go Просмотреть файл

124
 		return nil, 0, errors.Annotate(err, "Cannot create client stream")
124
 		return nil, 0, errors.Annotate(err, "Cannot create client stream")
125
 	}
125
 	}
126
 
126
 
127
-	wConn = newLogReadWriteCloser(wConn, s.logger, socketID, "client")
127
+	wConn = wrappers.NewLogRWC(wConn, s.logger, socketID, "client")
128
 	wConn = wrappers.NewStreamCipherRWC(wConn, obfs2.Encryptor, obfs2.Decryptor)
128
 	wConn = wrappers.NewStreamCipherRWC(wConn, obfs2.Encryptor, obfs2.Decryptor)
129
 	wConn = wrappers.NewCtxRWC(ctx, cancel, wConn)
129
 	wConn = wrappers.NewCtxRWC(ctx, cancel, wConn)
130
 
130
 
144
 		return nil, errors.Annotate(err, "Cannot write hadnshake frame")
144
 		return nil, errors.Annotate(err, "Cannot write hadnshake frame")
145
 	}
145
 	}
146
 
146
 
147
-	wConn = newLogReadWriteCloser(wConn, s.logger, socketID, "telegram")
147
+	wConn = wrappers.NewLogRWC(wConn, s.logger, socketID, "telegram")
148
 	wConn = wrappers.NewStreamCipherRWC(wConn, obfs2.Encryptor, obfs2.Decryptor)
148
 	wConn = wrappers.NewStreamCipherRWC(wConn, obfs2.Encryptor, obfs2.Decryptor)
149
 	wConn = wrappers.NewCtxRWC(ctx, cancel, wConn)
149
 	wConn = wrappers.NewCtxRWC(ctx, cancel, wConn)
150
 
150
 

proxy/logrwc.go → wrappers/logrwc.go Просмотреть файл

1
-package proxy
1
+package wrappers
2
 
2
 
3
 import (
3
 import (
4
 	"io"
4
 	"io"
36
 	return err
36
 	return err
37
 }
37
 }
38
 
38
 
39
-func newLogReadWriteCloser(conn io.ReadWriteCloser, logger *zap.SugaredLogger, sockid string, name string) io.ReadWriteCloser {
39
+func NewLogRWC(conn io.ReadWriteCloser, logger *zap.SugaredLogger, sockid string, name string) io.ReadWriteCloser {
40
 	return &LogReadWriteCloser{
40
 	return &LogReadWriteCloser{
41
 		conn:   conn,
41
 		conn:   conn,
42
 		logger: logger,
42
 		logger: logger,

Загрузка…
Отмена
Сохранить