refactor: rename to 'get_primary_key_or_insert'

Signed-off-by: ekexium <[email protected]>

Author: ekexium

src/transaction/buffer.rs

@@ -27,7 +27,7 @@ impl InnerBuffer {
         self.entry_map.insert(key, entry);
     }
 
-    pub fn get_primary_key_or(&mut self, key: &Key) -> &Key {
+    pub fn get_primary_key_or_insert(&mut self, key: &Key) -> &Key {
         self.primary_key.get_or_insert(key.clone())
     }
 }
@@ -50,8 +50,12 @@ impl Buffer {
     }
 
     /// Get the primary key of the buffer, if not exists, use `key` as the primary key.
-    pub async fn get_primary_key_or(&self, key: &Key) -> Key {
-        self.inner.lock().await.get_primary_key_or(key).clone()
+    pub async fn get_primary_key_or_insert(&self, key: &Key) -> Key {
+        self.inner
+            .lock()
+            .await
+            .get_primary_key_or_insert(key)
+            .clone()
     }
 
     /// Get a value from the buffer.

src/transaction/transaction.rs

@@ -674,7 +674,7 @@ impl<PdC: PdClient> Transaction<PdC> {
         let pairs = plan.execute().await;
 
         // primary key will be set here if needed
-        self.buffer.get_primary_key_or(&first_key).await;
+        self.buffer.get_primary_key_or_insert(&first_key).await;
 
         self.start_auto_heartbeat().await;