comparison mercurial/wireprototypes.py @ 40022:c537144fdbef

wireprotov2: support response caching One of the things I've learned from managing VCS servers over the years is that they are hard to scale. It is well known that some companies have very beefy (read: very expensive) servers to power their VCS needs. It is also known that specialized servers for various VCS exist in order to facilitate scaling servers. (Mercurial is in this boat.) One of the aspects that make a VCS server hard to scale is the high CPU load incurred by constant client clone/pull operations. To alleviate the scaling pain associated with data retrieval operations, I want to integrate caching into the Mercurial wire protocol server as robustly as possible such that servers can aggressively cache responses and defer as much server load as possible. This commit represents the initial implementation of a general caching layer in wire protocol version 2. We define a new interface and behavior for a wire protocol cacher in repository.py. (This is probably where a reviewer should look first to understand what is going on.) The bulk of the added code is in wireprotov2server.py, where we define how a command can opt in to being cached and integrate caching into command dispatching. From a very high-level: * A command can declare itself as cacheable by providing a callable that can be used to derive a cache key. * At dispatch time, if a command is cacheable, we attempt to construct a cacher and use it for serving the request and/or caching the request. * The dispatch layer handles the bulk of the business logic for caching, making cachers mostly "dumb content stores." * The mechanism for invalidating cached entries (one of the harder parts about caching in general) is by varying the cache key when state changes. As such, cachers don't need to be concerned with cache invalidation. Initially, we've hooked up support for caching "manifestdata" and "filedata" commands. These are the simplest to cache, as they should be immutable over time. Caching of commands related to changeset data is a bit harder (because cache validation is impacted by changes to bookmarks, phases, etc). This will be implemented later. (Strictly speaking, censoring a file should invalidate caches. I've added an inline TODO to track this edge case.) To prove it works, this commit implements a test-only extension providing in-memory caching backed by an lrucachedict. A new test showing this extension behaving properly is added. FWIW, the cacher is ~50 lines of code, demonstrating the relative ease with which a cache can be added to a server. While the test cacher is not suitable for production workloads, just for kicks I performed a clone of just the changeset and manifest data for the mozilla-unified repository. With a fully warmed cache (of just the manifest data since changeset data is not cached), server-side CPU usage dropped from ~73s to ~28s. That's pretty significant and demonstrates the potential that response caching has on server scalability! Differential Revision: https://phab.mercurial-scm.org/D4773
author Gregory Szorc <gregory.szorc@gmail.com>
date Wed, 26 Sep 2018 17:16:56 -0700
parents ed919b90acda
children b099e6032f38
comparison
equal deleted inserted replaced
40021:ed919b90acda 40022:c537144fdbef
230 """ 230 """
231 231
232 class commandentry(object): 232 class commandentry(object):
233 """Represents a declared wire protocol command.""" 233 """Represents a declared wire protocol command."""
234 def __init__(self, func, args='', transports=None, 234 def __init__(self, func, args='', transports=None,
235 permission='push'): 235 permission='push', cachekeyfn=None):
236 self.func = func 236 self.func = func
237 self.args = args 237 self.args = args
238 self.transports = transports or set() 238 self.transports = transports or set()
239 self.permission = permission 239 self.permission = permission
240 self.cachekeyfn = cachekeyfn
240 241
241 def _merge(self, func, args): 242 def _merge(self, func, args):
242 """Merge this instance with an incoming 2-tuple. 243 """Merge this instance with an incoming 2-tuple.
243 244
244 This is called when a caller using the old 2-tuple API attempts 245 This is called when a caller using the old 2-tuple API attempts