#!/usr/bin/python
"""
通过简单的数据读写方式，模拟十万级数据量的操作。在数据插入中途，通过混沌演练平台操作进行故障注入，观察前后变化。
本脚本只作为参考示例，真实演练时建议使用贴近生产环境的业务场景进行故障模拟。
"""
import pymongo
import random

# 演练时以实际注入故障的MongoDB实例Uri进行替换，具体可参考MongoDB [实例详情-网络配置-连接地址]
mongodbUri = 'mongodb://mongouser:thepasswordA1@ip:port/admin'
client = pymongo.MongoClient(mongodbUri)
# 数据库指定
db = client.somedb
# 数据集删除
db.user.drop()
# 自定义插入数据量。建议采用较大数据量，方便演练过程观察
element_num = 3 * 10 ** 6
for id in range(element_num):
    # 插入随机文档    
    name = random.choice(['R9', 'caÒt', 'owen', 'lee', 'J'])    
    sex = random.choice(['male', 'female'])    
    try:
        db.user.insert_one({'id': id, 'name': name, 'sex': sex})
    except Exception as e:
        print('error id', id)
# 查询全量文档
content = db.user.find()
for i in content:
    print(i)