| durin42 |
| hg-reviewers |
To aid readability. And to make it easier to search the code base
for protocol instances.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| # the server's network or CPU. | # the server's network or CPU. | ||||
| opts = {'level': self.ui.configint('server', 'zliblevel')} | opts = {'level': self.ui.configint('server', 'zliblevel')} | ||||
| return HGTYPE, util.compengines['zlib'], opts | return HGTYPE, util.compengines['zlib'], opts | ||||
| def iscmd(cmd): | def iscmd(cmd): | ||||
| return cmd in wireproto.commands | return cmd in wireproto.commands | ||||
| def callhttp(repo, req, cmd): | def callhttp(repo, req, cmd): | ||||
| p = webproto(req, repo.ui) | proto = webproto(req, repo.ui) | ||||
| def genversion2(gen, engine, engineopts): | def genversion2(gen, engine, engineopts): | ||||
| # application/mercurial-0.2 always sends a payload header | # application/mercurial-0.2 always sends a payload header | ||||
| # identifying the compression engine. | # identifying the compression engine. | ||||
| name = engine.wireprotosupport().name | name = engine.wireprotosupport().name | ||||
| assert 0 < len(name) < 256 | assert 0 < len(name) < 256 | ||||
| yield struct.pack('B', len(name)) | yield struct.pack('B', len(name)) | ||||
| yield name | yield name | ||||
| for chunk in gen: | for chunk in gen: | ||||
| yield chunk | yield chunk | ||||
| rsp = wireproto.dispatch(repo, p, cmd) | rsp = wireproto.dispatch(repo, proto, cmd) | ||||
| if isinstance(rsp, bytes): | if isinstance(rsp, bytes): | ||||
| req.respond(HTTP_OK, HGTYPE, body=rsp) | req.respond(HTTP_OK, HGTYPE, body=rsp) | ||||
| return [] | return [] | ||||
| elif isinstance(rsp, wireproto.streamres_legacy): | elif isinstance(rsp, wireproto.streamres_legacy): | ||||
| gen = rsp.gen | gen = rsp.gen | ||||
| req.respond(HTTP_OK, HGTYPE) | req.respond(HTTP_OK, HGTYPE) | ||||
| return gen | return gen | ||||
| elif isinstance(rsp, wireproto.streamres): | elif isinstance(rsp, wireproto.streamres): | ||||
| gen = rsp.gen | gen = rsp.gen | ||||
| # This code for compression should not be streamres specific. It | # This code for compression should not be streamres specific. It | ||||
| # is here because we only compress streamres at the moment. | # is here because we only compress streamres at the moment. | ||||
| mediatype, engine, engineopts = p.responsetype(rsp.prefer_uncompressed) | mediatype, engine, engineopts = proto.responsetype( | ||||
| rsp.prefer_uncompressed) | |||||
| gen = engine.compressstream(gen, engineopts) | gen = engine.compressstream(gen, engineopts) | ||||
| if mediatype == HGTYPE2: | if mediatype == HGTYPE2: | ||||
| gen = genversion2(gen, engine, engineopts) | gen = genversion2(gen, engine, engineopts) | ||||
| req.respond(HTTP_OK, mediatype) | req.respond(HTTP_OK, mediatype) | ||||
| return gen | return gen | ||||
| elif isinstance(rsp, wireproto.pushres): | elif isinstance(rsp, wireproto.pushres): | ||||
| val = p.restore() | val = proto.restore() | ||||
| rsp = '%d\n%s' % (rsp.res, val) | rsp = '%d\n%s' % (rsp.res, val) | ||||
| req.respond(HTTP_OK, HGTYPE, body=rsp) | req.respond(HTTP_OK, HGTYPE, body=rsp) | ||||
| return [] | return [] | ||||
| elif isinstance(rsp, wireproto.pusherr): | elif isinstance(rsp, wireproto.pusherr): | ||||
| # drain the incoming bundle | # drain the incoming bundle | ||||
| req.drain() | req.drain() | ||||
| p.restore() | proto.restore() | ||||
| rsp = '0\n%s\n' % rsp.res | rsp = '0\n%s\n' % rsp.res | ||||
| req.respond(HTTP_OK, HGTYPE, body=rsp) | req.respond(HTTP_OK, HGTYPE, body=rsp) | ||||
| return [] | return [] | ||||
| elif isinstance(rsp, wireproto.ooberror): | elif isinstance(rsp, wireproto.ooberror): | ||||
| rsp = rsp.message | rsp = rsp.message | ||||
| req.respond(HTTP_OK, HGERRTYPE, body=rsp) | req.respond(HTTP_OK, HGERRTYPE, body=rsp) | ||||
| return [] | return [] | ||||
| raise error.ProgrammingError('hgweb.protocol internal failure', rsp) | raise error.ProgrammingError('hgweb.protocol internal failure', rsp) | ||||