cpu-loader 0.0.3__tar.gz → 0.1.2__tar.gz

{cpu_loader-0.0.3/src/cpu_loader.egg-info → cpu_loader-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cpu-loader
-Version: 0.0.3
+Version: 0.1.2
 Summary: A tool to generate CPU load on a system with runtime configuration through a WebUI and REST API
 Author-email: the78mole <the78mole@users.noreply.github.com>
 License: MIT
@@ -33,6 +33,7 @@ Requires-Dist: pydantic==2.10.0
 Requires-Dist: requests==2.32.3
 Requires-Dist: psutil>=5.9.0
 Requires-Dist: websockets>=12.0
+Requires-Dist: paho-mqtt>=2.1.0
 Dynamic: license-file
 
 # CPU Loader
@@ -42,7 +43,7 @@ Dynamic: license-file
 [![Python Version](https://img.shields.io/pypi/pyversions/cpu-loader)](https://pypi.org/project/cpu-loader/)
 [![License](https://img.shields.io/github/license/the78mole/cpu-loader)](https://github.com/the78mole/cpu-loader/blob/main/LICENSE)
 
-A high-performance CPU load generator with **per-thread control** through an intuitive WebUI and REST API. Built with a native C extension using pthreads for accurate, nanosecond-precision load generation.
+A tool to generate CPU load on a system with runtime configuration through a WebUI and REST API.
 
 ## ✨ Key Features
 
@@ -53,6 +54,7 @@ A high-performance CPU load generator with **per-thread control** through an int
 - **🚀 REST API**: Complete programmatic control for automation and testing
 - **📱 Responsive Design**: Works seamlessly on desktop and mobile devices
 - **🔄 Instant Updates**: Changes take effect immediately with sub-second response time
+- 📡 **MQTT Integration**: Publish metrics and settings to MQTT broker for IoT/monitoring systems
 
 ## 🖼️ Screenshots
 
@@ -133,6 +135,22 @@ uv run src/main.py
 
 The server will start on `http://localhost:8000`
 
+#### Command-Line Options
+
+```bash
+uv run src/main.py --help
+```
+
+**Available Options:**
+- `--host HOST`: Host to bind the server to (default: 0.0.0.0)
+- `--port PORT`: Port to bind the server to (default: 8000)
+- `--mqtt-broker-host HOST`: MQTT broker hostname
+- `--mqtt-broker-port PORT`: MQTT broker port (default: 1883)
+- `--mqtt-username USER`: MQTT username
+- `--mqtt-password PASS`: MQTT password
+- `--mqtt-topic-prefix PREFIX`: MQTT topic prefix (default: cpu-loader)
+- `--mqtt-client-id ID`: MQTT client ID (default: cpu-loader)
+
 ### WebUI
 
 Open your browser and navigate to `http://localhost:8000`
@@ -188,6 +206,88 @@ curl -X POST http://localhost:8000/api/threads \
 
 Once the server is running, visit `http://localhost:8000/docs` for interactive API documentation powered by Swagger UI.
 
+## MQTT Publishing
+
+CPU Loader can publish real-time CPU metrics and load control settings to an MQTT broker for integration with home automation systems, monitoring tools, or custom applications.
+
+### MQTT Topics
+
+The application publishes to two topics:
+
+1. **`{prefix}/cpu_metrics`**: Published every second with current CPU utilization
+   ```json
+   {
+     "total_cpu_percent": 25.5,
+     "per_cpu_percent": [25.0, 26.0, 25.3, 25.7]
+   }
+   ```
+
+2. **`{prefix}/load_settings`**: Published when load settings change (retained message)
+   ```json
+   {
+     "num_threads": 4,
+     "loads": {"0": 25.0, "1": 25.0, "2": 25.0, "3": 25.0},
+     "average_load": 25.0
+   }
+   ```
+
+### Configuration
+
+MQTT can be configured using environment variables or command-line arguments. Command-line arguments take precedence over environment variables.
+
+#### Using Environment Variables
+
+```bash
+export MQTT_BROKER_HOST=mqtt.example.com
+export MQTT_BROKER_PORT=1883
+export MQTT_USERNAME=myuser
+export MQTT_PASSWORD=mypassword
+export MQTT_TOPIC_PREFIX=cpu-loader
+export MQTT_CLIENT_ID=cpu-loader-001
+
+uv run src/main.py
+```
+
+#### Using Command-Line Arguments
+
+```bash
+uv run src/main.py \
+  --mqtt-broker-host mqtt.example.com \
+  --mqtt-broker-port 1883 \
+  --mqtt-username myuser \
+  --mqtt-password mypassword \
+  --mqtt-topic-prefix cpu-loader \
+  --mqtt-client-id cpu-loader-001
+```
+
+#### Testing with Mosquitto
+
+To test MQTT publishing locally:
+
+```bash
+# Install mosquitto
+sudo apt-get install mosquitto mosquitto-clients
+
+# Start CPU Loader with MQTT
+uv run src/main.py --mqtt-broker-host localhost
+
+# Subscribe to all topics (in another terminal)
+mosquitto_sub -h localhost -t "cpu-loader/#" -v
+```
+
+### MQTT Settings
+
+| Setting | Environment Variable | CLI Argument | Default | Description |
+|---------|---------------------|--------------|---------|-------------|
+| Broker Host | `MQTT_BROKER_HOST` | `--mqtt-broker-host` | None | MQTT broker hostname or IP |
+| Broker Port | `MQTT_BROKER_PORT` | `--mqtt-broker-port` | 1883 | MQTT broker port |
+| Username | `MQTT_USERNAME` | `--mqtt-username` | None | MQTT authentication username |
+| Password | `MQTT_PASSWORD` | `--mqtt-password` | None | MQTT authentication password |
+| Topic Prefix | `MQTT_TOPIC_PREFIX` | `--mqtt-topic-prefix` | cpu-loader | Prefix for all MQTT topics |
+| Client ID | `MQTT_CLIENT_ID` | `--mqtt-client-id` | cpu-loader | MQTT client identifier |
+
+**Note:** If no MQTT broker host is configured, MQTT publishing will be disabled and the application will function normally without it.
+
 ## Example Script
 
 An example script (`src/example.py`) is provided to demonstrate programmatic control:
@@ -212,6 +312,7 @@ The example demonstrates:
 - **src/cpu_loader_core.c**: High-performance C implementation using pthreads for CPU load generation
 - **src/cpu_loader.py**: Python wrapper providing a clean API to the C extension
 - **src/main.py**: FastAPI application with REST API and embedded WebUI
+- **src/mqtt_publisher.py**: MQTT client for publishing metrics and settings
 - **Threading Model**: Native pthreads for maximum efficiency and precise timing
 - **Load Algorithm**: High-resolution busy-wait loops with nanosecond precision
 
@@ -221,6 +322,7 @@ The example demonstrates:
 - FastAPI 0.115.0+
 - Uvicorn 0.32.0+
 - Pydantic 2.10.0+
+- paho-mqtt 2.1.0+ (for MQTT publishing)
 - C compiler (gcc or clang) for building the extension
 
 ## Use Cases
@@ -230,6 +332,8 @@ The example demonstrates:
 - **Thermal Testing**: Check cooling system effectiveness
 - **Power Consumption Analysis**: Measure power usage at different load levels
 - **Benchmarking**: Create reproducible load scenarios
+- **IoT Integration**: Integrate with Home Assistant, Node-RED, or other MQTT-based systems
+- **Monitoring**: Feed CPU metrics into monitoring dashboards via MQTT
 
 ## Development
 

{cpu_loader-0.0.3 → cpu_loader-0.1.2}/README.md

@@ -5,7 +5,7 @@
 [![Python Version](https://img.shields.io/pypi/pyversions/cpu-loader)](https://pypi.org/project/cpu-loader/)
 [![License](https://img.shields.io/github/license/the78mole/cpu-loader)](https://github.com/the78mole/cpu-loader/blob/main/LICENSE)
 
-A high-performance CPU load generator with **per-thread control** through an intuitive WebUI and REST API. Built with a native C extension using pthreads for accurate, nanosecond-precision load generation.
+A tool to generate CPU load on a system with runtime configuration through a WebUI and REST API.
 
 ## ✨ Key Features
 
@@ -16,6 +16,7 @@ A high-performance CPU load generator with **per-thread control** through an int
 - **🚀 REST API**: Complete programmatic control for automation and testing
 - **📱 Responsive Design**: Works seamlessly on desktop and mobile devices
 - **🔄 Instant Updates**: Changes take effect immediately with sub-second response time
+- 📡 **MQTT Integration**: Publish metrics and settings to MQTT broker for IoT/monitoring systems
 
 ## 🖼️ Screenshots
 
@@ -96,6 +97,22 @@ uv run src/main.py
 
 The server will start on `http://localhost:8000`
 
+#### Command-Line Options
+
+```bash
+uv run src/main.py --help
+```
+
+**Available Options:**
+- `--host HOST`: Host to bind the server to (default: 0.0.0.0)
+- `--port PORT`: Port to bind the server to (default: 8000)
+- `--mqtt-broker-host HOST`: MQTT broker hostname
+- `--mqtt-broker-port PORT`: MQTT broker port (default: 1883)
+- `--mqtt-username USER`: MQTT username
+- `--mqtt-password PASS`: MQTT password
+- `--mqtt-topic-prefix PREFIX`: MQTT topic prefix (default: cpu-loader)
+- `--mqtt-client-id ID`: MQTT client ID (default: cpu-loader)
+
 ### WebUI
 
 Open your browser and navigate to `http://localhost:8000`
@@ -151,6 +168,88 @@ curl -X POST http://localhost:8000/api/threads \
 
 Once the server is running, visit `http://localhost:8000/docs` for interactive API documentation powered by Swagger UI.
 
+## MQTT Publishing
+
+CPU Loader can publish real-time CPU metrics and load control settings to an MQTT broker for integration with home automation systems, monitoring tools, or custom applications.
+
+### MQTT Topics
+
+The application publishes to two topics:
+
+1. **`{prefix}/cpu_metrics`**: Published every second with current CPU utilization
+   ```json
+   {
+     "total_cpu_percent": 25.5,
+     "per_cpu_percent": [25.0, 26.0, 25.3, 25.7]
+   }
+   ```
+
+2. **`{prefix}/load_settings`**: Published when load settings change (retained message)
+   ```json
+   {
+     "num_threads": 4,
+     "loads": {"0": 25.0, "1": 25.0, "2": 25.0, "3": 25.0},
+     "average_load": 25.0
+   }
+   ```
+
+### Configuration
+
+MQTT can be configured using environment variables or command-line arguments. Command-line arguments take precedence over environment variables.
+
+#### Using Environment Variables
+
+```bash
+export MQTT_BROKER_HOST=mqtt.example.com
+export MQTT_BROKER_PORT=1883
+export MQTT_USERNAME=myuser
+export MQTT_PASSWORD=mypassword
+export MQTT_TOPIC_PREFIX=cpu-loader
+export MQTT_CLIENT_ID=cpu-loader-001
+
+uv run src/main.py
+```
+
+#### Using Command-Line Arguments
+
+```bash
+uv run src/main.py \
+  --mqtt-broker-host mqtt.example.com \
+  --mqtt-broker-port 1883 \
+  --mqtt-username myuser \
+  --mqtt-password mypassword \
+  --mqtt-topic-prefix cpu-loader \
+  --mqtt-client-id cpu-loader-001
+```
+
+#### Testing with Mosquitto
+
+To test MQTT publishing locally:
+
+```bash
+# Install mosquitto
+sudo apt-get install mosquitto mosquitto-clients
+
+# Start CPU Loader with MQTT
+uv run src/main.py --mqtt-broker-host localhost
+
+# Subscribe to all topics (in another terminal)
+mosquitto_sub -h localhost -t "cpu-loader/#" -v
+```
+
+### MQTT Settings
+
+| Setting | Environment Variable | CLI Argument | Default | Description |
+|---------|---------------------|--------------|---------|-------------|
+| Broker Host | `MQTT_BROKER_HOST` | `--mqtt-broker-host` | None | MQTT broker hostname or IP |
+| Broker Port | `MQTT_BROKER_PORT` | `--mqtt-broker-port` | 1883 | MQTT broker port |
+| Username | `MQTT_USERNAME` | `--mqtt-username` | None | MQTT authentication username |
+| Password | `MQTT_PASSWORD` | `--mqtt-password` | None | MQTT authentication password |
+| Topic Prefix | `MQTT_TOPIC_PREFIX` | `--mqtt-topic-prefix` | cpu-loader | Prefix for all MQTT topics |
+| Client ID | `MQTT_CLIENT_ID` | `--mqtt-client-id` | cpu-loader | MQTT client identifier |
+
+**Note:** If no MQTT broker host is configured, MQTT publishing will be disabled and the application will function normally without it.
+
 ## Example Script
 
 An example script (`src/example.py`) is provided to demonstrate programmatic control:
@@ -175,6 +274,7 @@ The example demonstrates:
 - **src/cpu_loader_core.c**: High-performance C implementation using pthreads for CPU load generation
 - **src/cpu_loader.py**: Python wrapper providing a clean API to the C extension
 - **src/main.py**: FastAPI application with REST API and embedded WebUI
+- **src/mqtt_publisher.py**: MQTT client for publishing metrics and settings
 - **Threading Model**: Native pthreads for maximum efficiency and precise timing
 - **Load Algorithm**: High-resolution busy-wait loops with nanosecond precision
 
@@ -184,6 +284,7 @@ The example demonstrates:
 - FastAPI 0.115.0+
 - Uvicorn 0.32.0+
 - Pydantic 2.10.0+
+- paho-mqtt 2.1.0+ (for MQTT publishing)
 - C compiler (gcc or clang) for building the extension
 
 ## Use Cases
@@ -193,6 +294,8 @@ The example demonstrates:
 - **Thermal Testing**: Check cooling system effectiveness
 - **Power Consumption Analysis**: Measure power usage at different load levels
 - **Benchmarking**: Create reproducible load scenarios
+- **IoT Integration**: Integrate with Home Assistant, Node-RED, or other MQTT-based systems
+- **Monitoring**: Feed CPU metrics into monitoring dashboards via MQTT
 
 ## Development
 

{cpu_loader-0.0.3 → cpu_loader-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cpu-loader"
-version = "0.0.3"
+version = "0.1.2"
 description = "A tool to generate CPU load on a system with runtime configuration through a WebUI and REST API"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -33,6 +33,7 @@ dependencies = [
     "requests==2.32.3",
     "psutil>=5.9.0",
     "websockets>=12.0",
+    "paho-mqtt>=2.1.0",
 ]
 
 [project.scripts]
@@ -50,4 +51,4 @@ build-backend = "setuptools.build_meta"
 
 [tool.setuptools]
 package-dir = {"" = "src"}
-py-modules = ["cpu_loader", "main", "example"]
+py-modules = ["cpu_loader", "main", "example", "mqtt_publisher"]

{cpu_loader-0.0.3 → cpu_loader-0.1.2/src/cpu_loader.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cpu-loader
-Version: 0.0.3
+Version: 0.1.2
 Summary: A tool to generate CPU load on a system with runtime configuration through a WebUI and REST API
 Author-email: the78mole <the78mole@users.noreply.github.com>
 License: MIT
@@ -33,6 +33,7 @@ Requires-Dist: pydantic==2.10.0
 Requires-Dist: requests==2.32.3
 Requires-Dist: psutil>=5.9.0
 Requires-Dist: websockets>=12.0
+Requires-Dist: paho-mqtt>=2.1.0
 Dynamic: license-file
 
 # CPU Loader
@@ -42,7 +43,7 @@ Dynamic: license-file
 [![Python Version](https://img.shields.io/pypi/pyversions/cpu-loader)](https://pypi.org/project/cpu-loader/)
 [![License](https://img.shields.io/github/license/the78mole/cpu-loader)](https://github.com/the78mole/cpu-loader/blob/main/LICENSE)
 
-A high-performance CPU load generator with **per-thread control** through an intuitive WebUI and REST API. Built with a native C extension using pthreads for accurate, nanosecond-precision load generation.
+A tool to generate CPU load on a system with runtime configuration through a WebUI and REST API.
 
 ## ✨ Key Features
 
@@ -53,6 +54,7 @@ A high-performance CPU load generator with **per-thread control** through an int
 - **🚀 REST API**: Complete programmatic control for automation and testing
 - **📱 Responsive Design**: Works seamlessly on desktop and mobile devices
 - **🔄 Instant Updates**: Changes take effect immediately with sub-second response time
+- 📡 **MQTT Integration**: Publish metrics and settings to MQTT broker for IoT/monitoring systems
 
 ## 🖼️ Screenshots
 
@@ -133,6 +135,22 @@ uv run src/main.py
 
 The server will start on `http://localhost:8000`
 
+#### Command-Line Options
+
+```bash
+uv run src/main.py --help
+```
+
+**Available Options:**
+- `--host HOST`: Host to bind the server to (default: 0.0.0.0)
+- `--port PORT`: Port to bind the server to (default: 8000)
+- `--mqtt-broker-host HOST`: MQTT broker hostname
+- `--mqtt-broker-port PORT`: MQTT broker port (default: 1883)
+- `--mqtt-username USER`: MQTT username
+- `--mqtt-password PASS`: MQTT password
+- `--mqtt-topic-prefix PREFIX`: MQTT topic prefix (default: cpu-loader)
+- `--mqtt-client-id ID`: MQTT client ID (default: cpu-loader)
+
 ### WebUI
 
 Open your browser and navigate to `http://localhost:8000`
@@ -188,6 +206,88 @@ curl -X POST http://localhost:8000/api/threads \
 
 Once the server is running, visit `http://localhost:8000/docs` for interactive API documentation powered by Swagger UI.
 
+## MQTT Publishing
+
+CPU Loader can publish real-time CPU metrics and load control settings to an MQTT broker for integration with home automation systems, monitoring tools, or custom applications.
+
+### MQTT Topics
+
+The application publishes to two topics:
+
+1. **`{prefix}/cpu_metrics`**: Published every second with current CPU utilization
+   ```json
+   {
+     "total_cpu_percent": 25.5,
+     "per_cpu_percent": [25.0, 26.0, 25.3, 25.7]
+   }
+   ```
+
+2. **`{prefix}/load_settings`**: Published when load settings change (retained message)
+   ```json
+   {
+     "num_threads": 4,
+     "loads": {"0": 25.0, "1": 25.0, "2": 25.0, "3": 25.0},
+     "average_load": 25.0
+   }
+   ```
+
+### Configuration
+
+MQTT can be configured using environment variables or command-line arguments. Command-line arguments take precedence over environment variables.
+
+#### Using Environment Variables
+
+```bash
+export MQTT_BROKER_HOST=mqtt.example.com
+export MQTT_BROKER_PORT=1883
+export MQTT_USERNAME=myuser
+export MQTT_PASSWORD=mypassword
+export MQTT_TOPIC_PREFIX=cpu-loader
+export MQTT_CLIENT_ID=cpu-loader-001
+
+uv run src/main.py
+```
+
+#### Using Command-Line Arguments
+
+```bash
+uv run src/main.py \
+  --mqtt-broker-host mqtt.example.com \
+  --mqtt-broker-port 1883 \
+  --mqtt-username myuser \
+  --mqtt-password mypassword \
+  --mqtt-topic-prefix cpu-loader \
+  --mqtt-client-id cpu-loader-001
+```
+
+#### Testing with Mosquitto
+
+To test MQTT publishing locally:
+
+```bash
+# Install mosquitto
+sudo apt-get install mosquitto mosquitto-clients
+
+# Start CPU Loader with MQTT
+uv run src/main.py --mqtt-broker-host localhost
+
+# Subscribe to all topics (in another terminal)
+mosquitto_sub -h localhost -t "cpu-loader/#" -v
+```
+
+### MQTT Settings
+
+| Setting | Environment Variable | CLI Argument | Default | Description |
+|---------|---------------------|--------------|---------|-------------|
+| Broker Host | `MQTT_BROKER_HOST` | `--mqtt-broker-host` | None | MQTT broker hostname or IP |
+| Broker Port | `MQTT_BROKER_PORT` | `--mqtt-broker-port` | 1883 | MQTT broker port |
+| Username | `MQTT_USERNAME` | `--mqtt-username` | None | MQTT authentication username |
+| Password | `MQTT_PASSWORD` | `--mqtt-password` | None | MQTT authentication password |
+| Topic Prefix | `MQTT_TOPIC_PREFIX` | `--mqtt-topic-prefix` | cpu-loader | Prefix for all MQTT topics |
+| Client ID | `MQTT_CLIENT_ID` | `--mqtt-client-id` | cpu-loader | MQTT client identifier |
+
+**Note:** If no MQTT broker host is configured, MQTT publishing will be disabled and the application will function normally without it.
+
 ## Example Script
 
 An example script (`src/example.py`) is provided to demonstrate programmatic control:
@@ -212,6 +312,7 @@ The example demonstrates:
 - **src/cpu_loader_core.c**: High-performance C implementation using pthreads for CPU load generation
 - **src/cpu_loader.py**: Python wrapper providing a clean API to the C extension
 - **src/main.py**: FastAPI application with REST API and embedded WebUI
+- **src/mqtt_publisher.py**: MQTT client for publishing metrics and settings
 - **Threading Model**: Native pthreads for maximum efficiency and precise timing
 - **Load Algorithm**: High-resolution busy-wait loops with nanosecond precision
 
@@ -221,6 +322,7 @@ The example demonstrates:
 - FastAPI 0.115.0+
 - Uvicorn 0.32.0+
 - Pydantic 2.10.0+
+- paho-mqtt 2.1.0+ (for MQTT publishing)
 - C compiler (gcc or clang) for building the extension
 
 ## Use Cases
@@ -230,6 +332,8 @@ The example demonstrates:
 - **Thermal Testing**: Check cooling system effectiveness
 - **Power Consumption Analysis**: Measure power usage at different load levels
 - **Benchmarking**: Create reproducible load scenarios
+- **IoT Integration**: Integrate with Home Assistant, Node-RED, or other MQTT-based systems
+- **Monitoring**: Feed CPU metrics into monitoring dashboards via MQTT
 
 ## Development
 

{cpu_loader-0.0.3 → cpu_loader-0.1.2}/src/cpu_loader.egg-info/SOURCES.txt

@@ -7,6 +7,7 @@ src/cpu_loader.py
 src/cpu_loader_core.c
 src/example.py
 src/main.py
+src/mqtt_publisher.py
 src/cpu_loader.egg-info/PKG-INFO
 src/cpu_loader.egg-info/SOURCES.txt
 src/cpu_loader.egg-info/dependency_links.txt

{cpu_loader-0.0.3 → cpu_loader-0.1.2}/src/cpu_loader.egg-info/requires.txt

@@ -4,3 +4,4 @@ pydantic==2.10.0
 requests==2.32.3
 psutil>=5.9.0
 websockets>=12.0
+paho-mqtt>=2.1.0

{cpu_loader-0.0.3 → cpu_loader-0.1.2}/src/cpu_loader.egg-info/top_level.txt

@@ -2,3 +2,4 @@ cpu_loader
 cpu_loader_core
 example
 main
+mqtt_publisher

{cpu_loader-0.0.3 → cpu_loader-0.1.2}/src/main.py

@@ -2,9 +2,11 @@
 CPU Loader FastAPI Application
 Provides REST API and WebUI for controlling CPU load.
 """
+import argparse
 import asyncio
+import logging
 from contextlib import asynccontextmanager
-from typing import Dict, List, Set
+from typing import Dict, List, Optional, Set
 
 import psutil
 import uvicorn
@@ -13,6 +15,13 @@ from fastapi.responses import HTMLResponse
 from pydantic import BaseModel, Field
 
 from cpu_loader import CPULoader
+from mqtt_publisher import MQTTPublisher
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+)
+logger = logging.getLogger(__name__)
 
 
 # Pydantic models for request/response validation
@@ -44,8 +53,9 @@ class CPUMetricsResponse(BaseModel):
     per_cpu_percent: List[float]
 
 
-# Global CPU loader instance and WebSocket connections
+# Global CPU loader instance, MQTT publisher, and WebSocket connections
 cpu_loader = None
+mqtt_publisher: Optional[MQTTPublisher] = None
 websocket_connections: Set[WebSocket] = set()
 monitoring_task = None
 
@@ -82,17 +92,34 @@ async def cpu_monitoring_loop():
 
                 # Remove disconnected clients
                 websocket_connections.difference_update(disconnected)
+
+            # Publish to MQTT if enabled
+            if mqtt_publisher:
+                mqtt_publisher.publish_cpu_metrics(total_cpu, per_cpu)
+
         except Exception as e:
-            print(f"Error in CPU monitoring loop: {e}")
+            logger.error(f"Error in CPU monitoring loop: {e}")
             await asyncio.sleep(1.0)
 
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """Lifespan context manager for startup and shutdown events."""
-    global cpu_loader, monitoring_task
+    global cpu_loader, mqtt_publisher, monitoring_task
     # Startup
     cpu_loader = CPULoader()
+
+    # Initialize MQTT publisher with settings from arguments or environment
+    mqtt_args = getattr(app.state, "mqtt_args", {})
+    try:
+        mqtt_publisher = MQTTPublisher(**mqtt_args)
+    except ImportError:
+        logger.warning("MQTT publishing disabled: paho-mqtt not installed")
+        mqtt_publisher = None
+    except Exception as e:
+        logger.error(f"Failed to initialize MQTT publisher: {e}")
+        mqtt_publisher = None
+
     monitoring_task = asyncio.create_task(cpu_monitoring_loop())
     yield
     # Shutdown
@@ -102,6 +129,8 @@ async def lifespan(app: FastAPI):
             await monitoring_task
         except asyncio.CancelledError:
             pass
+    if mqtt_publisher:
+        mqtt_publisher.disconnect()
     cpu_loader.shutdown()
 
 
@@ -615,6 +644,13 @@ async def set_thread_count(request: ThreadCountRequest):
     """Set the number of threads."""
     try:
         cpu_loader.set_num_threads(request.num_threads)
+
+        # Publish updated settings to MQTT
+        if mqtt_publisher:
+            mqtt_publisher.publish_load_settings(
+                cpu_loader.get_num_threads(), cpu_loader.get_all_loads()
+            )
+
         return {
             "status": "success",
             "num_threads": request.num_threads,
@@ -629,6 +665,13 @@ async def set_thread_load(thread_id: int, request: ThreadLoadRequest):
     """Set the CPU load for a specific thread."""
     try:
         cpu_loader.set_thread_load(thread_id, request.load_percent)
+
+        # Publish updated settings to MQTT
+        if mqtt_publisher:
+            mqtt_publisher.publish_load_settings(
+                cpu_loader.get_num_threads(), cpu_loader.get_all_loads()
+            )
+
         return {
             "status": "success",
             "thread_id": thread_id,
@@ -644,6 +687,13 @@ async def set_all_thread_loads(request: AllThreadsLoadRequest):
     """Set the same CPU load for all threads."""
     try:
         cpu_loader.set_all_loads(request.load_percent)
+
+        # Publish updated settings to MQTT
+        if mqtt_publisher:
+            mqtt_publisher.publish_load_settings(
+                cpu_loader.get_num_threads(), cpu_loader.get_all_loads()
+            )
+
         return {
             "status": "success",
             "load_percent": request.load_percent,
@@ -654,5 +704,74 @@ async def set_all_thread_loads(request: AllThreadsLoadRequest):
         raise HTTPException(status_code=400, detail=str(e))
 
 
+def parse_args():
+    """Parse command-line arguments."""
+    parser = argparse.ArgumentParser(
+        description="CPU Loader - Generate controllable CPU load with REST API and WebUI"
+    )
+    parser.add_argument(
+        "--host",
+        default="0.0.0.0",
+        help="Host to bind the server to (default: 0.0.0.0)",
+    )
+    parser.add_argument(
+        "--port",
+        type=int,
+        default=8000,
+        help="Port to bind the server to (default: 8000)",
+    )
+
+    # MQTT arguments
+    mqtt_group = parser.add_argument_group("MQTT settings")
+    mqtt_group.add_argument(
+        "--mqtt-broker-host",
+        help="MQTT broker hostname (env: MQTT_BROKER_HOST)",
+    )
+    mqtt_group.add_argument(
+        "--mqtt-broker-port",
+        type=int,
+        help="MQTT broker port (env: MQTT_BROKER_PORT, default: 1883)",
+    )
+    mqtt_group.add_argument(
+        "--mqtt-username",
+        help="MQTT username (env: MQTT_USERNAME)",
+    )
+    mqtt_group.add_argument(
+        "--mqtt-password",
+        help="MQTT password (env: MQTT_PASSWORD)",
+    )
+    mqtt_group.add_argument(
+        "--mqtt-topic-prefix",
+        help="MQTT topic prefix (env: MQTT_TOPIC_PREFIX, default: cpu-loader)",
+    )
+    mqtt_group.add_argument(
+        "--mqtt-client-id",
+        help="MQTT client ID (env: MQTT_CLIENT_ID, default: cpu-loader)",
+    )
+
+    return parser.parse_args()
+
+
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    args = parse_args()
+
+    # Prepare MQTT arguments (only non-None values)
+    mqtt_args = {}
+    if args.mqtt_broker_host:
+        mqtt_args["broker_host"] = args.mqtt_broker_host
+    if args.mqtt_broker_port:
+        mqtt_args["broker_port"] = args.mqtt_broker_port
+    if args.mqtt_username:
+        mqtt_args["username"] = args.mqtt_username
+    if args.mqtt_password:
+        mqtt_args["password"] = args.mqtt_password
+    if args.mqtt_topic_prefix:
+        mqtt_args["topic_prefix"] = args.mqtt_topic_prefix
+    if args.mqtt_client_id:
+        mqtt_args["client_id"] = args.mqtt_client_id
+
+    # Store MQTT args in app state for lifespan to access
+    app.state.mqtt_args = mqtt_args
+
+    # Run the server
+    uvicorn.run(app, host=args.host, port=args.port)

cpu_loader-0.1.2/src/mqtt_publisher.py

@@ -0,0 +1,189 @@
+"""
+MQTT Publisher Module
+Publishes CPU load control settings and metrics to MQTT broker.
+"""
+import json
+import logging
+import os
+from typing import Dict, Optional
+
+try:
+    import paho.mqtt.client as mqtt
+except ImportError:
+    mqtt = None
+
+logger = logging.getLogger(__name__)
+
+
+class MQTTPublisher:
+    """Handles publishing CPU load data to MQTT broker."""
+
+    def __init__(
+        self,
+        broker_host: Optional[str] = None,
+        broker_port: Optional[int] = None,
+        username: Optional[str] = None,
+        password: Optional[str] = None,
+        topic_prefix: Optional[str] = None,
+        client_id: Optional[str] = None,
+    ):
+        """
+        Initialize MQTT publisher.
+
+        Args:
+            broker_host: MQTT broker hostname (env: MQTT_BROKER_HOST)
+            broker_port: MQTT broker port (env: MQTT_BROKER_PORT, default: 1883)
+            username: MQTT username (env: MQTT_USERNAME)
+            password: MQTT password (env: MQTT_PASSWORD)
+            topic_prefix: Topic prefix (env: MQTT_TOPIC_PREFIX, default: cpu-loader)
+            client_id: MQTT client ID (env: MQTT_CLIENT_ID, default: cpu-loader)
+        """
+        if mqtt is None:
+            raise ImportError(
+                "paho-mqtt not installed. Install with: pip install paho-mqtt"
+            )
+
+        # Get settings from arguments or environment variables
+        self.broker_host = broker_host or os.getenv("MQTT_BROKER_HOST")
+        self.broker_port = int(
+            broker_port or int(os.getenv("MQTT_BROKER_PORT", "1883"))
+        )
+        self.username = username or os.getenv("MQTT_USERNAME")
+        self.password = password or os.getenv("MQTT_PASSWORD")
+        self.topic_prefix = topic_prefix or os.getenv("MQTT_TOPIC_PREFIX", "cpu-loader")
+        self.client_id = client_id or os.getenv("MQTT_CLIENT_ID", "cpu-loader")
+
+        self.client: Optional[mqtt.Client] = None
+        self.connected = False
+
+        # Only connect if broker host is provided
+        if self.broker_host:
+            self._connect()
+        else:
+            logger.info("MQTT broker host not configured, MQTT publishing disabled")
+
+    def _connect(self):
+        """Connect to MQTT broker."""
+        try:
+            # Create MQTT client
+            self.client = mqtt.Client(
+                client_id=self.client_id,
+                protocol=mqtt.MQTTv311,
+            )
+
+            # Set callbacks
+            self.client.on_connect = self._on_connect
+            self.client.on_disconnect = self._on_disconnect
+
+            # Set credentials if provided
+            if self.username:
+                self.client.username_pw_set(self.username, self.password)
+
+            # Connect to broker
+            logger.info(
+                f"Connecting to MQTT broker at {self.broker_host}:{self.broker_port}"
+            )
+            self.client.connect(self.broker_host, self.broker_port, keepalive=60)
+
+            # Start network loop in background thread
+            self.client.loop_start()
+
+        except Exception as e:
+            logger.error(f"Failed to connect to MQTT broker: {e}")
+            self.client = None
+
+    def _on_connect(self, client, userdata, flags, rc):
+        """Callback when connected to MQTT broker."""
+        if rc == 0:
+            self.connected = True
+            logger.info("Connected to MQTT broker")
+        else:
+            self.connected = False
+            logger.error(f"Failed to connect to MQTT broker with code: {rc}")
+
+    def _on_disconnect(self, client, userdata, rc):
+        """Callback when disconnected from MQTT broker."""
+        self.connected = False
+        if rc != 0:
+            logger.warning(f"Unexpected disconnect from MQTT broker: {rc}")
+        else:
+            logger.info("Disconnected from MQTT broker")
+
+    def publish_load_settings(self, num_threads: int, loads: Dict[int, float]):
+        """
+        Publish load control settings to MQTT.
+
+        Args:
+            num_threads: Number of active threads
+            loads: Dictionary mapping thread ID to load percentage
+        """
+        if not self.connected or not self.client:
+            return
+
+        try:
+            # Calculate average load
+            avg_load = sum(loads.values()) / len(loads) if loads else 0.0
+
+            # Prepare payload
+            payload = {
+                "num_threads": num_threads,
+                "loads": loads,
+                "average_load": round(avg_load, 2),
+            }
+
+            # Publish to topic
+            topic = f"{self.topic_prefix}/load_settings"
+            self.client.publish(
+                topic,
+                json.dumps(payload),
+                qos=1,
+                retain=True,
+            )
+            logger.debug(f"Published load settings to {topic}")
+
+        except Exception as e:
+            logger.error(f"Failed to publish load settings: {e}")
+
+    def publish_cpu_metrics(self, total_cpu_percent: float, per_cpu_percent: list):
+        """
+        Publish CPU metrics to MQTT.
+
+        Args:
+            total_cpu_percent: Total CPU utilization percentage
+            per_cpu_percent: List of per-CPU utilization percentages
+        """
+        if not self.connected or not self.client:
+            return
+
+        try:
+            # Prepare payload
+            payload = {
+                "total_cpu_percent": round(total_cpu_percent, 1),
+                "per_cpu_percent": [round(cpu, 1) for cpu in per_cpu_percent],
+            }
+
+            # Publish to topic
+            topic = f"{self.topic_prefix}/cpu_metrics"
+            self.client.publish(
+                topic,
+                json.dumps(payload),
+                qos=0,
+                retain=False,
+            )
+            logger.debug(f"Published CPU metrics to {topic}")
+
+        except Exception as e:
+            logger.error(f"Failed to publish CPU metrics: {e}")
+
+    def disconnect(self):
+        """Disconnect from MQTT broker."""
+        if self.client:
+            try:
+                self.client.loop_stop()
+                self.client.disconnect()
+                logger.info("Disconnected from MQTT broker")
+            except Exception as e:
+                logger.error(f"Error disconnecting from MQTT broker: {e}")
+            finally:
+                self.client = None
+                self.connected = False