#!/usr/bin/env python

import sys
import nmdb
from random import randint, choice


class Mismatch (Exception):
	pass


# network db
ndb = nmdb.DB()
ndb.add_tipc_server()
ndb.add_tcp_server('localhost')
ndb.add_udp_server('localhost')

# local db
ldb = {}

# history of each key
history = {}

# check decorator
def checked(f):
	def newf(k, *args, **kwargs):
		try:
			return f(k, *args, **kwargs)
		except:
			print history[k]
			raise
	newf.__name__ = f.__name__
	return newf


# operations
@checked
def set(k, v):
	ndb[k] = v
	ldb[k] = v
	if k not in history:
		history[k] = []
	history[k].append((set, k, v))

@checked
def get(k):
	n = ndb[k]
	l = ldb[k]
	if l != n:
		raise Mismatch, (n, l)
	history[k].append((get, k))
	return n

@checked
def delete(k):
	del ndb[k]
	del ldb[k]
	history[k].append((delete, k))

@checked
def cas(k, ov, nv):
	prel = ldb[k]
	pren = ndb[k]
	n = ndb.cas(k, ov, nv)
	if not ldb.has_key(k):
		l = 0
	elif ldb[k] == ov:
		ldb[k] = nv
		l = 2
	else:
		l = 1
	if n != l:
		print k, ldb[k], ndb[k]
		print prel, pren
		print history[k]
		raise Mismatch, (n, l)
	history[k].append((cas, k, ov, nv))
	return n


def check():
	for k in ldb.keys():
		try:
			n = ndb[k]
			l = ldb[k]
		except:
			print history[k]
			raise Mismatch, (n, l)

		if n != n:
			print history[k]
			raise Mismatch, (n, l)


# Use integers because the normal random() generates floating point numbers,
# and they can mess up comparisons because of architecture details.
def getrand():
	return randint(0, 1000000000000000000)


if __name__ == '__main__':
	if len(sys.argv) != 2:
		print 'Use: random1.py number_of_keys'
		sys.exit(1)

	nkeys = int(sys.argv[1])

	# fill all the keys
	print 'populate'
	for i in xrange(nkeys):
		set(getrand(), getrand())

	lkeys = ldb.keys()

	# operate on them a bit
	print 'random operations'
	operations = ('set', 'delete', 'cas0', 'cas1')
	for i in xrange(nkeys / 2):
		op = choice(operations)
		k = choice(lkeys)
		if op == 'set':
			set(k, getrand())
		elif op == 'delete':
			delete(k)
			lkeys.remove(k)
		elif op == 'cas0':
			# unsucessful cas
			cas(k, -1, -1)
		elif op == 'cas1':
			# successful cas
			cas(k, ldb[k], getrand())

	print 'check'
	check()

	print 'delete'
	for k in lkeys:
		delete(k)

	print 'check'
	check()

	sys.exit(0)