Add TCP_KEEPALIVE option to http provider #24967
Conversation
potiuk
requested review from
kaxil,
mik-laj,
yuqian90,
josh-fell,
uranusjr,
ephraimbuddy and
jedcunningham
|
I think this one is worth breaking change (which is not really breaking in most cases) - enabling TCP Keepalive for SimpleHTTPOperator will have very limited impact in vast majority of cases (either no or very little increased traffic) however it will help out-of-the-box for a number of problems that people report with long running idle connections. The one "drawback" I can think of in making such a "breaking" change is that in some cases, the firewalls will actully NOT break connections that were broken before because they were idle. However, that is usually not something that should be seen as "bad" - the way how TCP/HTTP works, is that such firewalls breaking idle connections have rather bad effect on the client - the client does not get "disconnect" and the connection is kept open from the client, so the client does not realise tha the connection has already been broken. So even if the connetion is broken, airflow still thinks it is not. Adding Keepalive will actually make it behave much better in case the server stops responding as lack of ACK in Keepalive will actually signal to client that it should close the connection and error out. So in all cases, it seems like a better behaviour. And you can still disable it by setting "tcp_keep_alive" to False. It is a breaking change though, so we should increase the version of the provider. It's very difficult to test this one via unit tests, but I tested it in Breeze using `tcpdump' by setting the values to low ones (keep alives generated avery second after 1 second of inactivity) resulted in this: (continues every second while the long running HTTP request runs) Where setting tcp_keep_alive=False resulted in this (no regular keepalive messages - the initial exchange was the only traffic, it did not run continuously while the requests runs: |
Yeah @jedcunningham I wanted to do it but I found it difficult to test. This is very low level and it would actually require us to run HTTP server in process. The problem is that "requests" mock does a not-nice trick of replacing whatever adapters you have with "Mock Adapter" so that you can test the "logic" of requests parameters. The TCPKeepAliveAdapter is rather low-level one. I could likely do some HTTP server and unit testing with it, but I am not sure if this is really worth it? WDYT ? |
|
I was more thinking just mocking the session and making sure operator kwargs end up getting all the way there through the hook. If that doesn't work out, I don't think it's worth going further. |
I found a better way |
potiuk
force-pushed
the
add-keepalive-to-http-operator
branch
from
c2ca107
to
4ef7f22
Compare
OK. I think I got good balance between having tests and not complicating it too much. |
|
BTW. The user confirmed that lack of TCP_KEEPALIVE is what's causing SimpleHTTPRequest hangs on long requests: #21365 |
There is a lot (IMHO too much) of extra args handling in HTTP connection. We can attempt it as next step :) |
Oh absolutely. I wasn't saying it needed to be done here and now 🙂 |
potiuk
force-pushed
the
add-keepalive-to-http-operator
branch
2 times, most recently
from
385947c
to
e164679
Compare
Enabling TCP_KEEPALIVE allows to keep the idle connections opened when there are firewalls in-betweeen that close such connections. TCP_KEEPALIVE sends a "no-data" packet regularly (and expects ACK from the server). This should not be mistaken with Python requests "keep alive" feature - which reuses opened HTTP connections if you perform several requests to the same server (both are named keep alive but they both mean two completely different things and are implemented at differen layers of the OSI network stack. The Requests Keep Alive is Layer 7 (application) and TCP Keep Alive is at Layer 4 (transport). The "Requests" keep alive is enabled by default in the requests library while the TCP Keep Alive requires a TCPKeepAliveAdapter to be used (from requests_toolbelt library). Fixes: apache#21365
potiuk
force-pushed
the
add-keepalive-to-http-operator
branch
from
e164679
to
f0af3d4
Compare
| @@ -91,15 +100,26 @@ def __init__( | |||
| self.headers = headers or {} | |||
| self.extra_options = extra_options or {} | |||
| self.response_check = response_check | |||
|
|
|||
| self.hook = HttpHook(method=method, http_conn_id=http_conn_id) | |||
There was a problem hiding this comment.
@potiuk Is there a specific reason we would like to remove this assignment to the instance and are only creating the hook at a later stage in the poke method excluding it as an instance variable?
The async operator relies on this instance variable: https://github.com/astronomer/astronomer-providers/blob/main/astronomer/providers/http/sensors/http.py#L67 and the RC is failing for the current async implementation.
Should we create a new hook instance in the derived class?
There was a problem hiding this comment.
This is the error we get in our DAG run with the RC:
Traceback (most recent call last):
File "/usr/local/lib/python3.9/site-packages/astronomer/providers/http/sensors/http.py", line 67, in execute
method=self.hook.method, # TODO: Fix this to directly get method from ctor
AttributeError: 'HttpSensorAsync' object has no attribute 'hook'
There was a problem hiding this comment.
I have created a PR to reinstantiate the hook in the derived class: astronomer/astronomer-providers#515
There was a problem hiding this comment.
Well. I have not thought too much about this use case when I added it - it was just bringing a general approach that we usually do where Hooks are created in execute() method rather than in constructor.
But if there are good reasons why it should not be moved I can remove that one from the batch and make RC2 and restore the self.hook.
I think with async operators approach we miss a bit of best practices:
- should we create Hook in init()?
- should we have a Hook as Operator attribute (self.hook)?
- or should we restore the hook every time when we come back from deferred state?
I think the last option is most "reasonable" because first two might make (an incorrect) assumption that the same hook object is available during the whole lifecycle of all operator's methods, and if Hook is not "stateless", it might break when operator is deferred (because the Hook will be recreated when operator resumes from deferred state). If you keep the hook as self. property, you might (unconciously) rely on the fact that the hook is the same object in various methods of deferred operator.
WDYT? Does it sound like a good practice ? Also others @dstandish maybe? I think we had some discussion about this in the past (can't recall exactly)
There was a problem hiding this comment.
Makes sense to me. In that case, I think we can create a clean up story on our end to adhere to the best practices post our conclusion on this discussion. We do not need an RC2 for that then, and also for the sake that the operator still continues to work, I will for the time being, put the hack to create the hook in our operator init() with our PR.
Also, looping in @kaxil @bharanidharan14 @rajaths010494 @pankajastro for the discussion.
There was a problem hiding this comment.
That sounds good to me. Would be great to contribute such "best practicess for writing Defferable Operators" to the community after all the experiences Astronomer had with writing theirs. It would be a great contribution :).
Later on we could event try to automate some of those "best practices" in pre-commits for any future community async operators. This sounds like a doable task to ast-parse an operatort and if it is a Defferrable one, make sure no self.hook is used.
ephraimbuddy
added
the
changelog:skip
Enabling TCP_KEEPALIVE allows to keep the idle connections opened
when there are firewalls in-betweeen that close such connections.
TCP_KEEPALIVE sends a "no-data" packet regularly (and expects
ACK from the server). This should not be mistaken with Python
requests "keep alive" feature - which reuses opened HTTP connections
if you perform several requests to the same server (both are named
keep alive but they both mean two completely different things and
are implemented at differen layers of the OSI network stack. The
Requests Keep Alive is Layer 7 (application) and TCP Keep Alive
is at Layer 4 (transport).
The "Requests" keep alive is enabled by default in the requests
library while the TCP Keep Alive requires a TCPKeepAliveAdapter
to be used (from requests_toolbelt library).
Fixes: #21365
^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named
{pr_number}.significant.rstor{issue_number}.significant.rst, in newsfragments.