File: ssh_roundtrip.py

package info (click to toggle)
python-mitogen 0.3.39-1
  • links: PTS, VCS
  • area: main
  • in suites: forky
  • size: 6,732 kB
  • sloc: python: 24,733; sh: 198; makefile: 74; perl: 19; ansic: 18
file content (46 lines) | stat: -rw-r--r-- 1,163 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
"""
Measure latency of SSH RPC.
"""

import getpass

import mitogen.core
import mitogen.utils
import ansible_mitogen.affinity

mitogen.utils.setup_gil()
ansible_mitogen.affinity.policy.assign_worker()


def do_nothing():
    pass

@mitogen.main()
def main(router):
    import optparse
    parser = optparse.OptionParser(description=__doc__)
    parser.add_option(
        '-p', '--python', metavar='CMD', default='python3',
        help='Remote python path (default %default)')
    parser.add_option(
        '-u', '--user', metavar='S', default=getpass.getuser(),
        help='Remote username (default %default)')
    parser.add_option('--debug', action='store_true')
    opts, args = parser.parse_args()

    f = router.ssh(
        hostname=args[0], python_path=opts.python, username=opts.user,
        debug=opts.debug,
    )
    f.call(do_nothing)
    t0 = mitogen.core.now()
    end = mitogen.core.now() + 5.0
    i = 0
    while mitogen.core.now() < end:
        f.call(do_nothing)
        i += 1

    t1 = mitogen.core.now()
    iterations = i + 1
    mean = (t1 - t0) / iterations
    print('++ iterations %d, mean %.03f ms' % (iterations, 1e3 * mean))