import time
import requests

queries = [
	'{"query":{"bool":{"should":[{"match":{"_all":"filereader"}},{"match":{"extension":"java"}}]}}}',
	'{"query":{"bool":{"should":[{"match":{"code.class.name":"Stack"}},{"nested":{"path":"code.class.methods","query":{"bool":{"should":[{"match":{"code.class.methods.name":"push"}},{"term":{"code.class.methods.returntype":"void"}}]}}}},{"nested":{"path":"code.class.methods","query":{"bool":{"should":[{"match":{"code.class.methods.name":"pop"}},{"term":{"code.class.methods.returntype":"int"}}]}}}}]}}}',
	'{"query":{"bool":{"should":[{"match":{"files.code.class.name":"Export"}},{"match":{"files.code.class.extends":"WizardPage"}},{"match":{"files.code.imports":"eclipse"}}]}}}',
	'{"query":{"bool":{"should":[{"has_child":{"type":"files","query":{"match":{"code.class.implements":"Model"}}}},{"has_child":{"type":"files","query":{"match":{"code.class.implements":"View"}}}},{"has_child":{"type":"files","query":{"match":{"code.class.implements":"Controller"}}}},{"has_child":{"type":"files","query":{"match":{"code.class.extends":"JFrame"}}}}]}}}',
	'{"query":{"match":{"analyzedcontent":"File myfile = new File(\\"myfile.xml\\");\\nDocumentBuilderFactory dbFactory = DocumentBuilderFactory.newInstance();\\nDocumentBuilder dBuilder = dbFactory.newDocumentBuilder();\\nDocument doc = dBuilder.parse(myfile);"}}}'
]

repetitions = 5

total_time = 0
for i, query in enumerate(queries):
	query_times = []
	for repetition in range(repetitions):
		time_start = time.time()
		r = requests.post('http://agora.ee.auth.gr:8080/_search', data = query)
		if r.ok:
			time_elapsed = time.time() - time_start
			query_times.append(time_elapsed)
		else:
			print("Error on query %d" %(i+1))
			exit()
	avg_query_time = sum(query_times) / len(query_times)
	print("Average time per query for query %d: %.2f milliseconds" %(i + 1, avg_query_time))
	total_time += avg_query_time
print("Average time per query for all queries: %.2f milliseconds" %(total_time / len(queries)))

def add_project(self, project_address):
	"""
	Adds a project to the index or updates it if it already exists.
	
	:param project_address: the URL of the project to be added.
	"""
	project_id = '/'.join(project_address.split('/')[-2:])
	sys.stdout.write('\nDownloading project info for project ' + project_id)
	project, sourcefiles = self.gpdownloader.download_project(project_id)
	sys.stdout.write('. Done!\n')
	project_path = self.sourcecodedir + '/' + project['user'] + '/' + project['name']
	self.gitdownloader.git_pull_or_clone(project_id, project['git_url'], project_path, project['default_branch'])
	sys.stdout.write('Adding project to database!\n')
	sys.stdout.write('Compiling project')
	full_compiled_source = self.javaparser.parse_project(project_path)
	sys.stdout.write('. Done!\n')
	self.esclient.create_project(project)
	sys.stdout.write('Creating database entries')
	import time
	create_times = []
	for file_path, afile in self.get_enumerated_files_with_paths(project_path, sourcefiles):
		self.set_file_code_and_contents(file_path, afile, full_compiled_source)
		time_start = time.time()
		self.esclient.create_file(afile)
		create_times.append(time.time() - time_start)
	sys.stdout.write(' Done!\n')
	avg_create_time = sum(create_times) / len(create_times)
	sys.stdout.write("Average indexing time per file for project %s: %.3f milliseconds\n" %(project_id, avg_create_time))
	return avg_create_time

def add_projects(self, project_addresses):
	"""
	Adds or updates a list of projects in the index.
	
	:param project_addresses: a list of project addresses to be added.
	"""
	total_time = 0
	for project_address in project_addresses:
		avg_create_time = self.add_project(project_address)
		total_time += avg_create_time
	print("Average indexing time per file for all projects: %.3f milliseconds" %(total_time / len(project_addresses)))