#!/usr/bin/python
import pymongo
from pymongo import MongoClient, GEO2D
import time
import random
from bson.son import SON
import threading
from multiprocessing import Process
    

x_min_num=30
x_max_num=31

y_min_num=179
y_max_num=180

def mongodb_connect(): 
    connection = pymongo.Connection("192.168.0.102",27017)         
    #db = connection.admin 
    #db.authenticate("superuser","pwd") #用户认证
    db = connection.test 
    return db
    
    

def mongodb_save(collections,document):
     collections.insert(document)

'''
 db.users.ensureIndex({"gps":"2d"},{"min":-180,"max":180})
'''
def mongodb_createCollection():
    print "mongodb_createCollection"
    db  =  mongodb_connect() 
    my_collection = db.users
    my_collection.create_index([("gps", GEO2D)])
    #my_collection.create_index("mike")
   # my_collection.create_index([("mike", pymongo.DESCENDING), ("eliot", pymongo.ASCENDING)])
    #my_collection.ensure_index([("gps",pymongo.GEO2D),("min",-180),("max",180)],background=True)
    

  
def mongodb_destroyCollection():
    db  =  mongodb_connect()     
    my_collection = db.users    
    my_collection.drop()
    print "mongodb_destroyCollection"
    #my_collection.drop_index([("gps", GEO2D)])

    
def mongodb_insert_1w_data():    
    db  =  mongodb_connect()
    my_collection = db.users
    
    doc_list=list()
    for i in xrange(10000):
          doc=dict()
          x=random.uniform(x_min_num,x_max_num)
          y=random.uniform(y_min_num,y_max_num)
          doc["account"]=str(100+i)
          doc["gps"]=[  x,y ]        
          doc_list.append(doc)
          
    begin_time = time.time()
    my_collection.insert(doc_list)      
    end_time = time.time()
    print "my_collection.insert_many done:",end_time-begin_time 
  

def mongodb_findNear1km( location ):
    db  =  mongodb_connect()
    my_collection = db.users    
  
    
    begin_time = time.time()
    doc_list =  my_collection.find( {"gps": {"$near": location}} ).limit(100)  
    end_time = time.time()    
    
#    for doc in doc_list:
#        print "gps:",doc["gps"]
#        print "----------------"
#   
#    print "mongodb_findNear1km begin_time:",begin_time
   # print "mongodb_findNear1km end_time:",end_time
    #print "mongodb_findNear1km done:",end_time-begin_time ,begin_time,end_time    

    
class WorkerThread(threading.Thread):   
    def __init__(self,worker_index):
        threading.Thread.__init__(self)
        self.worker_index=worker_index
        
    def run(self):
         
         print "*"*20
         x=random.uniform(x_min_num,x_max_num)
         y=random.uniform(y_min_num,y_max_num)
         location= [x, y]         
         mongodb_findNear1km(location)
         print "worker:",self.worker_index,location
         
def mongodb_insert_some_data(how_many):            
    begin_time = time.time()
    for i in xrange(how_many):
        mongodb_insert_1w_data()
    end_time = time.time()   
    print "mongodb_insert_some_data done:",end_time-begin_time 
     
def  mongodb_test_concurrentQueryRequest(concurrent_num): 
    print "mongodb_test_concurrentQueryRequest:",  concurrent_num      
    for i in xrange(concurrent_num):
        worker = WorkerThread(i)
        worker.setDaemon(True)
        worker.start()   
     
def sleeper(name, seconds):  
    mongodb_test_concurrentQueryRequest(100)
    time.sleep(1)           
        
def testMore(num):
    process_list=list()
    for i in xrange(num):
        p = Process(target=sleeper, args=('bob', 5))  
        p.start()  
        process_list.append(p)
        #p.join()         
        
    for p in  process_list:
        p.join()
        
def mongodb_putdata():
    mongodb_destroyCollection()
    mongodb_createCollection()         
    mongodb_insert_some_data(10)  

def testone():
    x=random.uniform(x_min_num,x_max_num)
    y=random.uniform(y_min_num,y_max_num)
    location= [x, y]         
    mongodb_findNear1km(location)

def main():
    print "main"
    mongodb_putdata() 
    #testone()
    return
    #testMore(1)
    mongodb_test_concurrentQueryRequest(100)
    
    count=0
    while 1:
        time.sleep(1)
        count +=1
        #testone()
        print count
        if count > 100:
            print "count 100"
            break        
    
main()