<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> <html xmlns="http://www.w3.org/1999/xhtml"> <head> <meta http-equiv="Content-Type" content="text/html; charset=utf-8" /> <title> examples.performance.single_inserts — SQLAlchemy 1.2 Documentation </title> <!-- begin iterate through site-imported + sphinx environment css_files --> <link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" /> <link rel="stylesheet" href="../../../_static/docs.css" type="text/css" /> <link rel="stylesheet" href="../../../_static/changelog.css" type="text/css" /> <link rel="stylesheet" href="../../../_static/sphinx_paramlinks.css" type="text/css" /> <!-- end iterate through site-imported + sphinx environment css_files --> <!-- begin layout.mako headers --> <link rel="index" title="Index" href="../../../genindex.html" /> <link rel="search" title="Search" href="../../../search.html" /> <link rel="copyright" title="Copyright" href="../../../copyright.html" /> <link rel="top" title="SQLAlchemy 1.2 Documentation" href="../../../index.html" /> <link rel="up" title="Module code" href="../../index.html" /> <!-- end layout.mako headers --> </head> <body> <div id="docs-container"> <div id="docs-top-navigation-container" class="body-background"> <div id="docs-header"> <div id="docs-version-header"> Release: <span class="version-num">1.2.19</span> | Release Date: April 15, 2019 </div> <h1>SQLAlchemy 1.2 Documentation</h1> </div> </div> <div id="docs-body-container"> <div id="fixed-sidebar" class=""> <div id="index-nav"> <form class="search" action="../../../search.html" method="get"> <label> Search terms: <input type="text" placeholder="search..." name="q" size="12" /> </label> <input type="submit" value="Search" /> <input type="hidden" name="check_keywords" value="yes" /> <input type="hidden" name="area" value="default" /> </form> <p> <a href="../../../contents.html">Contents</a> | <a href="../../../genindex.html">Index</a> </p> </div> </div> <div id="docs-body" class="" > <h1>Source code for examples.performance.single_inserts</h1><div class="highlight"><pre> <span></span><span class="sd">"""In this series of tests, we're looking at a method that inserts a row</span> <span class="sd">within a distinct transaction, and afterwards returns to essentially a</span> <span class="sd">"closed" state. This would be analogous to an API call that starts up</span> <span class="sd">a database connection, inserts the row, commits and closes.</span> <span class="sd">"""</span> <span class="kn">from</span> <span class="nn">sqlalchemy</span> <span class="k">import</span> <span class="n">bindparam</span> <span class="kn">from</span> <span class="nn">sqlalchemy</span> <span class="k">import</span> <span class="n">Column</span> <span class="kn">from</span> <span class="nn">sqlalchemy</span> <span class="k">import</span> <span class="n">create_engine</span> <span class="kn">from</span> <span class="nn">sqlalchemy</span> <span class="k">import</span> <span class="n">Integer</span> <span class="kn">from</span> <span class="nn">sqlalchemy</span> <span class="k">import</span> <span class="n">pool</span> <span class="kn">from</span> <span class="nn">sqlalchemy</span> <span class="k">import</span> <span class="n">String</span> <span class="kn">from</span> <span class="nn">sqlalchemy.ext.declarative</span> <span class="k">import</span> <span class="n">declarative_base</span> <span class="kn">from</span> <span class="nn">sqlalchemy.orm</span> <span class="k">import</span> <span class="n">Session</span> <span class="kn">from</span> <span class="nn">.</span> <span class="k">import</span> <span class="n">Profiler</span> <span class="n">Base</span> <span class="o">=</span> <span class="n">declarative_base</span><span class="p">()</span> <span class="n">engine</span> <span class="o">=</span> <span class="kc">None</span> <span class="k">class</span> <span class="nc">Customer</span><span class="p">(</span><span class="n">Base</span><span class="p">):</span> <span class="n">__tablename__</span> <span class="o">=</span> <span class="s2">"customer"</span> <span class="nb">id</span> <span class="o">=</span> <span class="n">Column</span><span class="p">(</span><span class="n">Integer</span><span class="p">,</span> <span class="n">primary_key</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span> <span class="n">name</span> <span class="o">=</span> <span class="n">Column</span><span class="p">(</span><span class="n">String</span><span class="p">(</span><span class="mi">255</span><span class="p">))</span> <span class="n">description</span> <span class="o">=</span> <span class="n">Column</span><span class="p">(</span><span class="n">String</span><span class="p">(</span><span class="mi">255</span><span class="p">))</span> <span class="n">Profiler</span><span class="o">.</span><span class="n">init</span><span class="p">(</span><span class="s2">"single_inserts"</span><span class="p">,</span> <span class="n">num</span><span class="o">=</span><span class="mi">10000</span><span class="p">)</span> <span class="nd">@Profiler</span><span class="o">.</span><span class="n">setup</span> <span class="k">def</span> <span class="nf">setup_database</span><span class="p">(</span><span class="n">dburl</span><span class="p">,</span> <span class="n">echo</span><span class="p">,</span> <span class="n">num</span><span class="p">):</span> <span class="k">global</span> <span class="n">engine</span> <span class="n">engine</span> <span class="o">=</span> <span class="n">create_engine</span><span class="p">(</span><span class="n">dburl</span><span class="p">,</span> <span class="n">echo</span><span class="o">=</span><span class="n">echo</span><span class="p">)</span> <span class="k">if</span> <span class="n">engine</span><span class="o">.</span><span class="n">dialect</span><span class="o">.</span><span class="n">name</span> <span class="o">==</span> <span class="s2">"sqlite"</span><span class="p">:</span> <span class="n">engine</span><span class="o">.</span><span class="n">pool</span> <span class="o">=</span> <span class="n">pool</span><span class="o">.</span><span class="n">StaticPool</span><span class="p">(</span><span class="n">creator</span><span class="o">=</span><span class="n">engine</span><span class="o">.</span><span class="n">pool</span><span class="o">.</span><span class="n">_creator</span><span class="p">)</span> <span class="n">Base</span><span class="o">.</span><span class="n">metadata</span><span class="o">.</span><span class="n">drop_all</span><span class="p">(</span><span class="n">engine</span><span class="p">)</span> <span class="n">Base</span><span class="o">.</span><span class="n">metadata</span><span class="o">.</span><span class="n">create_all</span><span class="p">(</span><span class="n">engine</span><span class="p">)</span> <span class="nd">@Profiler</span><span class="o">.</span><span class="n">profile</span> <span class="k">def</span> <span class="nf">test_orm_commit</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="sd">"""Individual INSERT/COMMIT pairs via the ORM"""</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="n">session</span> <span class="o">=</span> <span class="n">Session</span><span class="p">(</span><span class="n">bind</span><span class="o">=</span><span class="n">engine</span><span class="p">)</span> <span class="n">session</span><span class="o">.</span><span class="n">add</span><span class="p">(</span> <span class="n">Customer</span><span class="p">(</span> <span class="n">name</span><span class="o">=</span><span class="s2">"customer name </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="n">description</span><span class="o">=</span><span class="s2">"customer description </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="p">)</span> <span class="p">)</span> <span class="n">session</span><span class="o">.</span><span class="n">commit</span><span class="p">()</span> <span class="nd">@Profiler</span><span class="o">.</span><span class="n">profile</span> <span class="k">def</span> <span class="nf">test_bulk_save</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="sd">"""Individual INSERT/COMMIT pairs using the "bulk" API """</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="n">session</span> <span class="o">=</span> <span class="n">Session</span><span class="p">(</span><span class="n">bind</span><span class="o">=</span><span class="n">engine</span><span class="p">)</span> <span class="n">session</span><span class="o">.</span><span class="n">bulk_save_objects</span><span class="p">(</span> <span class="p">[</span> <span class="n">Customer</span><span class="p">(</span> <span class="n">name</span><span class="o">=</span><span class="s2">"customer name </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="n">description</span><span class="o">=</span><span class="s2">"customer description </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="p">)</span> <span class="p">]</span> <span class="p">)</span> <span class="n">session</span><span class="o">.</span><span class="n">commit</span><span class="p">()</span> <span class="nd">@Profiler</span><span class="o">.</span><span class="n">profile</span> <span class="k">def</span> <span class="nf">test_bulk_insert_dictionaries</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="sd">"""Individual INSERT/COMMIT pairs using the "bulk" API with dictionaries"""</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="n">session</span> <span class="o">=</span> <span class="n">Session</span><span class="p">(</span><span class="n">bind</span><span class="o">=</span><span class="n">engine</span><span class="p">)</span> <span class="n">session</span><span class="o">.</span><span class="n">bulk_insert_mappings</span><span class="p">(</span> <span class="n">Customer</span><span class="p">,</span> <span class="p">[</span> <span class="nb">dict</span><span class="p">(</span> <span class="n">name</span><span class="o">=</span><span class="s2">"customer name </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="n">description</span><span class="o">=</span><span class="s2">"customer description </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="p">)</span> <span class="p">],</span> <span class="p">)</span> <span class="n">session</span><span class="o">.</span><span class="n">commit</span><span class="p">()</span> <span class="nd">@Profiler</span><span class="o">.</span><span class="n">profile</span> <span class="k">def</span> <span class="nf">test_core</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="sd">"""Individual INSERT/COMMIT pairs using Core."""</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="k">with</span> <span class="n">engine</span><span class="o">.</span><span class="n">begin</span><span class="p">()</span> <span class="k">as</span> <span class="n">conn</span><span class="p">:</span> <span class="n">conn</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span> <span class="n">Customer</span><span class="o">.</span><span class="n">__table__</span><span class="o">.</span><span class="n">insert</span><span class="p">(),</span> <span class="nb">dict</span><span class="p">(</span> <span class="n">name</span><span class="o">=</span><span class="s2">"customer name </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="n">description</span><span class="o">=</span><span class="s2">"customer description </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="p">),</span> <span class="p">)</span> <span class="nd">@Profiler</span><span class="o">.</span><span class="n">profile</span> <span class="k">def</span> <span class="nf">test_core_query_caching</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="sd">"""Individual INSERT/COMMIT pairs using Core with query caching"""</span> <span class="n">cache</span> <span class="o">=</span> <span class="p">{}</span> <span class="n">ins</span> <span class="o">=</span> <span class="n">Customer</span><span class="o">.</span><span class="n">__table__</span><span class="o">.</span><span class="n">insert</span><span class="p">()</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="k">with</span> <span class="n">engine</span><span class="o">.</span><span class="n">begin</span><span class="p">()</span> <span class="k">as</span> <span class="n">conn</span><span class="p">:</span> <span class="n">conn</span><span class="o">.</span><span class="n">execution_options</span><span class="p">(</span><span class="n">compiled_cache</span><span class="o">=</span><span class="n">cache</span><span class="p">)</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span> <span class="n">ins</span><span class="p">,</span> <span class="nb">dict</span><span class="p">(</span> <span class="n">name</span><span class="o">=</span><span class="s2">"customer name </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="n">description</span><span class="o">=</span><span class="s2">"customer description </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="p">),</span> <span class="p">)</span> <span class="nd">@Profiler</span><span class="o">.</span><span class="n">profile</span> <span class="k">def</span> <span class="nf">test_dbapi_raw_w_connect</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="sd">"""Individual INSERT/COMMIT pairs w/ DBAPI + connection each time"""</span> <span class="n">_test_dbapi_raw</span><span class="p">(</span><span class="n">n</span><span class="p">,</span> <span class="kc">True</span><span class="p">)</span> <span class="nd">@Profiler</span><span class="o">.</span><span class="n">profile</span> <span class="k">def</span> <span class="nf">test_dbapi_raw_w_pool</span><span class="p">(</span><span class="n">n</span><span class="p">):</span> <span class="sd">"""Individual INSERT/COMMIT pairs w/ DBAPI + connection pool"""</span> <span class="n">_test_dbapi_raw</span><span class="p">(</span><span class="n">n</span><span class="p">,</span> <span class="kc">False</span><span class="p">)</span> <span class="k">def</span> <span class="nf">_test_dbapi_raw</span><span class="p">(</span><span class="n">n</span><span class="p">,</span> <span class="n">connect</span><span class="p">):</span> <span class="n">compiled</span> <span class="o">=</span> <span class="p">(</span> <span class="n">Customer</span><span class="o">.</span><span class="n">__table__</span><span class="o">.</span><span class="n">insert</span><span class="p">()</span> <span class="o">.</span><span class="n">values</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="n">bindparam</span><span class="p">(</span><span class="s2">"name"</span><span class="p">),</span> <span class="n">description</span><span class="o">=</span><span class="n">bindparam</span><span class="p">(</span><span class="s2">"description"</span><span class="p">))</span> <span class="o">.</span><span class="n">compile</span><span class="p">(</span><span class="n">dialect</span><span class="o">=</span><span class="n">engine</span><span class="o">.</span><span class="n">dialect</span><span class="p">)</span> <span class="p">)</span> <span class="k">if</span> <span class="n">compiled</span><span class="o">.</span><span class="n">positional</span><span class="p">:</span> <span class="n">args</span> <span class="o">=</span> <span class="p">(</span> <span class="p">(</span><span class="s2">"customer name </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="s2">"customer description </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">)</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n</span><span class="p">)</span> <span class="p">)</span> <span class="k">else</span><span class="p">:</span> <span class="n">args</span> <span class="o">=</span> <span class="p">(</span> <span class="nb">dict</span><span class="p">(</span> <span class="n">name</span><span class="o">=</span><span class="s2">"customer name </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="n">description</span><span class="o">=</span><span class="s2">"customer description </span><span class="si">%d</span><span class="s2">"</span> <span class="o">%</span> <span class="n">i</span><span class="p">,</span> <span class="p">)</span> <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">n</span><span class="p">)</span> <span class="p">)</span> <span class="n">sql</span> <span class="o">=</span> <span class="nb">str</span><span class="p">(</span><span class="n">compiled</span><span class="p">)</span> <span class="k">if</span> <span class="n">connect</span><span class="p">:</span> <span class="k">for</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">args</span><span class="p">:</span> <span class="c1"># there's no connection pool, so if these were distinct</span> <span class="c1"># calls, we'd be connecting each time</span> <span class="n">conn</span> <span class="o">=</span> <span class="n">engine</span><span class="o">.</span><span class="n">pool</span><span class="o">.</span><span class="n">_creator</span><span class="p">()</span> <span class="n">cursor</span> <span class="o">=</span> <span class="n">conn</span><span class="o">.</span><span class="n">cursor</span><span class="p">()</span> <span class="n">cursor</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">sql</span><span class="p">,</span> <span class="n">arg</span><span class="p">)</span> <span class="n">lastrowid</span> <span class="o">=</span> <span class="n">cursor</span><span class="o">.</span><span class="n">lastrowid</span> <span class="n">conn</span><span class="o">.</span><span class="n">commit</span><span class="p">()</span> <span class="n">conn</span><span class="o">.</span><span class="n">close</span><span class="p">()</span> <span class="k">else</span><span class="p">:</span> <span class="k">for</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">args</span><span class="p">:</span> <span class="n">conn</span> <span class="o">=</span> <span class="n">engine</span><span class="o">.</span><span class="n">raw_connection</span><span class="p">()</span> <span class="n">cursor</span> <span class="o">=</span> <span class="n">conn</span><span class="o">.</span><span class="n">cursor</span><span class="p">()</span> <span class="n">cursor</span><span class="o">.</span><span class="n">execute</span><span class="p">(</span><span class="n">sql</span><span class="p">,</span> <span class="n">arg</span><span class="p">)</span> <span class="n">lastrowid</span> <span class="o">=</span> <span class="n">cursor</span><span class="o">.</span><span class="n">lastrowid</span> <span class="n">conn</span><span class="o">.</span><span class="n">commit</span><span class="p">()</span> <span class="n">conn</span><span class="o">.</span><span class="n">close</span><span class="p">()</span> <span class="k">if</span> <span class="vm">__name__</span> <span class="o">==</span> <span class="s2">"__main__"</span><span class="p">:</span> <span class="n">Profiler</span><span class="o">.</span><span class="n">main</span><span class="p">()</span></pre></div> </div> </div> <div id="docs-bottom-navigation" class="docs-navigation-links"> <div id="docs-copyright"> © <a href="../../../copyright.html">Copyright</a> 2007-2019, the SQLAlchemy authors and contributors. Created using <a href="http://sphinx.pocoo.org/">Sphinx</a> 2.0.1. </div> </div> </div> <script type="text/javascript"> var DOCUMENTATION_OPTIONS = { URL_ROOT: '../../../', VERSION: '1.2.19', COLLAPSE_MODINDEX: false, FILE_SUFFIX: '.html' }; </script> <script type="text/javascript" id="documentation_options" data-url_root="../../../" src="../../../_static/documentation_options.js"></script> <!-- begin iterate through sphinx environment script_files --> <script type="text/javascript" src="../../../_static/jquery.js"></script> <script type="text/javascript" src="../../../_static/underscore.js"></script> <script type="text/javascript" src="../../../_static/doctools.js"></script> <script type="text/javascript" src="../../../_static/language_data.js"></script> <!-- end iterate through sphinx environment script_files --> <script type="text/javascript" src="../../../_static/detectmobile.js"></script> <script type="text/javascript" src="../../../_static/init.js"></script> </body> </html>