|
35 | 35 | connection_info/1, connection_info/2, |
36 | 36 | connection_info_all/0, connection_info_all/1, |
37 | 37 | emit_connection_info_all/4, emit_connection_info_local/3, |
38 | | - close_connection/2, force_connection_event_refresh/1, accept_ack/2, |
| 38 | + close_connection/2, force_connection_event_refresh/1, |
39 | 39 | handshake/2, tcp_host/1]). |
40 | 40 |
|
41 | 41 | %% Used by TCP-based transports, e.g. STOMP adapter |
|
88 | 88 | [rabbit_types:infos()]. |
89 | 89 | -spec close_connection(pid(), string()) -> 'ok'. |
90 | 90 | -spec force_connection_event_refresh(reference()) -> 'ok'. |
91 | | --spec accept_ack(any(), rabbit_net:socket()) -> ok. |
92 | 91 |
|
93 | 92 | -spec on_node_down(node()) -> 'ok'. |
94 | 93 | -spec tcp_listener_addresses(listener_config()) -> [address()]. |
@@ -368,16 +367,16 @@ handshake(Ref, ProxyProtocol) -> |
368 | 367 | true -> |
369 | 368 | {ok, ProxyInfo} = ranch:recv_proxy_header(Ref, 1000), |
370 | 369 | {ok, Sock} = ranch:handshake(Ref), |
371 | | - tune_buffer_size(Sock), |
372 | | - ok = file_handle_cache:obtain(), |
| 370 | + setup_socket(Sock), |
373 | 371 | {ok, {rabbit_proxy_socket, Sock, ProxyInfo}}; |
374 | 372 | false -> |
375 | | - ranch:handshake(Ref) |
| 373 | + {ok, Sock} = ranch:handshake(Ref), |
| 374 | + setup_socket(Sock), |
| 375 | + {ok, Sock} |
376 | 376 | end. |
377 | 377 |
|
378 | | -accept_ack(Ref, Sock) -> |
379 | | - ok = ranch:accept_ack(Ref), |
380 | | - tune_buffer_size(Sock), |
| 378 | +setup_socket(Sock) -> |
| 379 | + ok = tune_buffer_size(Sock), |
381 | 380 | ok = file_handle_cache:obtain(). |
382 | 381 |
|
383 | 382 | tune_buffer_size(Sock) -> |
|
0 commit comments