From 3a481bbf2ec89662016c2a0cabb396a4c083bf27 Mon Sep 17 00:00:00 2001
From: Jessica Lucci <jessicalucci14@gmail.com>
Date: Sun, 19 May 2013 18:51:40 -0500
Subject: [PATCH] Implemented basic sqlalchemy session class

---
 taskflow/common/.config.py.swp    | Bin 0 -> 12288 bytes
 taskflow/common/__init__.py       |  19 +++++++
 taskflow/common/config.py         |  49 ++++++++++++++++++
 taskflow/db/sqlalchemy/api.py     |   5 ++
 taskflow/db/sqlalchemy/session.py |  83 ++++++++++++++++++++++++++++++
 5 files changed, 156 insertions(+)
 create mode 100644 taskflow/common/.config.py.swp
 create mode 100644 taskflow/common/__init__.py
 create mode 100644 taskflow/common/config.py

diff --git a/taskflow/common/.config.py.swp b/taskflow/common/.config.py.swp
new file mode 100644
index 0000000000000000000000000000000000000000..528e55ba62dcb44e4aece9551cdfdc264bd795a7
GIT binary patch
literal 12288
zcmeI2&ub(_6vr##!C$+|uHg0U1e`=BGn2_`mV|(_`LR1OnUI-evq2=(bl3Eh+tbzQ
zs?O|yD2s@ASJ|6*_2@xCJ^3%_!JEew5B>oP>bIt6k|9|_z>7*XeCZ!m_3FLPt5+~g
zwaRyDn>1gXW4K;m?8S$lZh!FW=93?;G4{65`ug1L4cm0ywOOvbI=i}Fy|sT{FUH;-
z87!?RFYSuR=+X|?_trwyD;Yer?SrSfT@~RLwPjF@`{&zqt&V^ra8Ux6*j#0{Y<o}U
zuF=&iyB8hk`Z)rQfFs}tI0BA<Bj5-)0*=7{fPjf7*n5cMiD4|W!~4Xk`|#D>905nb
z5pV<?0Y|_Qa0DCyN5Bzq1RMcJ;6F%!`;5&!#n{!S;XMBTAN~IS<$1;)LEl2(Ko6iL
zXc7ADGGm`Z??Vru4QL&jfWExMm=C=K{q-DUe?q@QKS4i2KR{nYUqN3$A3^KTAI~!O
zGxQzw8FUxggKj}DLyw+e>>+dr?LwEK@8R<&(8tg_&|PQ`T7+yoS0Hye0*-(q;0QPZ
zj({U@0Rn`}YX`;Ut@>Iq5rNc3BnK)sddd@_Ncw~4Y37SI9XF4)OieE^c8(Fh^*95=
zug8O}f$=M!%|Klo%H)Bd!&IaqKm8UP?}X=$2;-&vPDjwDiUPHAlW>wU5}uJNB5e{G
z1(a&xlTo7<`)DvAf7%uz<Y{P@@;HzgD8IE(8XVJHu;(;i50%Hmj?!kKGGCsZKR&>j
zFAU6Bf=!cb#b#>g`oZS*u<ew~G>Waoy?Y|^^V6A$v2G_(HMcjaq1O@J{>c4*6j)y$
zhK~>H0s)QuEwHil(-OUz<^PQTMs|ft4YfpDC4!_M3c~zjfoS4rNCb}sFq}#5C?RvN
zmCGlXYmpgXhN%N<^I$m7S=xD@8{Xm?3+LaC*ama5auLd~Kr+1TM0H8&P!+Rt!X&$~
zN{p>r#&WqFYk;$igeH)-d~udI9jyJ)f#X3hHfeK|M!rY}6|f;a5ouA(z7O99CTxA^
zSa`B6Jqmdgq_zkHbtIC=mQ*Z~uGAW95h;v(3T0OsZi`N5#`da>Mzy|ETW!)-gRr-(
z)OKoHb-b-nwZ2bp)aok*5)u<7B<{xv>YU&^0t=<^i_{dt`j}bsu_-08l^PWC8TUzQ
z8t;I@=Mm+qO{z6>bhFy5H4C&?+qu29yOWvQjY2zJA`VlTz+|gWJjT2pZ()gsyk|Y(
zK_aq!Lq^m~B=`me(yDEGShQ?1Gc{QB9b;ltRj=18@|dGqiz*39uxIJ&(nf80wccEv
zsT9kjVH`<{`=r|Ac_3~wqmeRp(?T0UYfJbf?t3CO1W#~Y9G>#X6VwB}@$p64S1ENl
z?6wS-QD$tI)^Tvvfs-_BDbO8}*i;f6q$!*F+^{J(y-5Ac=E(D9_?^h0V`yr5nkwbl
zIci`Z(=aLMHugumJZh0wq-q#a!}iqF5LzThNNeWO85<Sa=N+Z4oih+4k7Rd&P|(__
mc<DOnjzlh`Z#wn`c{y&vd0(1&Wd^K#c(XvMY0uoSU)bN&<Y*ZH

literal 0
HcmV?d00001

diff --git a/taskflow/common/__init__.py b/taskflow/common/__init__.py
new file mode 100644
index 00000000..2309e169
--- /dev/null
+++ b/taskflow/common/__init__.py
@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+#    Copyright (C) 2012 Yahoo! Inc. All Rights Reserved.
+#    Copyright (C) 2013 Rackspace Hosting Inc. All Rights Reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
diff --git a/taskflow/common/config.py b/taskflow/common/config.py
new file mode 100644
index 00000000..27d2cd3c
--- /dev/null
+++ b/taskflow/common/config.py
@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+#    Copyright (C) 2012 Yahoo! Inc. All Rights Reserved.
+#    Copyright (C) 2013 Rackspace Hosting Inc. All Rights Reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+"""
+Routines for configuring TaskFlow
+"""
+
+from oslo.config import cfg
+
+db_opts = [
+    cfg.StrOpt('sql_connection',
+               default='mysql://task:flow@localhost/taskflow',
+               help='The SQLAlchemy connection string used to connect to the '
+               'database'),
+    cfg.IntOpt('sql_idle_timeout',
+               default=3600,
+               help='timeout before idle sql connections are reaped')]
+
+celery_opts = [
+    cfg.StrOpt('celery_backend',
+               default='mysql://task:flow@localhost/taskflow',
+               help='The SQLAlchemy connection string used to connect to the '
+                    'celery backend'),
+    cfg.StrOpt('celery_MQ',
+               default='mongodb://task:flow@localhost:27017/taskflow',
+               help='The MongoDB connection string used to connect to the '
+                    'celery message queue')]
+
+def register_db_opts():
+    cfg.CONF.register_opts(db_opts)
+
+def register_celery_opts():
+    cfg.CONF.register_opts(celery_opts)
diff --git a/taskflow/db/sqlalchemy/api.py b/taskflow/db/sqlalchemy/api.py
index 275e7e13..837e0711 100644
--- a/taskflow/db/sqlalchemy/api.py
+++ b/taskflow/db/sqlalchemy/api.py
@@ -16,3 +16,8 @@
 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 #    License for the specific language governing permissions and limitations
 #    under the License.
+
+
+import logging
+
+LOG = logging.getLogger(__name__)
diff --git a/taskflow/db/sqlalchemy/session.py b/taskflow/db/sqlalchemy/session.py
index 275e7e13..35e71a8c 100644
--- a/taskflow/db/sqlalchemy/session.py
+++ b/taskflow/db/sqlalchemy/session.py
@@ -16,3 +16,86 @@
 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 #    License for the specific language governing permissions and limitations
 #    under the License.
+
+"""Session Handling for SQLAlchemy backend."""
+
+import logging
+import sqlalchemy
+
+from taskflow.db import api as db_api
+
+LOG = logging.getLogger(__name__)
+
+_ENGINE = None
+_MAKER = None
+
+
+def get_session(autocommit=True, expire_on_commit=True):
+    """Return a SQLAlchemy session."""
+    global _MAKER
+
+    if _MAKER is None:
+        _MAKER = get_maker(get_engine(), autocommit, expire_on_commit)
+    return _MAKER()
+
+
+def synchronous_switch_listener(dbapi_conn, connection_rec):
+    """Switch sqlite connections to non-synchronous mode"""
+    dbapi_conn.execute("PRAGMA synchronous = OFF")
+
+
+def ping_listener(dbapi_conn, connection_rec, connection_proxy):
+    """
+Ensures that MySQL connections checked out of the
+pool are alive.
+
+Borrowed from:
+http://groups.google.com/group/sqlalchemy/msg/a4ce563d802c929f
+"""
+    try:
+        dbapi_conn.cursor().execute('select 1')
+    except dbapi_conn.OperationalError, ex:
+        if ex.args[0] in (2006, 2013, 2014, 2045, 2055):
+            LOG.warn(_('Got mysql server has gone away: %s'), ex)
+            raise DisconnectionError("Database server went away")
+        else:
+            raise
+
+
+def get_engine():
+    """Return a SQLAlchemy engine."""
+    global _ENGINE
+    if _ENGINE is None:
+        connection_dict = sqlalchemy.engine.url.make_url(_get_sql_connection())
+        engine_args = {
+            "pool_recycle": _get_sql_idle_timeout(),
+            "echo": False,
+            "convert_unicode": True
+        }
+
+    if "sqlite" in connection_dict.drivername:
+        engine_args['poolclass'] = NullPool
+
+    if "mysql" in connection_dict.drivername:
+        engine_args['listeners'] = [MySQLPingListener()]
+
+    _ENGINE = sqlqlchemy.create_engine(_get_sql_connection(),
+                                       **engine_args)
+
+    #TODO: Check to make sure engine connected
+
+    return _ENGINE
+
+
+def get_maker(engine, autocommit=True, expire_on_commit=False):
+    "Return a SQLAlchemy sessionmaker using the given engine."""
+    return sqlalchemy.orm.sessionmaker(bind=engine,
+                                       autocommit=autocommit,
+                                       expire_on_commit=expire_on_commit)
+
+def _get_sql_connection():
+    return db_api.SQL_CONNECTION
+
+
+def _get_sql_idle_timeout():
+    return db_api.SQL_IDLE_TIMEOUT