# HG changeset patch
# User Raphaël Gomès <rgomes@octobus.net>
# Date 2019-10-16 08:14:26
# Node ID 5d4046594d6fb6f4056d63a4144a32e5bd4ff0c4
# Parent  726bd0b612a2f2822879921d0302963c27fc2823

rust-dirstatemap: remove additional lookups in dirstatemap

We're copying this shortcut from the Python implementation, pretty standard
for this codebase.

Differential Revision: https://phab.mercurial-scm.org/D7117

diff --git a/mercurial/dirstate.py b/mercurial/dirstate.py
--- a/mercurial/dirstate.py
+++ b/mercurial/dirstate.py
@@ -1784,6 +1784,10 @@ if rustmod is not None:
             if parents and not self._dirtyparents:
                 self.setparents(*parents)
 
+            self.__contains__ = self._rustmap.__contains__
+            self.__getitem__ = self._rustmap.__getitem__
+            self.get = self._rustmap.get
+
         def write(self, st, now):
             parents = self.parents()
             st.write(self._rustmap.write(parents[0], parents[1], now))