Skip to content
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

[STACKED #1228] [ENH] Grpc Coordinator/SysDB #1229

Merged
merged 3 commits into from
Oct 16, 2023
Merged

Conversation

HammadB
Copy link
Collaborator

@HammadB HammadB commented Oct 11, 2023

Description of changes

This PR is stacked on #1228, please TAL at that before this.

Summarize the changes made by this PR.

  • Improvements & Bug fixes
    • n/a
  • New functionality
    • Introduced a grpc sysdb implementation.
    • Migrate the protobufs to support Null vs Presence where needed.
    • Added minimal protobuf definitions to implement the grpc coordinator
    • Add a mock grpc sysdb server that reimplements the core functionality in memory for testing.

Will address the following open questions in stacked PRs!
Open questions/issues:

Test plan

How are these changes tested?
I added a mock gRPC server that implements the basic functionality of sysdb with in memory data structures.

We will run the coordinator tests with this impl in the bin/cluster test once the go coordinator service is ready from @Ishiihara

Documentation Changes

None required.

@github-actions
Copy link

Reviewer Checklist

Please leverage this checklist to ensure your code review is thorough before approving

Testing, Bugs, Errors, Logs, Documentation

  • Can you think of any use case in which the code does not behave as intended? Have they been tested?
  • Can you think of any inputs or external events that could break the code? Is user input validated and safe? Have they been tested?
  • If appropriate, are there adequate property based tests?
  • If appropriate, are there adequate unit tests?
  • Should any logging, debugging, tracing information be added or removed?
  • Are error messages user-friendly?
  • Have all documentation changes needed been made?
  • Have all non-obvious changes been commented?

System Compatibility

  • Are there any potential impacts on other parts of the system or backward compatibility?
  • Does this change intersect with any items on our roadmap, and if so, is there a plan for fitting them together?

Quality

  • Is this code of a unexpectedly high quality (Readbility, Modularity, Intuitiveness)

@HammadB HammadB changed the base branch from main to hammad/move_idl October 11, 2023 03:36
@HammadB HammadB changed the title [WIP] Grpc Coordinator/SysDB [WIP] [STACKED #1228] [ENH] Grpc Coordinator/SysDB Oct 11, 2023
@HammadB HammadB requested review from Ishiihara and beggers and removed request for Ishiihara and beggers October 11, 2023 03:41
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

todo: factor out the services that are not core data models

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Will do in follow-up PR, is a separate concern.

@HammadB HammadB force-pushed the hammad/coordinator_sysdb branch from 0cdb5d7 to 6310595 Compare October 12, 2023 08:16
@HammadB HammadB changed the title [WIP] [STACKED #1228] [ENH] Grpc Coordinator/SysDB [STACKED #1228] [ENH] Grpc Coordinator/SysDB Oct 12, 2023
@@ -35,8 +37,19 @@ def sqlite_persistent() -> Generator[SysDB, None, None]:
shutil.rmtree(save_path)


def grpc_with_mock_server() -> Generator[SysDB, None, None]:
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Adds a test fixture

@HammadB
Copy link
Collaborator Author

HammadB commented Oct 12, 2023

test failures are just a config key bug, will patch.

Copy link
Contributor

@Ishiihara Ishiihara left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

idl/chromadb/proto/chroma.proto Show resolved Hide resolved
chromadb/db/impl/grpc/server.py Show resolved Hide resolved
chromadb/db/impl/grpc/server.py Show resolved Hide resolved
chromadb/db/impl/grpc/server.py Show resolved Hide resolved
@HammadB HammadB force-pushed the hammad/coordinator_sysdb branch from 6fd33f9 to f35e576 Compare October 13, 2023 21:05
@HammadB HammadB changed the base branch from hammad/move_idl to main October 13, 2023 21:05
@HammadB HammadB force-pushed the hammad/coordinator_sysdb branch from f35e576 to f2cf8a6 Compare October 13, 2023 21:28
@HammadB HammadB force-pushed the hammad/coordinator_sysdb branch from f2cf8a6 to b25c8bd Compare October 13, 2023 23:20
@HammadB HammadB merged commit a9d654b into main Oct 16, 2023
85 checks passed
HammadB added a commit that referenced this pull request Oct 16, 2023
… creation into SysDB using the policy (#1237)

## Description of changes

*Summarize the changes made by this PR.*
 - Improvements & Bug fixes
- Adds a CollectionAssignmentPolicy and moves topic creation down into
SysDB
- Changes SysDB to not accept a Collection Object in create_collection()
but params instead
 - New functionality
       - None

TODO: 
- [ ] delete topic should also live in the sysdb now.

## Test plan
*How are these changes tested?*
Existing tests were modified with the refactor and ensured to pass.

## Documentation Changes
None required
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants