diff options
Diffstat (limited to 'docker/storperf-master/storperf/db/configuration_db.py')
-rw-r--r-- | docker/storperf-master/storperf/db/configuration_db.py | 120 |
1 files changed, 120 insertions, 0 deletions
diff --git a/docker/storperf-master/storperf/db/configuration_db.py b/docker/storperf-master/storperf/db/configuration_db.py new file mode 100644 index 0000000..5b996c7 --- /dev/null +++ b/docker/storperf-master/storperf/db/configuration_db.py @@ -0,0 +1,120 @@ +############################################################################## +# Copyright (c) 2015 EMC and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## + +from sqlite3 import OperationalError +from threading import Lock +import logging +import sqlite3 + +db_mutex = Lock() + + +class ConfigurationDB(object): + + db_name = "StorPerfConfig.db" + + def __init__(self): + """ + Creates the StorPerfConfig.db and configuration tables on demand + """ + + self.logger = logging.getLogger(__name__) + self.logger.debug("Connecting to " + ConfigurationDB.db_name) + with db_mutex: + db = sqlite3.connect(ConfigurationDB.db_name) + + cursor = db.cursor() + try: + cursor.execute('''CREATE TABLE configuration + (configuration_name text, + key text, + value text)''') + self.logger.debug("Created configuration table") + except OperationalError: + self.logger.debug("Configuration table exists") + + cursor.execute('SELECT * FROM configuration') + db.commit() + db.close() + + def delete_configuration_value(self, configuration_name, key): + """Deletes the value associated with the given key + """ + + with db_mutex: + db = sqlite3.connect(ConfigurationDB.db_name) + cursor = db.cursor() + + cursor.execute("delete from configuration where " + "configuration_name=? and key=?", + (configuration_name, key)) + + self.logger.debug("Deleted " + configuration_name + ":" + key) + + db.commit() + db.close() + + def get_configuration_value(self, configuration_name, key): + """Returns a string representation of the value stored + with this key under the given configuration name. + """ + + with db_mutex: + db = sqlite3.connect(ConfigurationDB.db_name) + cursor = db.cursor() + + cursor.execute( + """select value from configuration + where configuration_name = ? + and key = ?""", + (configuration_name, key,)) + + row = cursor.fetchone() + + return_value = None + + if (row is None): + self.logger.debug( + configuration_name + ":" + key + " does not exist") + else: + self.logger.debug( + configuration_name + ":" + key + " is " + str(row[0])) + return_value = str(row[0]) + + db.close() + + return return_value + + def set_configuration_value(self, configuration_name, key, value): + """Updates or creates the key under the given configuration + name so that it holds the value specified. + """ + + if (value is None): + return self.delete_configuration_value(configuration_name, key) + + with db_mutex: + value = str(value) + + db = sqlite3.connect(ConfigurationDB.db_name) + cursor = db.cursor() + + cursor.execute("delete from configuration where " + "configuration_name=? and key=?", + (configuration_name, key)) + + cursor.execute( + """insert into configuration(configuration_name, key, value) + values (?,?,?)""", (configuration_name, key, value)) + + self.logger.debug( + configuration_name + ":" + key + " set to " + value) + + db.commit() + db.close() |