mirror of
https://github.com/knex/knex.git
synced 2025-12-16 09:38:25 +00:00
60 lines
14 KiB
HTML
60 lines
14 KiB
HTML
|
<!DOCTYPE html><html lang="en"><head><title>lib/dialects/postgres/runner</title></head><meta http-equiv="Content-Type" content="text/html; charset=utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0, minimum-scale=1.0, maximum-scale=1.0"><meta name="groc-relative-root" content="../../../"><meta name="groc-document-path" content="lib/dialects/postgres/runner"><meta name="groc-project-path" content="lib/dialects/postgres/runner.js"><link rel="stylesheet" type="text/css" media="all" href="../../../assets/style.css"><script type="text/javascript" src="../../../assets/behavior.js"></script><body><div id="meta"><div class="file-path">lib/dialects/postgres/runner.js</div></div><div id="document"><div class="segment"><div class="code"><div class="wrapper"><span class="nx">module</span><span class="p">.</span><span class="nx">exports</span> <span class="o">=</span> <span class="kd">function</span><span class="p">(</span><span class="nx">client</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
|
||
|
|
<span class="kd">var</span> <span class="nx">_</span> <span class="o">=</span> <span class="nx">require</span><span class="p">(</span><span class="s1">'lodash'</span><span class="p">);</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">inherits</span> <span class="o">=</span> <span class="nx">require</span><span class="p">(</span><span class="s1">'inherits'</span><span class="p">);</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">Promise</span> <span class="o">=</span> <span class="nx">require</span><span class="p">(</span><span class="s1">'../../promise'</span><span class="p">);</span>
|
||
|
|
|
||
|
|
<span class="kd">var</span> <span class="nx">Runner</span> <span class="o">=</span> <span class="nx">require</span><span class="p">(</span><span class="s1">'../../runner'</span><span class="p">);</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">utils</span> <span class="o">=</span> <span class="nx">require</span><span class="p">(</span><span class="s1">'../../utils'</span><span class="p">);</span></div></div></div><div class="segment"><div class="comments "><div class="wrapper"><p>Inherit from the <code>Runner</code> constructor's prototype,
|
||
|
|
so we can add the correct <code>then</code> method.</p></div></div><div class="code"><div class="wrapper"><span class="kd">function</span> <span class="nx">Runner_PG</span><span class="p">()</span> <span class="p">{</span>
|
||
|
|
<span class="k">this</span><span class="p">.</span><span class="nx">client</span> <span class="o">=</span> <span class="nx">client</span><span class="p">;</span>
|
||
|
|
<span class="nx">Runner</span><span class="p">.</span><span class="nx">apply</span><span class="p">(</span><span class="k">this</span><span class="p">,</span> <span class="nx">arguments</span><span class="p">);</span>
|
||
|
|
<span class="p">}</span>
|
||
|
|
<span class="nx">inherits</span><span class="p">(</span><span class="nx">Runner_PG</span><span class="p">,</span> <span class="nx">Runner</span><span class="p">);</span>
|
||
|
|
|
||
|
|
<span class="kd">var</span> <span class="nx">PGQueryStream</span><span class="p">;</span>
|
||
|
|
<span class="nx">Runner_PG</span><span class="p">.</span><span class="nx">prototype</span><span class="p">.</span><span class="nx">_stream</span> <span class="o">=</span> <span class="nx">Promise</span><span class="p">.</span><span class="nx">method</span><span class="p">(</span><span class="kd">function</span><span class="p">(</span><span class="nx">sql</span><span class="p">,</span> <span class="nx">stream</span><span class="p">,</span> <span class="nx">options</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="nx">PGQueryStream</span> <span class="o">=</span> <span class="nx">PGQueryStream</span> <span class="o">||</span> <span class="nx">require</span><span class="p">(</span><span class="s1">'pg-query-stream'</span><span class="p">);</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">runner</span> <span class="o">=</span> <span class="k">this</span><span class="p">;</span>
|
||
|
|
<span class="k">return</span> <span class="k">new</span> <span class="nx">Promise</span><span class="p">(</span><span class="kd">function</span><span class="p">(</span><span class="nx">resolver</span><span class="p">,</span> <span class="nx">rejecter</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="nx">stream</span><span class="p">.</span><span class="nx">on</span><span class="p">(</span><span class="s1">'error'</span><span class="p">,</span> <span class="nx">rejecter</span><span class="p">);</span>
|
||
|
|
<span class="nx">stream</span><span class="p">.</span><span class="nx">on</span><span class="p">(</span><span class="s1">'end'</span><span class="p">,</span> <span class="nx">resolver</span><span class="p">);</span>
|
||
|
|
<span class="nx">runner</span><span class="p">.</span><span class="nx">connection</span><span class="p">.</span><span class="nx">query</span><span class="p">(</span><span class="k">new</span> <span class="nx">PGQueryStream</span><span class="p">(</span><span class="nx">sql</span><span class="p">.</span><span class="nx">sql</span><span class="p">,</span> <span class="nx">sql</span><span class="p">.</span><span class="nx">bindings</span><span class="p">,</span> <span class="nx">options</span><span class="p">)).</span><span class="nx">pipe</span><span class="p">(</span><span class="nx">stream</span><span class="p">);</span>
|
||
|
|
<span class="p">});</span>
|
||
|
|
<span class="p">});</span></div></div></div><div class="segment"><div class="comments "><div class="wrapper"><p>Runs the query on the specified connection, providing the bindings
|
||
|
|
and any other necessary prep work.</p></div></div><div class="code"><div class="wrapper"><span class="nx">Runner_PG</span><span class="p">.</span><span class="nx">prototype</span><span class="p">.</span><span class="nx">_query</span> <span class="o">=</span> <span class="nx">Promise</span><span class="p">.</span><span class="nx">method</span><span class="p">(</span><span class="kd">function</span><span class="p">(</span><span class="nx">obj</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">connection</span> <span class="o">=</span> <span class="k">this</span><span class="p">.</span><span class="nx">connection</span><span class="p">;</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">sql</span> <span class="o">=</span> <span class="nx">obj</span><span class="p">.</span><span class="nx">sql</span> <span class="o">=</span> <span class="nx">utils</span><span class="p">.</span><span class="nx">pgBindings</span><span class="p">(</span><span class="nx">obj</span><span class="p">.</span><span class="nx">sql</span><span class="p">);</span>
|
||
|
|
<span class="k">if</span> <span class="p">(</span><span class="k">this</span><span class="p">.</span><span class="nx">isDebugging</span><span class="p">())</span> <span class="k">this</span><span class="p">.</span><span class="nx">debug</span><span class="p">(</span><span class="nx">obj</span><span class="p">);</span>
|
||
|
|
<span class="k">if</span> <span class="p">(</span><span class="nx">obj</span><span class="p">.</span><span class="nx">options</span><span class="p">)</span> <span class="nx">sql</span> <span class="o">=</span> <span class="nx">_</span><span class="p">.</span><span class="nx">extend</span><span class="p">({</span><span class="nx">text</span><span class="o">:</span> <span class="nx">sql</span><span class="p">},</span> <span class="nx">obj</span><span class="p">.</span><span class="nx">options</span><span class="p">);</span>
|
||
|
|
<span class="k">return</span> <span class="k">new</span> <span class="nx">Promise</span><span class="p">(</span><span class="kd">function</span><span class="p">(</span><span class="nx">resolver</span><span class="p">,</span> <span class="nx">rejecter</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="nx">connection</span><span class="p">.</span><span class="nx">query</span><span class="p">(</span><span class="nx">sql</span><span class="p">,</span> <span class="nx">obj</span><span class="p">.</span><span class="nx">bindings</span><span class="p">,</span> <span class="kd">function</span><span class="p">(</span><span class="nx">err</span><span class="p">,</span> <span class="nx">response</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="k">if</span> <span class="p">(</span><span class="nx">err</span><span class="p">)</span> <span class="k">return</span> <span class="nx">rejecter</span><span class="p">(</span><span class="nx">err</span><span class="p">);</span>
|
||
|
|
<span class="nx">obj</span><span class="p">.</span><span class="nx">response</span> <span class="o">=</span> <span class="nx">response</span><span class="p">;</span>
|
||
|
|
<span class="nx">resolver</span><span class="p">(</span><span class="nx">obj</span><span class="p">);</span>
|
||
|
|
<span class="p">});</span>
|
||
|
|
<span class="p">});</span>
|
||
|
|
<span class="p">});</span></div></div></div><div class="segment"><div class="comments "><div class="wrapper"><p>Ensures the response is returned in the same format as other clients.</p></div></div><div class="code"><div class="wrapper"><span class="nx">Runner_PG</span><span class="p">.</span><span class="nx">prototype</span><span class="p">.</span><span class="nx">processResponse</span> <span class="o">=</span> <span class="kd">function</span><span class="p">(</span><span class="nx">obj</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="k">if</span> <span class="p">(</span><span class="nx">obj</span><span class="p">.</span><span class="nx">output</span><span class="p">)</span> <span class="k">return</span> <span class="nx">obj</span><span class="p">.</span><span class="nx">output</span><span class="p">.</span><span class="nx">call</span><span class="p">(</span><span class="k">this</span><span class="p">,</span> <span class="nx">obj</span><span class="p">.</span><span class="nx">response</span><span class="p">);</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">resp</span> <span class="o">=</span> <span class="nx">obj</span><span class="p">.</span><span class="nx">response</span><span class="p">;</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">returning</span> <span class="o">=</span> <span class="nx">obj</span><span class="p">.</span><span class="nx">returning</span><span class="p">;</span>
|
||
|
|
<span class="k">if</span> <span class="p">(</span><span class="nx">resp</span><span class="p">.</span><span class="nx">command</span> <span class="o">===</span> <span class="s1">'SELECT'</span><span class="p">)</span> <span class="k">return</span> <span class="nx">resp</span><span class="p">.</span><span class="nx">rows</span><span class="p">;</span>
|
||
|
|
<span class="k">if</span> <span class="p">(</span><span class="nx">returning</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">returns</span> <span class="o">=</span> <span class="p">[];</span>
|
||
|
|
<span class="k">for</span> <span class="p">(</span><span class="kd">var</span> <span class="nx">i</span> <span class="o">=</span> <span class="mi">0</span><span class="p">,</span> <span class="nx">l</span> <span class="o">=</span> <span class="nx">resp</span><span class="p">.</span><span class="nx">rows</span><span class="p">.</span><span class="nx">length</span><span class="p">;</span> <span class="nx">i</span> <span class="o"><</span> <span class="nx">l</span><span class="p">;</span> <span class="nx">i</span><span class="o">++</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="kd">var</span> <span class="nx">row</span> <span class="o">=</span> <span class="nx">resp</span><span class="p">.</span><span class="nx">rows</span><span class="p">[</span><span class="nx">i</span><span class="p">];</span>
|
||
|
|
<span class="k">if</span> <span class="p">(</span><span class="nx">returning</span> <span class="o">===</span> <span class="s1">'*'</span> <span class="o">||</span> <span class="nx">_</span><span class="p">.</span><span class="nx">isArray</span><span class="p">(</span><span class="nx">returning</span><span class="p">))</span> <span class="p">{</span>
|
||
|
|
<span class="nx">returns</span><span class="p">[</span><span class="nx">i</span><span class="p">]</span> <span class="o">=</span> <span class="nx">row</span><span class="p">;</span>
|
||
|
|
<span class="p">}</span> <span class="k">else</span> <span class="p">{</span>
|
||
|
|
<span class="nx">returns</span><span class="p">[</span><span class="nx">i</span><span class="p">]</span> <span class="o">=</span> <span class="nx">row</span><span class="p">[</span><span class="nx">returning</span><span class="p">];</span>
|
||
|
|
<span class="p">}</span>
|
||
|
|
<span class="p">}</span>
|
||
|
|
<span class="k">return</span> <span class="nx">returns</span><span class="p">;</span>
|
||
|
|
<span class="p">}</span>
|
||
|
|
<span class="k">if</span> <span class="p">(</span><span class="nx">resp</span><span class="p">.</span><span class="nx">command</span> <span class="o">===</span> <span class="s1">'UPDATE'</span> <span class="o">||</span> <span class="nx">resp</span><span class="p">.</span><span class="nx">command</span> <span class="o">===</span> <span class="s1">'DELETE'</span><span class="p">)</span> <span class="p">{</span>
|
||
|
|
<span class="k">return</span> <span class="nx">resp</span><span class="p">.</span><span class="nx">rowCount</span><span class="p">;</span>
|
||
|
|
<span class="p">}</span>
|
||
|
|
<span class="k">return</span> <span class="nx">resp</span><span class="p">;</span>
|
||
|
|
<span class="p">};</span></div></div></div><div class="segment"><div class="comments "><div class="wrapper"><p>Assign the newly extended <code>Runner</code> constructor to the client object.</p></div></div><div class="code"><div class="wrapper"><span class="nx">client</span><span class="p">.</span><span class="nx">Runner</span> <span class="o">=</span> <span class="nx">Runner_PG</span><span class="p">;</span>
|
||
|
|
|
||
|
|
<span class="p">};</span></div></div></div></div></body></html>
|