diff --git a/README.md b/README.md index b6ddb826..533c8fff 100644 --- a/README.md +++ b/README.md @@ -2,34 +2,14 @@ ## Как запустить -Запускаем mongodb и приложение +[mongo-sharding](mongo-sharding/README.md) -```shell -docker compose up -d -``` +[mongo-sharding-repl](mongo-sharding-repl/README.md) -Заполняем mongodb данными - -```shell -./scripts/mongo-init.sh -``` +[sharding-repl-cache](sharding-repl-cache/README.md) ## Как проверить -### Если вы запускаете проект на локальной машине - Откройте в браузере http://localhost:8080 -### Если вы запускаете проект на предоставленной виртуальной машине - -Узнать белый ip виртуальной машины - -```shell -curl --silent http://ifconfig.me -``` - -Откройте в браузере http://:8080 - -## Доступные эндпоинты - -Список доступных эндпоинтов, swagger http://:8080/docs \ No newline at end of file +[Итоговая схема](diagram.drawio) \ No newline at end of file diff --git a/diagram.drawio b/diagram.drawio new file mode 100644 index 00000000..c55a7a88 --- /dev/null +++ b/diagram.drawio @@ -0,0 +1,225 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/mongo-sharding-repl/README.md b/mongo-sharding-repl/README.md new file mode 100644 index 00000000..80dea2df --- /dev/null +++ b/mongo-sharding-repl/README.md @@ -0,0 +1,45 @@ +# mongo-sharding-repl + +## Как запустить + +1. Запуск mongodb и приложения + +```shell + +docker compose up -d + +``` + +2. Инициализация + +```shell + +./scripts/init.sh + +``` + +3. Наполнение + +```shell + +./scripts/fill.sh + +``` + +## Как проверить + +1. Тест + +```shell + +./scripts/tests.sh + +``` + +2. FastAPI + +```shell + +http://localhost:8080/docs + +``` \ 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..6dde742d --- /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.0 +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..0150e15b --- /dev/null +++ b/mongo-sharding-repl/compose.yaml @@ -0,0 +1,135 @@ +name: mongo-sharding-repl +services: + + # api + pymongo_api: + image: kazhem/pymongo_api:1.0.0 + ports: + - "8080:8080" + environment: + MONGODB_URL: "mongodb://mongos_router:27020" + MONGODB_DATABASE_NAME: "somedb" + + # сервер конфигурации + configSrv: + image: mongo:latest + container_name: configSrv + restart: always + ports: + - "27017:27017" + volumes: + - config-data:/data/db + command: mongod --configsvr --replSet config-server --bind_ip_all --port 27017 + healthcheck: + test: ["CMD", "mongosh", "--port", "27017", "--eval", "db.adminCommand('ping')"] + interval: 5s + start_period: 10s + + # роутер + mongos_router: + image: mongo:latest + container_name: mongos_router + restart: always + ports: + - "27020:27020" + command: mongos --configdb config-server/configSrv:27017 --bind_ip_all --port 27020 + healthcheck: + test: ["CMD", "mongosh", "--port", "27020", "--eval", "db.adminCommand('ping')"] + interval: 5s + start_period: 10s + + # 1-й шард + shard1_1: + image: mongo:latest + container_name: shard1_1 + restart: always + ports: + - "27010:27010" + volumes: + - shard1_1:/data/db + command: mongod --shardsvr --replSet shard1 --bind_ip_all --port 27010 + healthcheck: + test: ["CMD", "mongosh", "--port", "27010", "--eval", "db.adminCommand('ping')"] + interval: 5s + start_period: 10s + + shard1_2: + image: mongo:latest + container_name: shard1_2 + restart: always + ports: + - "27011:27011" + volumes: + - shard1_2:/data/db + command: mongod --shardsvr --replSet shard1 --bind_ip_all --port 27011 + healthcheck: + test: ["CMD", "mongosh", "--port", "27011", "--eval", "db.adminCommand('ping')"] + interval: 5s + start_period: 10s + + shard1_3: + image: mongo:latest + container_name: shard1_3 + restart: always + ports: + - "27012:27012" + volumes: + - shard1_3:/data/db + command: mongod --shardsvr --replSet shard1 --bind_ip_all --port 27012 + healthcheck: + test: ["CMD", "mongosh", "--port", "27012", "--eval", "db.adminCommand('ping')"] + interval: 5s + start_period: 10s + + # 2-й шард + shard2_1: + image: mongo:latest + container_name: shard2_1 + restart: always + ports: + - "27013:27013" + volumes: + - shard2_1:/data/db + command: mongod --shardsvr --replSet shard2 --bind_ip_all --port 27013 + healthcheck: + test: ["CMD", "mongosh", "--port", "27013", "--eval", "db.adminCommand('ping')"] + interval: 5s + start_period: 10s + + shard2_2: + image: mongo:latest + container_name: shard2_2 + restart: always + ports: + - "27014:27014" + volumes: + - shard2_2:/data/db + command: mongod --shardsvr --replSet shard2 --bind_ip_all --port 27014 + healthcheck: + test: ["CMD", "mongosh", "--port", "27014", "--eval", "db.adminCommand('ping')"] + interval: 5s + start_period: 10s + + shard2_3: + image: mongo:latest + container_name: shard2_3 + restart: always + ports: + - "27015:27015" + volumes: + - shard2_3:/data/db + command: mongod --shardsvr --replSet shard2 --bind_ip_all --port 27015 + healthcheck: + test: ["CMD", "mongosh", "--port", "27015", "--eval", "db.adminCommand('ping')"] + interval: 5s + start_period: 10s + +volumes: + config-data: + config-data-configdb: + shard1_1: + shard1_2: + shard1_3: + shard2_1: + shard2_2: + shard2_3: \ No newline at end of file diff --git a/mongo-sharding-repl/scripts/fill.sh b/mongo-sharding-repl/scripts/fill.sh new file mode 100644 index 00000000..ff3c7984 --- /dev/null +++ b/mongo-sharding-repl/scripts/fill.sh @@ -0,0 +1,21 @@ +#!/bin/bash + +### +# Заполнение бд +### + +docker compose exec -T mongos_router mongosh --port 27020 --quiet <