redis與mongoDB使用python測試獲取數據差距

# -*- coding: utf-8 -*-
from common.common_setting import mongo_db_addr
from common.server_log.logger import start_twisted_logging, smart_log

import redis
import pymongo
import datetime

redis_count = 0
mongodb_count = 0

def connect_redis():
    return redis.StrictRedis(host='192.168.0.7', port=6379, db=0)

def redis_set():
    user_map = dict(userId=1,
                    userName='YourSis')
    connect_redis().set('userInfo', user_map)

def redis_get(connect, thread_id=1):
    global redis_count
    redis_count += 1
    user_info = connect.get('userInfo')
    smart_log("{},redis-{}: {}".format(thread_id, redis_count, user_info))


def start_redis_get():

    connect = connect_redis()

    current_time = datetime.datetime.now()
    minutes_test = current_time + datetime.timedelta(minutes=1)
    seconds_test = current_time + datetime.timedelta(seconds=1)

    while current_time < seconds_test:
    # while current_time < minutes_test :
        redis_get(connect)
        current_time = datetime.datetime.now()

def connect_mongodb():
    conn = pymongo.Connection(mongo_db_addr[0], mongo_db_addr[1])
    return conn.test

def mongodb_set():
    connect_mongodb().collection.insert({'userId': 1, 'userName': 'YourSis'})

def mongodb_get(connect, thread_id=1):
    global mongodb_count
    mongodb_count += 1
    user_info = connect.collection.find({'userId': 1})
    smart_log("{},mongodb-{}: {}".format(thread_id, mongodb_count, user_info[0]))

def start_mongodb_get():
    connect = connect_mongodb()

    current_time = datetime.datetime.now()
    minutes_test = current_time + datetime.timedelta(minutes=1)
    seconds_test = current_time + datetime.timedelta(seconds=1)

    while current_time < seconds_test:
    # while current_time < minutes_test :
        mongodb_get(connect)
        current_time = datetime.datetime.now()

if __name__ == '__main__':

    start_twisted_logging('robot_called_single_routine')

    # 查看 redis 獲取數
    # start_redis_get()

    # 查看 mongodb 獲取數
    start_mongodb_get()

這個本沒有什麼好測的,不過公司須要知道具體數據級,就寫了python

相關文章
相關標籤/搜索