Mercurial > public > mercurial-scm > hg-stable
comparison mercurial/revset.py @ 20536:532b114a6e02
revset: added generatorset class with cached __contains__ method
| author | Lucas Moscovicz <lmoscovicz@fb.com> |
|---|---|
| date | Wed, 05 Feb 2014 15:23:11 -0800 |
| parents | 4849f574aa24 |
| children | fe220013e4db |
comparison
equal
deleted
inserted
replaced
| 20534:4849f574aa24 | 20536:532b114a6e02 |
|---|---|
| 2174 self._subset.reverse() | 2174 self._subset.reverse() |
| 2175 | 2175 |
| 2176 def set(self): | 2176 def set(self): |
| 2177 return set([r for r in self]) | 2177 return set([r for r in self]) |
| 2178 | 2178 |
| 2179 class generatorset(object): | |
| 2180 """Wrapper structure for generators that provides lazy membership.""" | |
| 2181 def __init__(self, gen): | |
| 2182 self._gen = gen | |
| 2183 self._iter = iter(gen) | |
| 2184 self._cache = {} | |
| 2185 | |
| 2186 def __contains__(self, x): | |
| 2187 if x in self._cache: | |
| 2188 return self._cache[x] | |
| 2189 | |
| 2190 while True: | |
| 2191 try: | |
| 2192 l = self._iter.next() | |
| 2193 self._cache[l] = True | |
| 2194 if l == x: | |
| 2195 return True | |
| 2196 except (StopIteration): | |
| 2197 break | |
| 2198 | |
| 2199 self._cache[x] = False | |
| 2200 return False | |
| 2201 | |
| 2202 def __iter__(self): | |
| 2203 for item in self._gen: | |
| 2204 self._cache[item] = True | |
| 2205 yield item | |
| 2206 | |
| 2207 def set(self): | |
| 2208 return self | |
| 2209 | |
| 2179 class spanset(object): | 2210 class spanset(object): |
| 2180 """Duck type for baseset class which represents a range of revisions and | 2211 """Duck type for baseset class which represents a range of revisions and |
| 2181 can work lazily and without having all the range in memory | 2212 can work lazily and without having all the range in memory |
| 2182 """ | 2213 """ |
| 2183 def __init__(self, repo, start=0, end=None): | 2214 def __init__(self, repo, start=0, end=None): |