Mercurial > hg
comparison tests/badserverext.py @ 33192:c538fca0d511
configitems: register the 'badserver.closeafterrecvbytes' config
| author | Pierre-Yves David <pierre-yves.david@octobus.net> |
|---|---|
| date | Fri, 30 Jun 2017 03:27:44 +0200 |
| parents | 8065b4ab0ed7 |
| children | cbb50fd830ea |
comparison
equal
deleted
inserted
replaced
| 33191:8065b4ab0ed7 | 33192:c538fca0d511 |
|---|---|
| 44 configtable = {} | 44 configtable = {} |
| 45 configitem = registrar.configitem(configtable) | 45 configitem = registrar.configitem(configtable) |
| 46 | 46 |
| 47 configitem('badserver', 'closeafteraccept', | 47 configitem('badserver', 'closeafteraccept', |
| 48 default=False, | 48 default=False, |
| 49 ) | |
| 50 configitem('badserver', 'closeafterrecvbytes', | |
| 51 default=0, | |
| 49 ) | 52 ) |
| 50 | 53 |
| 51 # We can't adjust __class__ on a socket instance. So we define a proxy type. | 54 # We can't adjust __class__ on a socket instance. So we define a proxy type. |
| 52 class socketproxy(object): | 55 class socketproxy(object): |
| 53 __slots__ = ( | 56 __slots__ = ( |
| 265 # self.finish_request() which calls self.RequestHandlerClass() which | 268 # self.finish_request() which calls self.RequestHandlerClass() which |
| 266 # is a hgweb.server._httprequesthandler. | 269 # is a hgweb.server._httprequesthandler. |
| 267 def process_request(self, socket, address): | 270 def process_request(self, socket, address): |
| 268 # Wrap socket in a proxy if we need to count bytes. | 271 # Wrap socket in a proxy if we need to count bytes. |
| 269 closeafterrecvbytes = self._ui.configint('badserver', | 272 closeafterrecvbytes = self._ui.configint('badserver', |
| 270 'closeafterrecvbytes', 0) | 273 'closeafterrecvbytes') |
| 271 closeaftersendbytes = self._ui.configint('badserver', | 274 closeaftersendbytes = self._ui.configint('badserver', |
| 272 'closeaftersendbytes', 0) | 275 'closeaftersendbytes', 0) |
| 273 | 276 |
| 274 if closeafterrecvbytes or closeaftersendbytes: | 277 if closeafterrecvbytes or closeaftersendbytes: |
| 275 socket = socketproxy(socket, self.errorlog, | 278 socket = socketproxy(socket, self.errorlog, |