diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2007-08-14 21:53:32 +0000 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2007-08-14 21:53:32 +0000 |
| commit | 087f235c33c1be4e0778231e8344a50dc4005c59 (patch) | |
| tree | d47c35d1e520e43c05ec869304870c0b6c87f736 /lib/sqlalchemy/engine | |
| parent | e58063aa91d893d76e9f34fbc3ea21818185844d (diff) | |
| download | sqlalchemy-087f235c33c1be4e0778231e8344a50dc4005c59.tar.gz | |
- merged "fasttypes" branch. this branch changes the signature
of convert_bind_param() and convert_result_value() to callable-returning
bind_processor() and result_processor() methods. if no callable is
returned, no pre/post processing function is called.
- hooks added throughout base/sql/defaults to optimize the calling
of bind param/result processors so that method call overhead is minimized.
special cases added for executemany() scenarios such that unneeded "last row id"
logic doesn't kick in, parameters aren't excessively traversed.
- new performance tests show a combined mass-insert/mass-select test as having 68%
fewer function calls than the same test run against 0.3.
- general performance improvement of result set iteration is around 10-20%.
Diffstat (limited to 'lib/sqlalchemy/engine')
| -rw-r--r-- | lib/sqlalchemy/engine/base.py | 24 | ||||
| -rw-r--r-- | lib/sqlalchemy/engine/default.py | 91 |
2 files changed, 64 insertions, 51 deletions
diff --git a/lib/sqlalchemy/engine/base.py b/lib/sqlalchemy/engine/base.py index 284be6dfe..8fe34bf3f 100644 --- a/lib/sqlalchemy/engine/base.py +++ b/lib/sqlalchemy/engine/base.py @@ -1127,20 +1127,17 @@ class ResultProxy(object): col3 = row[mytable.c.mycol] # access via Column object. ResultProxy also contains a map of TypeEngine objects and will - invoke the appropriate ``convert_result_value()`` method before + invoke the appropriate ``result_processor()`` method before returning columns, as well as the ExecutionContext corresponding to the statement execution. It provides several methods for which to obtain information from the underlying ExecutionContext. """ - class AmbiguousColumn(object): - def __init__(self, key): - self.key = key - def dialect_impl(self, dialect): - return self - def convert_result_value(self, arg, engine): - raise exceptions.InvalidRequestError("Ambiguous column name '%s' in result set! try 'use_labels' option on select statement." % (self.key)) - + def __ambiguous_processor(self, colname): + def process(value): + raise exceptions.InvalidRequestError("Ambiguous column name '%s' in result set! try 'use_labels' option on select statement." % colname) + return process + def __init__(self, context): """ResultProxy objects are constructed via the execute() method on SQLEngine.""" self.context = context @@ -1185,13 +1182,13 @@ class ResultProxy(object): else: type = typemap.get(item[1], types.NULLTYPE) - rec = (type, type.dialect_impl(self.dialect), i) + rec = (type, type.dialect_impl(self.dialect).result_processor(self.dialect), i) if rec[0] is None: raise exceptions.InvalidRequestError( "None for metadata " + colname) if self.__props.setdefault(colname.lower(), rec) is not rec: - self.__props[colname.lower()] = (type, ResultProxy.AmbiguousColumn(colname), 0) + self.__props[colname.lower()] = (type, self.__ambiguous_processor(colname), 0) self.__keys.append(colname) self.__props[i] = rec @@ -1298,7 +1295,10 @@ class ResultProxy(object): def _get_col(self, row, key): rec = self._key_cache[key] - return rec[1].convert_result_value(row[rec[2]], self.dialect) + if rec[1]: + return rec[1](row[rec[2]]) + else: + return row[rec[2]] def _fetchone_impl(self): return self.cursor.fetchone() diff --git a/lib/sqlalchemy/engine/default.py b/lib/sqlalchemy/engine/default.py index 02802452a..ccaf080e7 100644 --- a/lib/sqlalchemy/engine/default.py +++ b/lib/sqlalchemy/engine/default.py @@ -163,12 +163,17 @@ class DefaultExecutionContext(base.ExecutionContext): self.statement = unicode(compiled) if parameters is None: self.compiled_parameters = compiled.construct_params({}) + self.executemany = False elif not isinstance(parameters, (list, tuple)): self.compiled_parameters = compiled.construct_params(parameters) + self.executemany = False else: self.compiled_parameters = [compiled.construct_params(m or {}) for m in parameters] if len(self.compiled_parameters) == 1: self.compiled_parameters = self.compiled_parameters[0] + self.executemany = False + else: + self.executemany = True elif statement is not None: self.typemap = self.column_labels = None self.parameters = self.__encode_param_keys(parameters) @@ -206,22 +211,26 @@ class DefaultExecutionContext(base.ExecutionContext): return proc(params) def __convert_compiled_params(self, parameters): - executemany = parameters is not None and isinstance(parameters, list) encode = not self.dialect.supports_unicode_statements() # the bind params are a CompiledParams object. but all the DBAPI's hate # that object (or similar). so convert it to a clean # dictionary/list/tuple of dictionary/tuple of list if parameters is not None: - if self.dialect.positional: - if executemany: - parameters = [p.get_raw_list() for p in parameters] + if self.executemany: + processors = parameters[0].get_processors() + else: + processors = parameters.get_processors() + + if self.dialect.positional: + if self.executemany: + parameters = [p.get_raw_list(processors) for p in parameters] else: - parameters = parameters.get_raw_list() - else: - if executemany: - parameters = [p.get_raw_dict(encode_keys=encode) for p in parameters] + parameters = parameters.get_raw_list(processors) + else: + if self.executemany: + parameters = [p.get_raw_dict(processors, encode_keys=encode) for p in parameters] else: - parameters = parameters.get_raw_dict(encode_keys=encode) + parameters = parameters.get_raw_dict(processors, encode_keys=encode) return parameters def is_select(self): @@ -311,28 +320,31 @@ class DefaultExecutionContext(base.ExecutionContext): """generate default values for compiled insert/update statements, and generate last_inserted_ids() collection.""" - # TODO: cleanup if self.isinsert: - if isinstance(self.compiled_parameters, list): - plist = self.compiled_parameters - else: - plist = [self.compiled_parameters] drunner = self.dialect.defaultrunner(self) - for param in plist: + if self.executemany: + # executemany doesn't populate last_inserted_ids() + firstparam = self.compiled_parameters[0] + processors = firstparam.get_processors() + for c in self.compiled.statement.table.c: + if c.default is not None: + params = self.compiled_parameters + for param in params: + if not c.key in param or param.get_original(c.key) is None: + self.compiled_parameters = param + newid = drunner.get_column_default(c) + if newid is not None: + param.set_value(c.key, newid) + self.compiled_parameters = params + else: + param = self.compiled_parameters + processors = param.get_processors() last_inserted_ids = [] - # check the "default" status of each column in the table for c in self.compiled.statement.table.c: - # check if it will be populated by a SQL clause - we'll need that - # after execution. if c in self.compiled.inline_params: self._postfetch_cols.add(c) if c.primary_key: last_inserted_ids.append(None) - # check if its not present at all. see if theres a default - # and fire it off, and add to bind parameters. if - # its a pk, add the value to our last_inserted_ids list, - # or, if its a SQL-side default, let it fire off on the DB side, but we'll need - # the SQL-generated value after execution. elif not c.key in param or param.get_original(c.key) is None: if isinstance(c.default, schema.PassiveDefault): self._postfetch_cols.add(c) @@ -340,32 +352,33 @@ class DefaultExecutionContext(base.ExecutionContext): if newid is not None: param.set_value(c.key, newid) if c.primary_key: - last_inserted_ids.append(param.get_processed(c.key)) + last_inserted_ids.append(param.get_processed(c.key, processors)) elif c.primary_key: last_inserted_ids.append(None) - # its an explicitly passed pk value - add it to - # our last_inserted_ids list. elif c.primary_key: - last_inserted_ids.append(param.get_processed(c.key)) - # TODO: we arent accounting for executemany() situations - # here (hard to do since lastrowid doesnt support it either) + last_inserted_ids.append(param.get_processed(c.key, processors)) self._last_inserted_ids = last_inserted_ids self._last_inserted_params = param + + elif self.isupdate: - if isinstance(self.compiled_parameters, list): - plist = self.compiled_parameters - else: - plist = [self.compiled_parameters] drunner = self.dialect.defaultrunner(self) - for param in plist: - # check the "onupdate" status of each column in the table + if self.executemany: + for c in self.compiled.statement.table.c: + if c.onupdate is not None: + params = self.compiled_parameters + for param in params: + if not c.key in param or param.get_original(c.key) is None: + self.compiled_parameters = param + value = drunner.get_column_onupdate(c) + if value is not None: + param.set_value(c.key, value) + self.compiled_parameters = params + else: + param = self.compiled_parameters for c in self.compiled.statement.table.c: - # it will be populated by a SQL clause - we'll need that - # after execution. if c in self.compiled.inline_params: self._postfetch_cols.add(c) - # its not in the bind parameters, and theres an "onupdate" defined for the column; - # execute it and add to bind params elif c.onupdate is not None and (not c.key in param or param.get_original(c.key) is None): value = drunner.get_column_onupdate(c) if value is not None: |
