partition-sim 0.1.0

Simulate network partitions in a docker cluster.
Documentation 
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

#!/usr/bin/env python3

import uuid
import requests
import argparse
import socket
from pprint import pprint
import os


def my_local_ip():
    return socket.gethostbyname(socket.getfqdn())


def parse_args():
    parser = argparse.ArgumentParser("Register a service with Consul")
    parser.add_argument(
        "--consul-base-url",
        type=str,
        default=f'http://{os.getenv("CONSUL_ADDR", "127.0.0.1")}:8500',
        help="Base URL (without trailing slash) for the Consul HTTP API"
    )
    parser.add_argument(
        "--name",
        type=str,
        required=True,
        help="Name of the service to register"
    )
    parser.add_argument(
        "--port",
        type=int,
        nargs="+",
        required=True,
        help="Ports for the service to register"
    )
    parser.add_argument(
        "--address",
        type=str,
        default=my_local_ip(),
        help="Address of the service to register"
    )
    return parser.parse_args()



def main():
    args = parse_args()
    for port in args.port:
        output = register_service(
            name=args.name,
            id=f"{args.name}-{str(args.address)}-{str(port)}",
            port=port,
            address=args.address,
            base_url=args.consul_base_url
        )
        pprint(output)


def register_service(
    base_url: str,
    name: str,
    id: str,
    port: int,
    address: str
):
    data = {
        "Name": name,
        "ID": id,
        "Port": port,
        "Address": address,
        "Check": {
            "DeregisterCriticalServiceAfter": "15s",
            "HTTP": f"http://{address}:{port}/health",
            "Interval": "10s"
        },
        "Meta": {
            "raft-infra-test-node": "true"
        }
    }
    pprint(data, indent=4)
    response = requests.put(
        f"{base_url}/v1/agent/service/register",
        json=data,
        headers={"Content-Type": "application/json"}
    )
    response.raise_for_status()
    return response


if __name__ == '__main__':
    main()