diff --git a/mongo-sharding-repl/README.md b/mongo-sharding-repl/README.md new file mode 100644 index 00000000..cbb01603 --- /dev/null +++ b/mongo-sharding-repl/README.md @@ -0,0 +1,14 @@ +# Задание 2. Шардирование + +Запуск проекта и заполнение его тестовыми данными происходит автоматически скриптом ./setup_mongo_sharding-repl.sh. + +Для запуска скрипт нужно сделать исполняемым. В папке проекта выполните: + + chmod +x setup_mongo_sharding-repl.sh + +Затем запустите скрипт командой: + + ./setup_mongo_sharding-repl.sh + +Дождитесь сообщения "Настройка шардированного кластера с репликацией и загрузка данных завершены". +Теперь с проектом можно работать. \ No newline at end of file diff --git a/mongo-sharding-repl/api_app/Dockerfile b/mongo-sharding-repl/api_app/Dockerfile new file mode 100644 index 00000000..46f6c9d0 --- /dev/null +++ b/mongo-sharding-repl/api_app/Dockerfile @@ -0,0 +1,10 @@ +FROM python:3.12.1-slim +WORKDIR /app +EXPOSE 8080 +COPY requirements.txt ./ +# Устанавливаем зависимости python не пересобирая их +RUN pip install --no-cache --no-cache-dir -r requirements.txt +# Копирование кода приложения +COPY app.py /app/ +ENTRYPOINT ["uvicorn"] +CMD ["app:app", "--host", "0.0.0.0", "--port", "8080"] diff --git a/mongo-sharding-repl/api_app/app.py b/mongo-sharding-repl/api_app/app.py new file mode 100644 index 00000000..9b19c017 --- /dev/null +++ b/mongo-sharding-repl/api_app/app.py @@ -0,0 +1,192 @@ +import json +import logging +import os +import time +from typing import List, Optional + +import motor.motor_asyncio +from bson import ObjectId +from fastapi import Body, FastAPI, HTTPException, status +from fastapi_cache import FastAPICache +from fastapi_cache.backends.redis import RedisBackend +from fastapi_cache.decorator import cache +from logmiddleware import RouterLoggingMiddleware, logging_config +from pydantic import BaseModel, ConfigDict, EmailStr, Field +from pydantic.functional_validators import BeforeValidator +from pymongo import errors +from redis import asyncio as aioredis +from typing_extensions import Annotated + +# Configure JSON logging +logging.config.dictConfig(logging_config) +logger = logging.getLogger(__name__) + +app = FastAPI() +app.add_middleware( + RouterLoggingMiddleware, + logger=logger, +) + +DATABASE_URL = os.environ["MONGODB_URL"] +DATABASE_NAME = os.environ["MONGODB_DATABASE_NAME"] +REDIS_URL = os.getenv("REDIS_URL", None) + + +def nocache(*args, **kwargs): + def decorator(func): + return func + + return decorator + + +if REDIS_URL: + cache = cache +else: + cache = nocache + + +client = motor.motor_asyncio.AsyncIOMotorClient(DATABASE_URL) +db = client[DATABASE_NAME] + +# Represents an ObjectId field in the database. +# It will be represented as a `str` on the model so that it can be serialized to JSON. +PyObjectId = Annotated[str, BeforeValidator(str)] + + +@app.on_event("startup") +async def startup(): + if REDIS_URL: + redis = aioredis.from_url(REDIS_URL, encoding="utf8", decode_responses=True) + FastAPICache.init(RedisBackend(redis), prefix="api:cache") + + +class UserModel(BaseModel): + """ + Container for a single user record. + """ + + id: Optional[PyObjectId] = Field(alias="_id", default=None) + age: int = Field(...) + name: str = Field(...) + + +class UserCollection(BaseModel): + """ + A container holding a list of `UserModel` instances. + """ + + users: List[UserModel] + + +@app.get("/") +async def root(): + collection_names = await db.list_collection_names() + collections = {} + for collection_name in collection_names: + collection = db.get_collection(collection_name) + collections[collection_name] = { + "documents_count": await collection.count_documents({}) + } + try: + replica_status = await client.admin.command("replSetGetStatus") + replica_status = json.dumps(replica_status, indent=2, default=str) + except errors.OperationFailure: + replica_status = "No Replicas" + + topology_description = client.topology_description + read_preference = client.client_options.read_preference + topology_type = topology_description.topology_type_name + replicaset_name = topology_description.replica_set_name + + shards = None + if topology_type == "Sharded": + shards_list = await client.admin.command("listShards") + shards = {} + for shard in shards_list.get("shards", {}): + shards[shard["_id"]] = shard["host"] + + cache_enabled = False + if REDIS_URL: + cache_enabled = FastAPICache.get_enable() + + return { + "mongo_topology_type": topology_type, + "mongo_replicaset_name": replicaset_name, + "mongo_db": DATABASE_NAME, + "read_preference": str(read_preference), + "mongo_nodes": client.nodes, + "mongo_primary_host": client.primary, + "mongo_secondary_hosts": client.secondaries, + "mongo_address": client.address, + "mongo_is_primary": client.is_primary, + "mongo_is_mongos": client.is_mongos, + "collections": collections, + "shards": shards, + "cache_enabled": cache_enabled, + "status": "OK", + } + + +@app.get("/{collection_name}/count") +async def collection_count(collection_name: str): + collection = db.get_collection(collection_name) + items_count = await collection.count_documents({}) + # status = await client.admin.command('replSetGetStatus') + # import ipdb; ipdb.set_trace() + return {"status": "OK", "mongo_db": DATABASE_NAME, "items_count": items_count} + + +@app.get( + "/{collection_name}/users", + response_description="List all users", + response_model=UserCollection, + response_model_by_alias=False, +) +@cache(expire=60 * 1) +async def list_users(collection_name: str): + """ + List all of the user data in the database. + The response is unpaginated and limited to 1000 results. + """ + time.sleep(1) + collection = db.get_collection(collection_name) + return UserCollection(users=await collection.find().to_list(1000)) + + +@app.get( + "/{collection_name}/users/{name}", + response_description="Get a single user", + response_model=UserModel, + response_model_by_alias=False, +) +async def show_user(collection_name: str, name: str): + """ + Get the record for a specific user, looked up by `name`. + """ + + collection = db.get_collection(collection_name) + if (user := await collection.find_one({"name": name})) is not None: + return user + + raise HTTPException(status_code=404, detail=f"User {name} not found") + + +@app.post( + "/{collection_name}/users", + response_description="Add new user", + response_model=UserModel, + status_code=status.HTTP_201_CREATED, + response_model_by_alias=False, +) +async def create_user(collection_name: str, user: UserModel = Body(...)): + """ + Insert a new user record. + + A unique `id` will be created and provided in the response. + """ + collection = db.get_collection(collection_name) + new_user = await collection.insert_one( + user.model_dump(by_alias=True, exclude=["id"]) + ) + created_user = await collection.find_one({"_id": new_user.inserted_id}) + return created_user diff --git a/mongo-sharding-repl/api_app/requirements.txt b/mongo-sharding-repl/api_app/requirements.txt new file mode 100644 index 00000000..61b29b87 --- /dev/null +++ b/mongo-sharding-repl/api_app/requirements.txt @@ -0,0 +1,6 @@ +fastapi==0.110.2 +uvicorn[standard]==0.29.0 +motor==3.5.3 +redis==4.4.2 +fastapi-cache2==0.2.0 +logmiddleware==0.0.4 \ No newline at end of file diff --git a/mongo-sharding-repl/compose.yaml b/mongo-sharding-repl/compose.yaml new file mode 100644 index 00000000..d67d8b5b --- /dev/null +++ b/mongo-sharding-repl/compose.yaml @@ -0,0 +1,244 @@ +name: mongo-sharding-repl + +x-common: &commonSettings + image: mongo:latest + restart: always + networks: + app-network: + healthcheck: + interval: 5s + start_period: 10s + +services: + # app server + pymongo_api: + build: + context: api_app + dockerfile: Dockerfile + image: kazhem/pymongo_api:1.0.0 + depends_on: [mongos_router] + ports: + - 8080:8080 + networks: + app-network: + ipv4_address: 172.18.0.5 + environment: + MONGODB_URL: "mongodb://mongos_router:27020" + MONGODB_DATABASE_NAME: "somedb" + + # Config Servers Replica Set + config1: + <<: *commonSettings + hostname: config1 + ports: + - 27017:27017 + networks: + app-network: + ipv4_address: 172.18.0.10 + volumes: + - config1-data:/data/db + command: + [ + "mongod", + "--configsvr", + "--replSet", + "configReplSet", + "--bind_ip_all", + "--port", + "27017" + ] + + config2: + <<: *commonSettings + hostname: config2 + networks: + app-network: + ipv4_address: 172.18.0.11 + volumes: + - config2-data:/data/db + command: + [ + "mongod", + "--configsvr", + "--replSet", + "configReplSet", + "--bind_ip_all", + "--port", + "27017" + ] + + config3: + <<: *commonSettings + hostname: config3 + networks: + app-network: + ipv4_address: 172.18.0.12 + volumes: + - config3-data:/data/db + command: + [ + "mongod", + "--configsvr", + "--replSet", + "configReplSet", + "--bind_ip_all", + "--port", + "27017" + ] + + # mongos router + mongos_router: + <<: *commonSettings + hostname: mongos_router + ports: + - 27020:27020 + networks: + app-network: + ipv4_address: 172.18.0.7 + volumes: + - router-data:/data/db + command: + [ + "mongos", + "--configdb", + "configReplSet/config1:27017,config2:27017,config3:27017", + "--bind_ip_all", + "--port", + "27020" + ] + + # Shard 1 Replica Set + shard1a: + <<: *commonSettings + hostname: shard1a + networks: + app-network: + ipv4_address: 172.18.0.20 + volumes: + - shard1a-data:/data/db + command: + [ + "mongod", + "--shardsvr", + "--replSet", + "shard1ReplSet", + "--bind_ip_all", + "--port", + "27018" + ] + + shard1b: + <<: *commonSettings + hostname: shard1b + networks: + app-network: + ipv4_address: 172.18.0.21 + volumes: + - shard1b-data:/data/db + command: + [ + "mongod", + "--shardsvr", + "--replSet", + "shard1ReplSet", + "--bind_ip_all", + "--port", + "27018" + ] + + shard1c: + <<: *commonSettings + hostname: shard1c + networks: + app-network: + ipv4_address: 172.18.0.22 + volumes: + - shard1c-data:/data/db + command: + [ + "mongod", + "--shardsvr", + "--replSet", + "shard1ReplSet", + "--bind_ip_all", + "--port", + "27018" + ] + + # Shard 2 Replica Set + shard2a: + <<: *commonSettings + hostname: shard2a + networks: + app-network: + ipv4_address: 172.18.0.30 + volumes: + - shard2a-data:/data/db + command: + [ + "mongod", + "--shardsvr", + "--replSet", + "shard2ReplSet", + "--bind_ip_all", + "--port", + "27019" + ] + + shard2b: + <<: *commonSettings + hostname: shard2b + networks: + app-network: + ipv4_address: 172.18.0.31 + volumes: + - shard2b-data:/data/db + command: + [ + "mongod", + "--shardsvr", + "--replSet", + "shard2ReplSet", + "--bind_ip_all", + "--port", + "27019" + ] + + shard2c: + <<: *commonSettings + hostname: shard2c + networks: + app-network: + ipv4_address: 172.18.0.32 + volumes: + - shard2c-data:/data/db + command: + [ + "mongod", + "--shardsvr", + "--replSet", + "shard2ReplSet", + "--bind_ip_all", + "--port", + "27019" + ] + +networks: + app-network: + driver: bridge + ipam: + driver: default + config: + - subnet: 172.18.0.0/16 + +volumes: + config1-data: + config2-data: + config3-data: + router-data: + shard1a-data: + shard1b-data: + shard1c-data: + shard2a-data: + shard2b-data: + shard2c-data: diff --git a/mongo-sharding-repl/setup_mongo_sharding-repl.sh b/mongo-sharding-repl/setup_mongo_sharding-repl.sh new file mode 100755 index 00000000..60f16398 --- /dev/null +++ b/mongo-sharding-repl/setup_mongo_sharding-repl.sh @@ -0,0 +1,83 @@ +#!/bin/bash + +GREEN='\033[0;32m' +NC='\033[0m' + +PROJECT_NAME="mongo-sharding-repl" + +docker-compose -p $PROJECT_NAME down -v + +docker-compose -p $PROJECT_NAME up -d + +echo -e "${GREEN}Ожидание инициализации MongoDB контейнеров...${NC}" +sleep 20 + +echo -e "${GREEN}Инициализация реплицированного набора configReplSet...${NC}" +docker compose -p $PROJECT_NAME exec -T config1 mongosh --port 27017 --quiet < + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +