python 内存NoSQL数据库

python 内存NoSQL数据库

来自于网络，经过修改，秉承Open Source精神，回馈网络!

#!/usr/bin/python
#-*- coding: UTF-8 -*-
#
# memdb.py
#   python memory db
#
# 2015-12
########################################################################
# The MIT License (MIT)
#    http://opensource.org/licenses/MIT
#
#  Copyright (c) 2015 copyright cheungmine
#
# Permission is hereby granted, free of charge, to any person obtaining
#  a copy of this software and associated documentation files (the
#  "Software"), to deal in the Software without restriction, including
#  without limitation the rights to use, copy, modify, merge, publish,
#  distribute, sublicense, and/or sell copies of the Software, and to
#  permit persons to whom the Software is furnished to do so, subject
#  to the following conditions:
#
# The above copyright notice and this permission notice shall be
#  included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
# IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
# CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
# TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
########################################################################

from multiprocessing import RLock

# sync for threading
class ObjectSync:
    def __init__(self, name):
        self.refcount = 0
        self.synclock = RLock()
        self.keyname  = name

    def Lock(self):
        self.synclock.acquire()
        self.refcount = self.refcount + 1

    def Unlock(self):
        self.refcount = self.refcount - 1
        self.synclock.release()

class ObjectSyncFactory:
    def __init__(self):
        self.globalLock = ObjectSync("")
        self.rowlocks = {}

    def __RemoveLock(self, sem, keyname):
        self.globalLock.Lock()
        self.rowlocks[keyname] = None
        self.globalLock.Unlock()

    def GetLock(self, tablename, key):
        keyname = tablename + "," + str(key)
        self.globalLock.Lock()

        l = None
        try:
            l = self.rowlocks[keyname]
            if l == None:
                self.rowlocks[keyname] = ObjectSync(keyname)
                l = self.rowlocks[keyname]
        except:
            self.rowlocks[keyname] = ObjectSync(keyname)
            l = self.rowlocks[keyname]

        self.globalLock.Unlock()

        return l

class PairGuard:
    def __init__(self, factory, sem):
        self.syncfactory = factory
        self.host = sem
        self.host.Lock()

    def __del__(self):
        self.host.Unlock()
        if self.host.refcount == 0 :
            self.syncfactory._ObjectSyncFactory__RemoveLock(self.host, self.host.keyname)

########################################
# Database table
class MemTable:
    def __init__(self):
        self.rows = {}
        self.tableLock = ObjectSync("")

    def GetRowCount(self):
        return len(self.rows)

    def DeleteAll(self):
        self.tableLock.Lock()
        self.rows = {}
        self.tableLock.Unlock()

    def __DeleteAll(self):
        self.rows = {}

    def GetAllValue(self):
        return self.rows

    # throw KeyError if key not found.
    def GetValue(self, key):
        return self.rows[key]

    # not exist: Add
    # exist: Update
    def AddValue(self, key, value):
        self.tableLock.Lock()
        self.rows[key] = value
        self.tableLock.Unlock()

    def __AddValue(self, key, value):
        self.rows[key] = value

    def DelValue(self, key):
        self.AddValue(key,None)

    def __DelValue(self, key):
        self._MemTable__AddValue(key, None)

########################################
# MemDB
class MemDB:
    def __init__(self):
        self.tables = {}
        self.syncFactory = ObjectSyncFactory()

    # is not thread safed
    def CreateTable(self, tablename):
        self.tables[tablename] = MemTable()

    # is not thread safed
    def DropTable(self, tablename):
        self.tables[tablename] = None

    def GetValue(self, tablename, key):
        mt = self.tables[tablename]
        PairGuard(self.syncFactory, self.syncFactory.GetLock(tablename, key))
        return mt.GetValue(key)

    def AddValue(self, tablename, key, value):
        mt = self.tables[tablename]
        PairGuard(self.syncFactory, self.syncFactory.GetLock(tablename, key))
        mt.AddValue(key, value)

    def DelValue(self, tablename, key):
        mt = self.tables[tablename]
        PairGuard(self.syncFactory, self.syncFactory.GetLock(tablename, key))
        mt.DelValue(key)

    def __GetValue(self, tablename, key):
        mt = self.tables[tablename]
        return mt.GetValue(key)

    def __AddValue(self, tablename, key, value):
        mt = self.tables[tablename]
        mt._MemTable__AddValue(key, value)

    def __DelValue(self, tablename, key):
        mt = self.tables[tablename]
        mt._MemTable__DelValue(key)

class Transaction:
    def __init__(self, conn):
        self.dbconn = conn
        self.logs = []

    def Commit(self):
        syncs = []
        tables = {}

        for p in self.logs:
            tables[p[0]] = True

        for name in tables:
            syncTable = self.dbconn.memdb.syncFactory.GetLock(name, 'table')
            syncs.append( (syncTable.keyname, syncTable) )

        syncs.sort()

        #lock
        guards = []
        for sync in syncs:
            guards.append(PairGuard(self.dbconn.memdb.syncFactory, sync[1]))

        #commit
        self.logs.reverse()
        while True:
            if len(self.logs) == 0:
                break
            p = self.logs.pop()
            self.dbconn.memdb._MemDB__AddValue(p[0], p[1], p[2])

        #unlock
        guards.reverse()
        while True:
            if len(guards) == 0:
                break
            guards.pop()

        self.dbconn._MemDBConnect__EndTransaction()

    def Rollback(self):
        self.dbconn._MemDBConnect__EndTransaction()

    def LogPoint(self, tablename, key, value):
        self.logs.append((tablename, key, value))

class MemDBConnect:
    def __init__(self, db):
        self.memdb = db
        self.bTransaction = False
        self.trans = None

    def BeginTransaction(self):
        self.bTransaction = True
        self.trans = Transaction(self)
        return self.trans

    def __EndTransaction(self):
        self.bTransaction = False
        self.trans = None

    def CommitTransaction(self):
        if self.bTransaction:
            self.bTransaction = False
            ts = self.trans
            self.trans = None
            if ts:
                ts.Commit()

    def RollbackTransaction(self):
        if self.bTransaction:
            self.bTransaction = False
            ts = self.trans
            self.trans = None
            if ts:
                ts.Rollback()

    # not thread safe
    def CreateTable(self, tablename):
        self.memdb.CreateTable(tablename)

    # not thread safe
    def DropTable(self, tablename):
        self.memdb.DropTable(tablename)

    # not thread safe
    def HasTable(self, tablename):
        if self.memdb.tables.get(tablename):
            return True
        else:
            return False

    def GetValue(self, tablename, key):
        if self.bTransaction:
            return self.memdb._MemDB__GetValue(tablename, key)
        else:
            return self.memdb.GetValue(tablename, key)

    def AddValue(self, tablename, key, value):
        if self.bTransaction:
            self.trans.LogPoint(tablename, key, value)
        else:
            self.memdb.AddValue(tablename, key, value)

    def DelValue(self, tablename, key):
        if self.bTransaction:
            self.trans.LogPoint(tablename, key, None)
        else:
            self.memdb.DelValue(tablename, key)

    def QueryTablesNothrow(self):
        tables = []
        try:
            self.BeginTransaction()

            for tablename,_ in self.memdb.tables.items():
                tables.append(tablename)

            self.CommitTransaction()
        except:
            tables = []
            self.RollbackTransaction()
        finally:
            return tables

    def QueryTableKeysNothrow(self, tablename):
        keys = []
        try:
            self.BeginTransaction()

            if self.HasTable(tablename):
                rows_dict = self.memdb.tables[tablename].rows
                for key, _ in rows_dict.items():
                    keys.append(key)

            self.CommitTransaction()
        except:
            keys = []
            self.RollbackTransaction()
        finally:
            return keys

    def CreateTableNothrow(self, tablename):
        try:
            self.BeginTransaction()

            if not self.HasTable(tablename):
                self.memdb.CreateTable(tablename)
                self.CommitTransaction()
        except:
            self.RollbackTransaction()
        finally:
            pass

    def DropTableNothrow(self, tablename):
        try:
            self.BeginTransaction()

            if self.HasTable(tablename):
                self.memdb.DropTable(tablename)
                self.CommitTransaction()
        except:
            self.RollbackTransaction()
        finally:
            pass

    def GetValueNothrow(self, tablename, key, defaultvalue):
        result = defaultvalue
        try:
            self.BeginTransaction()

            result = self.GetValue(tablename, key)

            self.CommitTransaction()
        except:
            self.RollbackTransaction()
        finally:
            return result

    def AddValueNothrow(self, tablename, key, value):
        result = False
        try:
            self.BeginTransaction()

            self.AddValue(tablename, key, value)

            self.CommitTransaction()

            result = True
        except:
            self.RollbackTransaction()
        finally:
            return result

    def DelValueNothrow(self, tablename, key):
        result = False
        try:
            self.BeginTransaction()

            self.DelValue(tablename, key)

            self.CommitTransaction()

            result = True
        except:
            self.RollbackTransaction()
        finally:
            return result

    def AppendValueListNothrow(self, tablename, key, value, non_repeated_value):
        try:
            self.BeginTransaction()

            if self.HasTable(tablename):
                try:
                    values = self.GetValue(tablename, key)

                    if non_repeated_value:
                        if value not in values:
                            values.append(value)
                            self.AddValue(tablename, key, values)
                    else:
                        values.append(value)
                        self.AddValue(tablename, key, values)
                except KeyError:
                    self.AddValue(tablename, key, [value])
                finally:
                    self.CommitTransaction()
        except:
            self.RollbackTransaction()
        finally:
            pass

    def AppendValueListMultiNothrow(self, tablenames, keys, values, non_repeated_values):
        try:
            self.BeginTransaction()

            for i in range(0, len(tablenames)):
                t, k, v, nrv = tablenames[i], keys[i], values[i], non_repeated_values[i]

                if self.HasTable(t):
                    try:
                        vals = self.GetValue(t, k)

                        if nrv:
                            if v not in vals:
                                vals.append(v)
                                self.AddValue(t, k, vals)
                        else:
                            vals.append(v)
                            self.AddValue(t, k, vals)
                    except KeyError:
                        self.AddValue(t, k, [v])

            self.CommitTransaction()
        except:
            self.RollbackTransaction()
        finally:
            pass

用法1：

可以直接嵌入到python中：

def test():
    db = MemDB()
    tname = "table1"
    db.CreateTable(tname)

    #for i in range(100000):
    #    db.AddValue(tname,i,"sdfsd")
    db.AddValue(tname,11,"sdfsd")
    print db.GetValue(tname, 11)

    db.AddValue(tname,11,"dddddd")
    print db.GetValue(tname,11)

    db.AddValue(tname,12,"dsfdsfd")
    print db.GetValue(tname,12)

    conn = MemDBConnect(db)
    t = conn.BeginTransaction()
    for i in range(100000):
        conn.AddValue(tname,i,"sdfsd")
    conn.AddValue(tname,12,"sfdas")
    conn.AddValue(tname,12,"ddddd")
    t.Commit()
    print db.GetValue(tname,12)