gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

add a context to Connection.ReceiveMessage (#3926)

Browse Source 
* add context to ReceiveMessage

* add newlines

---------

Co-authored-by: Marten Seemann <martenseemann@gmail.com>
This commit is contained in:
Glonee
2023-06-28 02:29:30 +08:00
committed by GitHub
parent f3875147b9
commit 435444af7e
7 changed files with 35 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/mocks/quic/early_conn.go
View File

@@ -212,18 +212,18 @@ func (mr *MockEarlyConnectionMockRecorder) OpenUniStreamSync(arg0 interface{}) *
}
// ReceiveMessage mocks base method.
func (m *MockEarlyConnection) ReceiveMessage() ([]byte, error) {
func (m *MockEarlyConnection) ReceiveMessage(arg0 context.Context) ([]byte, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ReceiveMessage")
ret := m.ctrl.Call(m, "ReceiveMessage", arg0)
ret0, _ := ret[0].([]byte)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// ReceiveMessage indicates an expected call of ReceiveMessage.
func (mr *MockEarlyConnectionMockRecorder) ReceiveMessage() *gomock.Call {
func (mr *MockEarlyConnectionMockRecorder) ReceiveMessage(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReceiveMessage", reflect.TypeOf((*MockEarlyConnection)(nil).ReceiveMessage))
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReceiveMessage", reflect.TypeOf((*MockEarlyConnection)(nil).ReceiveMessage), arg0)
}
// RemoteAddr mocks base method.