Instance Types

1#!/usr/bin/env python3
2"""Timestream for InfluxDB instance type reference and sizing helper."""
3
4INSTANCE_TYPES = {
5    "db.influx.medium": {"vcpu": 1, "memory_gib": 8, "network": "Up to 10 Gbps"},
6    "db.influx.large": {"vcpu": 2, "memory_gib": 16, "network": "Up to 10 Gbps"},
7    "db.influx.xlarge": {"vcpu": 4, "memory_gib": 32, "network": "Up to 10 Gbps"},
8    "db.influx.2xlarge": {"vcpu": 8, "memory_gib": 64, "network": "Up to 10 Gbps"},
9    "db.influx.4xlarge": {"vcpu": 16, "memory_gib": 128, "network": "Up to 10 Gbps"},
10    "db.influx.8xlarge": {"vcpu": 32, "memory_gib": 256, "network": "10 Gbps"},
11    "db.influx.12xlarge": {"vcpu": 48, "memory_gib": 384, "network": "12 Gbps"},
12    "db.influx.16xlarge": {"vcpu": 64, "memory_gib": 512, "network": "20 Gbps"},
13    "db.influx.24xlarge": {"vcpu": 96, "memory_gib": 768, "network": "25 Gbps"},
14}
15
16
17def list_types():
18    print(f"{'Instance Type':<25} {'vCPU':>5} {'Memory (GiB)':>13} {'Network':<15}")
19    print("-" * 60)
20    for name, spec in INSTANCE_TYPES.items():
21        print(f"{name:<25} {spec['vcpu']:>5} {spec['memory_gib']:>13} {spec['network']:<15}")
22
23
24def recommend(write_rate_per_sec: int, query_concurrency: int, data_retention_days: int):
25    """Simple sizing recommendation based on workload parameters."""
26    if write_rate_per_sec < 1000 and query_concurrency < 5:
27        rec = "db.influx.medium"
28    elif write_rate_per_sec < 10000 and query_concurrency < 20:
29        rec = "db.influx.large"
30    elif write_rate_per_sec < 50000 and query_concurrency < 50:
31        rec = "db.influx.xlarge"
32    elif write_rate_per_sec < 100000:
33        rec = "db.influx.2xlarge"
34    else:
35        rec = "db.influx.4xlarge"
36    # Bump up for high-retention workloads (more storage and memory pressure)
37    type_names = list(INSTANCE_TYPES.keys())
38    idx = type_names.index(rec)
39    if data_retention_days > 365:
40        idx = min(idx + 2, len(type_names) - 1)
41    elif data_retention_days > 90:
42        idx = min(idx + 1, len(type_names) - 1)
43    rec = type_names[idx]
44    spec = INSTANCE_TYPES[rec]
45    print(f"Recommended: {rec} ({spec['vcpu']} vCPU, {spec['memory_gib']} GiB)")
46    print(
47        f"  Based on: {write_rate_per_sec} writes/s, {query_concurrency} concurrent queries, {data_retention_days}d retention"
48    )
49    print(f"  Note: This is a starting point. Monitor CloudWatch metrics and adjust.")
50
51
52if __name__ == "__main__":
53    import sys
54
55    if len(sys.argv) > 1 and sys.argv[1] == "recommend":
56        recommend(int(sys.argv[2]), int(sys.argv[3]), int(sys.argv[4]))
57    else:
58        list_types()