mongodb生成数据-sharding测试准备

JS函数准备

注：准备几个js函数，为后续测试mongodb分片生成数据

主要生成个人姓名，性别，年龄，省份，手机号码等字段；

#性别

function getSex() {  
    var sex_array=["男","女"]
    var sex = sex_array[Math.floor(Math.random() *sex_array.length )];
    return sex;
}

#名字

function getName(sex){
    var first = ["赵",  "钱",  "孙",  "李",  "周",  "吴",  "郑",  "王",  "冯",  "陈" ];
    var male_last = [ "宇轩","梓轩","俊宇","梓睿","铭轩","梓豪","睿","伟","杰","军","强","勇","建华" ];
    var female_last = [ "芷晴","可昕","梓晴","秀英 ","桂英","秀兰","玥","悦","可","悠","钰","敏","艳","丽","静","玉兰" ];
    var last=[]
    if (sex=='男')
    {
        last=male_last 
    }else{
       last=female_last;
   };
    var name = first[Math.floor(Math.random()*first.length)];
    name+=last[Math.floor(Math.random()*last.length)];
    return name;
}

#手机号码

function getPhone() {
    const phonePrefixes = ['134', '135', '136', '137', '138', '139', '147', '150', '151', '152', '157', '158', '159', '165', '172', '178', '182', '183', '184', '187', '188', '198'];
    const prefix = phonePrefixes[Math.floor(Math.random() * phonePrefixes.length)];
    const middleDigits = Math.floor(Math.random() * (8999)) + 1000;
    const lastDigits = Math.floor(Math.random() * (8999)) + 1000;
    return `${prefix}${middleDigits}${lastDigits}`;
}

#省份

function getProvinces() {
    const provinces = [
        '广东', '北京', '上海', '江苏', '浙江', '四川',
        '河南', '湖南', '湖北', '江西', '安徽', '浙江',
        '山东', '河北', '山西', '内蒙古', '甘肃', '云南',
        '甘肃', '广西', '西藏', '甘肃', '甘肃', '甘肃'
    ];
    const index = Math.floor(Math.random() * provinces.length);
    return provinces[index];
}

#年龄

function getAge() {  
    var age =Math.floor(Math.random() *99+1 );
    return age;
}

连接mongos造数

连接mongos

mongo --port 27000 --host 127.0.0.1
use admin
db.auth("admin","Admin@01");

执行

use testdb;

生成数据

启用sharding

如：按_id来做分区键

sh.enableSharding("testdb")
sh.shardCollection("testdb.person",{_id:"hashed"})

生成1万条记录

for(var i=0; i<200; i++){ 
    var docs = new Array();
     for(var j=0; j<50; j++) {

                    var sex = getSex() ;
                    var name=getName(sex);                                   
                    var age =getAge(); 
                    var province=getProvinces();
                    var phonenumber=getPhone();
                    var id=ObjectId()
                    docs[j] = {"_id":id,"name":name,"sex":sex,"age":age,"province":province,"phonenumber":phonenumber};                            
 }          
db.person.insertMany(docs);                
}

load scripts

注：保存脚本到文件中，避免每次都需要手动输入

vi /home/mongo/testdb_func.js
function getSex() {  
    var sex_array=["男","女"]
    var sex = sex_array[Math.floor(Math.random() *sex_array.length )];
    return sex;
}
function getName(sex){
    var first = ["赵",  "钱",  "孙",  "李",  "周",  "吴",  "郑",  "王",  "冯",  "陈" ];
    var male_last = [ "宇轩","梓轩","俊宇","梓睿","铭轩","梓豪","睿","伟","杰","军","强","勇","建华" ];
    var female_last = [ "芷晴","可昕","梓晴","秀英 ","桂英","秀兰","玥","悦","可","悠","钰","敏","艳","丽","静","玉兰" ];
    var last=[]
    if (sex=='男')
    {
        last=male_last 
    }else{
       last=female_last;
   };
    var name = first[Math.floor(Math.random()*first.length)];
    name+=last[Math.floor(Math.random()*last.length)];
    return name;
}
function getPhone() {
    const phonePrefixes = ['134', '135', '136', '137', '138', '139', '147', '150', '151', '152', '157', '158', '159', '165', '172', '178', '182', '183', '184', '187', '188', '198'];
    const prefix = phonePrefixes[Math.floor(Math.random() * phonePrefixes.length)];
    const middleDigits = Math.floor(Math.random() * (8999)) + 1000;
    const lastDigits = Math.floor(Math.random() * (8999)) + 1000;
    return `${prefix}${middleDigits}${lastDigits}`;
}
function getProvinces() {
    const provinces = [
        '广东', '北京', '上海', '江苏', '浙江', '四川',
        '河南', '湖南', '湖北', '江西', '安徽', '浙江',
        '山东', '河北', '山西', '内蒙古', '甘肃', '云南',
        '甘肃', '广西', '西藏', '甘肃', '甘肃', '甘肃'
    ];
    const index = Math.floor(Math.random() * provinces.length);
    return provinces[index];
}
function getAge() {  
    var age =Math.floor(Math.random() *99+1 );
    return age;
}

for(var i=0; i<200; i++){ 
    var docs = new Array();
     for(var j=0; j<50; j++) {

                    var sex = getSex() ;
                    var name=getName(sex);                                   
                    var age =getAge(); 
                    var province=getProvinces();
                    var phonenumber=getPhone();
                    var id=ObjectId()
                    docs[j] = {"_id":id,"name":name,"sex":sex,"age":age,"province":province,"phonenumber":phonenumber};                            
 }          
db.person.insertMany(docs);                
}

加载脚本

use testdb
load("/home/mongo/testdb_func.js")