def main():
db = Database("/tmp/tut_update")
db.create()
x_ind = WithXIndex(db.path, "x")
db.add_index(x_ind)
# full examples so we had to add first the data
# the same code as in previous step
for x in xrange(100):
db.insert(dict(x=x))
for y in xrange(100):
db.insert(dict(y=y))
# end of insert part
print(db.count(db.all, "x"))
for curr in db.all("x", with_doc=True):
doc = curr["doc"]
if curr["key"] % 7 == 0:
db.delete(doc)
elif curr["key"] % 5 == 0:
doc["updated"] = True
db.update(doc)
print(db.count(db.all, "x"))
for curr in db.all("x", with_doc=True):
print(curr)
def test_to_many_shards(self, tmpdir):
db = Database(str(tmpdir) + "/db")
db.create(with_id_index=False)
# it's ok to use sharded directly there
with pytest.raises(IndexPreconditionsException):
db.add_index(ShardedUniqueHashIndex(db.path, "id", sh_nums=300))
with pytest.raises(IndexPreconditionsException):
db.add_index(ShardedUniqueHashIndex(db.path, "id", sh_nums=256))
def test_compact_shards(self, tmpdir):
db = Database(str(tmpdir) + "/db")
db.create(with_id_index=False)
db.add_index(ShardedUniqueHashIndex5(db.path, "id"))
for x in range(100):
db.insert({"x": x})
db.compact()
assert db.count(db.all, "id") == 100
def main():
db = Database("/tmp/tut5_2")
db.create()
x_ind = WithXIndex(db.path, "x")
db.add_index(x_ind)
for x in xrange(100):
db.insert(dict(x=x, t=random.random()))
print(db.run("x", "avg", start=10, end=30))
def test_insert_get(self, tmpdir, sh_nums):
db = Database(str(tmpdir) + "/db")
db.create(with_id_index=False)
n = globals()["ShardedUniqueHashIndex%d" % sh_nums]
db.add_index(n(db.path, "id"))
l = []
for x in range(10000):
l.append(db.insert(dict(x=x))["_id"])
for curr in l:
assert db.get("id", curr)["_id"] == curr
def main():
db = Database("/tmp/tut2")
db.create()
x_ind = WithXIndex(db.path, "x")
db.add_index(x_ind)
for x in xrange(100):
db.insert(dict(x=x))
for y in xrange(100):
db.insert(dict(y=y))
print(db.get("x", 10, with_doc=True))
def main():
db = Database("/tmp/tut5_1")
db.create()
x_ind = WithXIndex(db.path, "x")
db.add_index(x_ind)
for x in xrange(100):
db.insert(dict(x=x, t=random.random()))
l = []
for curr in db.get_many("x", start=10, end=30, limit=-1, with_doc=True):
l.append(curr["doc"]["t"])
print(sum(l) / len(l))
def main():
db = Database("/tmp/tut4")
db.create()
x_ind = WithXIndex(db.path, "x")
db.add_index(x_ind)
for x in xrange(100):
db.insert(dict(x=x))
for y in xrange(100):
db.insert(dict(y=y))
print(db.get("x", 10, with_doc=True))
for curr in db.get_many("x", start=15, end=25, limit=-1, with_doc=True):
print(curr)
def test_num_shards(self, tmpdir, sh_nums):
db = Database(str(tmpdir) + "/db")
db.create(with_id_index=False)
n = globals()["ShardedUniqueHashIndex%d" % sh_nums]
db.add_index(n(db.path, "id"))
assert db.id_ind.sh_nums == sh_nums
def test_create(self, tmpdir):
db = Database(str(tmpdir) + "/db")
db.create(with_id_index=False)
db.add_index(ShardedUniqueHashIndex(db.path, "id", sh_nums=3))
class MySharded(ShardedHashIndex):
custom_header = "from codernitydb3.sharded_hash import ShardedHashIndex"
def __init__(self, *args, **kwargs):
kwargs["sh_nums"] = 10
kwargs["key_format"] = "I"
kwargs["use_make_keys"] = True
super(MySharded, self).__init__(*args, **kwargs)
def make_key_value(self, data):
return data["x"], None
def calculate_shard(self, key):
return key % self.sh_nums
y = 1500 * "y"
db = Database("/tmp/shard")
db.create(with_id_index=False)
db.add_index(CustomIdSharded(db.path, "id"))
db.add_index(MySharded(db.path, "x"))
# it makes non sense to use sharding with such small number of records
for x in xrange(10**4):
db.insert({"x": x, "y": y})
print(db.get("x", random.randint(0, 10**4))["_id"])