fix(bindings): stabilize refs listing and branch metadata

Author: majin1102

java/lance-jni/src/blocking_dataset.rs

@@ -2374,14 +2374,15 @@ fn inner_list_tags<'local>(
     env: &mut JNIEnv<'local>,
     java_dataset: JObject,
 ) -> Result<JObject<'local>> {
-    let tag_map = {
+    let mut tags: Vec<_> = {
         let dataset_guard =
             unsafe { env.get_rust_field::<_, _, BlockingDataset>(java_dataset, NATIVE_DATASET) }?;
-        dataset_guard.list_tags()?
+        dataset_guard.list_tags()?.into_iter().collect()
     };
+    tags.sort_unstable_by(|(left_name, _), (right_name, _)| left_name.cmp(right_name));
     let array_list = env.new_object("java/util/ArrayList", "()V", &[])?;
 
-    for (tag_name, tag_contents) in tag_map {
+    for (tag_name, tag_contents) in tags {
         let branch_name: JObject = if let Some(branch_name) = tag_contents.branch.as_ref() {
             env.new_string(branch_name)?.into()
         } else {
@@ -2544,11 +2545,12 @@ fn inner_list_branches<'local>(
     env: &mut JNIEnv<'local>,
     java_dataset: JObject,
 ) -> Result<JObject<'local>> {
-    let branches = {
+    let mut branches: Vec<_> = {
         let dataset_guard =
             unsafe { env.get_rust_field::<_, _, BlockingDataset>(java_dataset, NATIVE_DATASET) }?;
-        dataset_guard.list_branches()?
+        dataset_guard.list_branches()?.into_iter().collect()
     };
+    branches.sort_unstable_by(|(left_name, _), (right_name, _)| left_name.cmp(right_name));
     let array_list = env.new_object("java/util/ArrayList", "()V", &[])?;
 
     for (name, contents) in branches {

python/python/tests/test_dataset.py

@@ -5278,6 +5278,9 @@ def test_branches(tmp_path: Path):
     assert b1_meta["manifest_size"] > 0
     assert "create_at" in b1_meta
     assert b1_meta["metadata"] == "branch one"
+    ordered_branches = dict(ds_main.branches.list_ordered())
+    assert ordered_branches["branch1"]["metadata"] == "branch one"
+    assert "metadata" in ordered_branches["branch2"]
 
     try:
         ds_main.checkout_version("branch_not_exists")

python/src/dataset.rs

@@ -1881,6 +1881,7 @@ impl Dataset {
             dict.set_item("parent_version", meta.parent_version)?;
             dict.set_item("create_at", meta.create_at)?;
             dict.set_item("manifest_size", meta.manifest_size)?;
+            dict.set_item("metadata", meta.metadata.clone())?;
             out.push((name, dict.into_py_any(py)?));
         }
         Ok(out)