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

Refactor ctxrw to wrappers

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

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

126
 
126
 
127
 	wConn = newLogReadWriteCloser(wConn, s.logger, socketID, "client")
127
 	wConn = newLogReadWriteCloser(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 = newCtxReadWriteCloser(ctx, cancel, wConn)
129
+	wConn = wrappers.NewCtxRWC(ctx, cancel, wConn)
130
 
130
 
131
 	return wConn, dc, nil
131
 	return wConn, dc, nil
132
 }
132
 }
146
 
146
 
147
 	wConn = newLogReadWriteCloser(wConn, s.logger, socketID, "telegram")
147
 	wConn = newLogReadWriteCloser(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 = newCtxReadWriteCloser(ctx, cancel, wConn)
149
+	wConn = wrappers.NewCtxRWC(ctx, cancel, wConn)
150
 
150
 
151
 	return wConn, nil
151
 	return wConn, nil
152
 }
152
 }

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

1
-package proxy
1
+package wrappers
2
 
2
 
3
 import (
3
 import (
4
 	"context"
4
 	"context"
48
 	return c.conn.Close()
48
 	return c.conn.Close()
49
 }
49
 }
50
 
50
 
51
-func newCtxReadWriteCloser(ctx context.Context, cancel context.CancelFunc, conn io.ReadWriteCloser) io.ReadWriteCloser {
51
+func NewCtxRWC(ctx context.Context, cancel context.CancelFunc, conn io.ReadWriteCloser) io.ReadWriteCloser {
52
 	return &CtxReadWriteCloser{
52
 	return &CtxReadWriteCloser{
53
 		conn:   conn,
53
 		conn:   conn,
54
 		ctx:    ctx,
54
 		ctx:    ctx,

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