palantir / atlasdb

Transactional Distributed Database Layer
https://palantir.github.io/atlasdb/
Apache License 2.0
53 stars 9 forks source link

SnapshotTransaction stream refactor proof-of-concept #7003

Open wi11dey opened 7 months ago

wi11dey commented 7 months ago

Could probably also separate out all the postfiltering logic into its own PostFilter class

Birthday present for @rhuffy

General

Before this PR:

After this PR:

==COMMIT_MSG== ==COMMIT_MSG==

Priority:

Concerns / possible downsides (what feedback would you like?):

Is documentation needed?:

Compatibility

Does this PR create any API breaks (e.g. at the Java or HTTP layers) - if so, do we have compatibility?:

Does this PR change the persisted format of any data - if so, do we have forward and backward compatibility?:

The code in this PR may be part of a blue-green deploy. Can upgrades from previous versions safely coexist? (Consider restarts of blue or green nodes.):

Does this PR rely on statements being true about other products at a deployment - if so, do we have correct product dependencies on these products (or other ways of verifying that these statements are true)?:

Does this PR need a schema migration?

Testing and Correctness

What, if any, assumptions are made about the current state of the world? If they change over time, how will we find out?:

What was existing testing like? What have you done to improve it?:

If this PR contains complex concurrent or asynchronous code, is it correct? The onus is on the PR writer to demonstrate this.:

If this PR involves acquiring locks or other shared resources, how do we ensure that these are always released?:

Execution

How would I tell this PR works in production? (Metrics, logs, etc.):

Has the safety of all log arguments been decided correctly?:

Will this change significantly affect our spending on metrics or logs?:

How would I tell that this PR does not work in production? (monitors, etc.):

If this PR does not work as expected, how do I fix that state? Would rollback be straightforward?:

If the above plan is more complex than “recall and rollback”, please tag the support PoC here (if it is the end of the week, tag both the current and next PoC):

Scale

Would this PR be expected to pose a risk at scale? Think of the shopping product at our largest stack.:

Would this PR be expected to perform a large number of database calls, and/or expensive database calls (e.g., row range scans, concurrent CAS)?:

Would this PR ever, with time and scale, become the wrong thing to do - and if so, how would we know that we need to do something differently?:

Development Process

Where should we start reviewing?:

If this PR is in excess of 500 lines excluding versions lock-files, why does it not make sense to split it?:

Please tag any other people who should be aware of this PR:

changelog-app[bot] commented 7 months ago

Generate changelog in changelog-dir>`changelog/@unreleased`</changelog-dir

What do the change types mean? - `feature`: A new feature of the service. - `improvement`: An incremental improvement in the functionality or operation of the service. - `fix`: Remedies the incorrect behaviour of a component of the service in a backwards-compatible way. - `break`: Has the potential to break consumers of this service's API, inclusive of both Palantir services and external consumers of the service's API (e.g. customer-written software or integrations). - `deprecation`: Advertises the intention to remove service functionality without any change to the operation of the service itself. - `manualTask`: Requires the possibility of manual intervention (running a script, eyeballing configuration, performing database surgery, ...) at the time of upgrade for it to succeed. - `migration`: A fully automatic upgrade migration task with no engineer input required. _Note: only one type should be chosen._
How are new versions calculated? - ❗The `break` and `manual task` changelog types will result in a major release! - 🐛 The `fix` changelog type will result in a minor release in most cases, and a patch release version for patch branches. This behaviour is configurable in autorelease. - ✨ All others will result in a minor version release.

Type

- [ ] Feature - [ ] Improvement - [ ] Fix - [ ] Break - [ ] Deprecation - [ ] Manual task - [ ] Migration

Description

SnapshotTransaction stream refactor proof-of-concept **Check the box to generate changelog(s)** - [ ] Generate changelog entry
rhuffy commented 7 months ago

As an exercise I think this is fine, but to actually merge this, we'd probably need more than just "streams are easier to read" as a justification, especially if there's additional allocation cost in the hot read path.

jeremyk-91 commented 7 months ago

Could probably also separate out all the postfiltering logic into its own PostFilter class

Yep, there's actually an internal project ongoing to slice this into many parts (see #7000 or internal RFCs). I'm curious if there was something that motivated this though?

Birthday present for @rhuffy

🎉

rhuffy commented 7 months ago

I was talking to Will about this refactor in general, and how some of the control flow (specifically the use of a resultsAccumulator) is a bit confusing for the uninitiated.

wi11dey commented 7 months ago

yup, motivation was just a proof-of-concept since raymond and i were discussing the code in the context of db upgrading and that it was difficult to plug in the kvs migrator, and also the code was a bit hard to follow for the uninitiated because of the various future chaining. i suggested streams and either raymond or jakub said it wouldn't be possible to use them async so this was a proof of concept using streams while keeping async behavior

the extra allocation of keysToDeleteOrReload could be removed with a map view -- we were already allocating for the result map, just renamed it to filteredResults. the extra iterations could probably also be removed with a little effort

I think this version is easier to reason about since there's no bouncing back and forth between getWithPostFilteringIterate/getWithPostFilteringInternal, i believe @Output in general should be avoided when possible so that the flow of values matches the code flow, and you need to keep less state in your head while reading through the code

anyway not suggesting we merge this right now, but the Future Spliterator approach could be something to keep in mind/reference if this code is going to be refactored anyway