-
Notifications
You must be signed in to change notification settings - Fork 187
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
bug: support lag for non-UTC datetime cursor fields #2170
bug: support lag for non-UTC datetime cursor fields #2170
Conversation
✅ Deploy Preview for dlt-hub-docs canceled.
|
5235731
to
ba4151e
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
thanks for catching edge cases for the lag! the problem you've found is deeper than just empty results. please see my comments.
dlt/extract/incremental/__init__.py
Outdated
# write back state | ||
self._cached_state["last_value"] = transformer.last_value | ||
# Skip writing back state if no valid rows for last_value in transformer may be wrong | ||
if rows is not None: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this is really a good catch but the problem is deeper than this. we cannot let the state go back and indeed when there's a lag this can unfortunately happen (like in your test, where the "last value" record is missing in the second read).
so IMO this code will fix it
if self.lag and (cached_last_value := self._cached_state.get("last_value")):
transformer.last_value = self.last_value_func(transformer.last_value, cached_last_value)
self._cached_state["last_value"] = transformer.last_value
it uses last_value_func to make sure that "highest" last_value is preserved ("highest" per order established by the last_value_func)
rows
is of TDataItems type: so it may be a single item or a list of items. items may be dicts, panda frames, arrow tables or batches.
so for example rows may be a pandas frame or a list of pandas frames. that's way the checks below are not sufficient. but as I said the problem is deeper (but solution is at the end simpler)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks, @rudolfix! I think this modification aligns well with the dlt code style. As you mentioned, there might be deeper issues that I’m not aware of at the moment.
so for example rows may be a pandas frame or a list of pandas frames. that's way the checks below are not sufficient. but as I said the problem is deeper (but solution is at the end simpler)
I have a question. If rows is an empty list of pandas DataFrames, I think the code rows != []
will handle this condition. This is because of the following code:
if isinstance(rows, list):
rows = [
item
for item in (self._transform_item(transformer, row) for row in rows)
if item is not None
]
else:
rows = self._transform_item(transformer, rows)
00e2cd4
to
ead27f8
Compare
- Add `ensure_pendulum_datetime_non_utc` to parse datetime strings into non-UTC datetime objects. - Add `_datetime_obj_to_str` to preserve the colon in the timezone when converting datetime objects back to strings. - Skip writing back state if no valid rows are found for `last_value` in the transformer, which may otherwise cause incorrect behavior.
ead27f8
to
e18fb40
Compare
Hi, @rudolfix. I have resovled all the changes you ask! Please review again. Thank you! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
This PR:
ensure_pendulum_datetime_non_utc
to parse datetime strings into non-UTC datetime objects._datetime_obj_to_str
to preserve the colon in the timezone when converting datetime objects back to strings.last_value
in the transformer, which may otherwise cause incorrect behavior.Fix issure: #2169
Please let me know if there are any problems with my code. Thank you!