Merge "Implement generic update-on-match feature"

1 files changed, 22 insertions, 0 deletions

diff --git a/oslo_db/sqlalchemy/session.py b/oslo_db/sqlalchemy/session.py
index 6a0355d..ce347ce 100644
--- a/oslo_db/sqlalchemy/session.py
+++ b/oslo_db/sqlalchemy/session.py
@@ -297,6 +297,7 @@ from oslo_db import exception
 from oslo_db import options
 from oslo_db.sqlalchemy import compat
 from oslo_db.sqlalchemy import exc_filters
+from oslo_db.sqlalchemy import update_match
 from oslo_db.sqlalchemy import utils
 
 LOG = logging.getLogger(__name__)
@@ -599,6 +600,27 @@ class Query(sqlalchemy.orm.query.Query):
                             'deleted_at': timeutils.utcnow()},
                            synchronize_session=synchronize_session)
 
+    def update_returning_pk(self, values, surrogate_key):
+        """Perform an UPDATE, returning the primary key of the matched row.
+
+        This is a method-version of
+        oslo_db.sqlalchemy.update_match.update_returning_pk(); see that
+        function for usage details.
+
+        """
+        return update_match.update_returning_pk(self, values, surrogate_key)
+
+    def update_on_match(self, specimen, surrogate_key, values, **kw):
+        """Emit an UPDATE statement matching the given specimen.
+
+        This is a method-version of
+        oslo_db.sqlalchemy.update_match.update_on_match(); see that function
+        for usage details.
+
+        """
+        return update_match.update_on_match(
+            self, specimen, surrogate_key, values, **kw)
+
 
 class Session(sqlalchemy.orm.session.Session):
     """Custom Session class to avoid SqlAlchemy Session monkey patching."""