tmud-2.10.1/benchmark/
tmud-2.10.1/cmd/
tmud-2.10.1/cmd/objects/
tmud-2.10.1/cmd/tiny/
tmud-2.10.1/doc/classes/SQLite/
tmud-2.10.1/doc/classes/SQLite3/
tmud-2.10.1/doc/classes/TernaryTrie/
tmud-2.10.1/doc/files/cmd/objects/
tmud-2.10.1/doc/files/cmd/tiny/
tmud-2.10.1/doc/files/lib/
tmud-2.10.1/doc/files/lib/engine/
tmud-2.10.1/doc/files/lib/farts/
tmud-2.10.1/farts/
tmud-2.10.1/lib/
tmud-2.10.1/lib/core/
tmud-2.10.1/lib/engine/
tmud-2.10.1/lib/farts/
tmud-2.10.1/logs/
<?xml version="1.0" encoding="iso-8859-1"?>
<!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" xml:lang="en" lang="en">
<head>
  <title>Class: Loader</title>
  <meta http-equiv="Content-Type" content="text/html; charset=iso-8859-1" />
  <meta http-equiv="Content-Script-Type" content="text/javascript" />
  <link rel="stylesheet" href=".././rdoc-style.css" type="text/css" media="screen" />
  <script type="text/javascript">
  // <![CDATA[

  function popupCode( url ) {
    window.open(url, "Code", "resizable=yes,scrollbars=yes,toolbar=no,status=no,height=150,width=400")
  }

  function toggleCode( id ) {
    if ( document.getElementById )
      elem = document.getElementById( id );
    else if ( document.all )
      elem = eval( "document.all." + id );
    else
      return false;

    elemStyle = elem.style;
    
    if ( elemStyle.display != "block" ) {
      elemStyle.display = "block"
    } else {
      elemStyle.display = "none"
    }

    return true;
  }
  
  // Make codeblocks hidden by default
  document.writeln( "<style type=\"text/css\">div.method-source-code { display: none }</style>" )
  
  // ]]>
  </script>

</head>
<body>



    <div id="classHeader">
        <table class="header-table">
        <tr class="top-aligned-row">
          <td><strong>Class</strong></td>
          <td class="class-name-in-header">Loader</td>
        </tr>
        <tr class="top-aligned-row">
            <td><strong>In:</strong></td>
            <td>
                <a href="../files/dbload_rb.html">
                dbload.rb
                </a>
        <br />
            </td>
        </tr>

        <tr class="top-aligned-row">
            <td><strong>Parent:</strong></td>
            <td>
                Object
            </td>
        </tr>
        </table>
    </div>
  <!-- banner header -->

  <div id="bodyContent">



  <div id="contextContent">
    <div id="diagram">
      <map id="map" name="map">
  <area shape="rect" coords="5,5,77,53"      href="Loader.html" alt="Loader
" />
</map>
<img src="../dot/f_5.png" usemap="#map" border="0" alt="dot/f_5.png">
    </div>

    <div id="description">
      <p>
This utility program loads a yaml file to a database
</p>

    </div>


   </div>

    <div id="method-list">
      <h3 class="section-bar">Methods</h3>

      <div class="name-list">
      <a href="#M000036">get_options</a>&nbsp;&nbsp;
      <a href="#M000035">new</a>&nbsp;&nbsp;
      <a href="#M000037">run</a>&nbsp;&nbsp;
      </div>
    </div>

  </div>


    <!-- if includes -->

    <div id="section">


    <div id="constants-list">
      <h3 class="section-bar">Constants</h3>

      <div class="name-list">
        <table summary="Constants">
        <tr class="top-aligned-row context-row">
          <td class="context-item-name">VERSION</td>
          <td>=</td>
          <td class="context-item-value">&quot;0.2.0&quot;</td>
        </tr>
        <tr class="top-aligned-row context-row">
          <td class="context-item-name">DATABASES</td>
          <td>=</td>
          <td class="context-item-value">[:dbm, :gdbm, :sdbm, :sqlite, :sqlite3]</td>
        </tr>
        </table>
      </div>
    </div>



    <div id="attribute-list">
      <h3 class="section-bar">Attributes</h3>

      <div class="name-list">
        <table>
        <tr class="top-aligned-row context-row">
          <td class="context-item-name">opts</td>
          <td class="context-item-value">&nbsp;[RW]&nbsp;</td>
          <td class="context-item-desc"></td>
        </tr>
        </table>
      </div>
    </div>
      


    <!-- if method_list -->
    <div id="methods">
      <h3 class="section-bar">Public Class methods</h3>

      <div id="method-M000035" class="method-detail">
        <a name="M000035"></a>

        <div class="method-heading">
          <a href="#M000035" class="method-signature">
          <span class="method-name">new</span><span class="method-args">()</span>
          </a>
        </div>
      
        <div class="method-description">
          <p><a class="source-toggle" href="#"
            onclick="toggleCode('M000035-source');return false;">[Source]</a></p>
          <div class="method-source-code" id="M000035-source">
<pre>
<span class="ruby-comment cmt"># File dbload.rb, line 37</span>
  <span class="ruby-keyword kw">def</span> <span class="ruby-identifier">initialize</span>
    <span class="ruby-ivar">@opts</span> = <span class="ruby-identifier">get_options</span>
    <span class="ruby-keyword kw">case</span> <span class="ruby-ivar">@opts</span>.<span class="ruby-identifier">dbtype</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:dbm</span>
      <span class="ruby-identifier">require</span> <span class="ruby-value str">'dbm'</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:gdbm</span>
      <span class="ruby-identifier">require</span> <span class="ruby-value str">'gdbm'</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:sdbm</span>
      <span class="ruby-identifier">require</span> <span class="ruby-value str">'sdbm'</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:sqlite</span>
      <span class="ruby-identifier">require</span> <span class="ruby-value str">'sqlite'</span>
      <span class="ruby-identifier">require</span> <span class="ruby-value str">'storage/sqlitehash'</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:sqlite3</span>
      <span class="ruby-identifier">require</span> <span class="ruby-value str">'sqlite3'</span>
      <span class="ruby-identifier">require</span> <span class="ruby-value str">'storage/sqlite3hash'</span>
    <span class="ruby-keyword kw">end</span>
    <span class="ruby-ivar">@dbtop</span> = <span class="ruby-value">0</span>
    <span class="ruby-ivar">@db</span> = {}
    <span class="ruby-ivar">@count</span> = <span class="ruby-value">0</span>
  <span class="ruby-keyword kw">end</span>
</pre>
          </div>
        </div>
      </div>

      <h3 class="section-bar">Public Instance methods</h3>

      <div id="method-M000036" class="method-detail">
        <a name="M000036"></a>

        <div class="method-heading">
          <a href="#M000036" class="method-signature">
          <span class="method-name">get_options</span><span class="method-args">()</span>
          </a>
        </div>
      
        <div class="method-description">
          <p>
Processes command line arguments
</p>
          <p><a class="source-toggle" href="#"
            onclick="toggleCode('M000036-source');return false;">[Source]</a></p>
          <div class="method-source-code" id="M000036-source">
<pre>
<span class="ruby-comment cmt"># File dbload.rb, line 61</span>
  <span class="ruby-keyword kw">def</span> <span class="ruby-identifier">get_options</span>

    <span class="ruby-comment cmt"># The myopts specified on the command line will be collected in *myopts*.
</span>
    <span class="ruby-comment cmt"># We set default values here.
</span>
    <span class="ruby-identifier">myopts</span> = <span class="ruby-constant">OpenStruct</span>.<span class="ruby-identifier">new</span>
    <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ifile</span> = <span class="ruby-keyword kw">nil</span>
    <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ofile</span> = <span class="ruby-keyword kw">nil</span>
    <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">dbtype</span> = <span class="ruby-keyword kw">nil</span>

    <span class="ruby-identifier">opts</span> = <span class="ruby-constant">OptionParser</span>.<span class="ruby-identifier">new</span> <span class="ruby-keyword kw">do</span> <span class="ruby-operator">|</span><span class="ruby-identifier">opts</span><span class="ruby-operator">|</span>
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">banner</span> = <span class="ruby-node">&quot;Database Loader #{VERSION}&quot;</span>
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">separator</span> <span class="ruby-value str">&quot;&quot;</span>
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">separator</span> <span class="ruby-node">&quot;Usage: ruby #{$0} [options]&quot;</span>
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">separator</span> <span class="ruby-value str">&quot;&quot;</span>
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">separator</span> <span class="ruby-value str">&quot;Options:&quot;</span>
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">on</span>(<span class="ruby-value str">&quot;-i&quot;</span>, <span class="ruby-value str">&quot;--ifile FILE&quot;</span>, <span class="ruby-constant">String</span>,
              <span class="ruby-value str">&quot;Select the yaml file to read&quot;</span>,
              <span class="ruby-value str">&quot;  defaults to same as database&quot;</span>) {<span class="ruby-operator">|</span><span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ifile</span><span class="ruby-operator">|</span>}
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">on</span>(<span class="ruby-value str">&quot;-o&quot;</span>, <span class="ruby-value str">&quot;--ofile FILE&quot;</span>, <span class="ruby-constant">String</span>,
              <span class="ruby-value str">&quot;Select the database file to write&quot;</span>,
              <span class="ruby-value str">&quot;  extension determined automatically&quot;</span>) {<span class="ruby-operator">|</span><span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ofile</span><span class="ruby-operator">|</span>}
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">on</span>(<span class="ruby-value str">&quot;-t&quot;</span>, <span class="ruby-value str">&quot;--type DBTYPE&quot;</span>, <span class="ruby-constant">DATABASES</span>,
              <span class="ruby-value str">&quot;Select the database type - required (no default)&quot;</span>,
              <span class="ruby-node">&quot;  One of: #{DATABASES.join(&quot;, &quot;)}&quot;</span>,
              <span class="ruby-value str">&quot;    Example: -t gdbm&quot;</span>) {<span class="ruby-operator">|</span><span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">dbtype</span><span class="ruby-operator">|</span>}
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">on_tail</span>(<span class="ruby-value str">&quot;-h&quot;</span>, <span class="ruby-value str">&quot;--help&quot;</span>, <span class="ruby-value str">&quot;Show this message&quot;</span>) <span class="ruby-keyword kw">do</span>
        <span class="ruby-identifier">puts</span> <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">help</span>
        <span class="ruby-identifier">exit</span>
      <span class="ruby-keyword kw">end</span>
      <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">on_tail</span>(<span class="ruby-value str">&quot;-v&quot;</span>, <span class="ruby-value str">&quot;--version&quot;</span>, <span class="ruby-value str">&quot;Show version&quot;</span>) <span class="ruby-keyword kw">do</span>
        <span class="ruby-identifier">puts</span> <span class="ruby-node">&quot;Database Loader #{VERSION}&quot;</span>
        <span class="ruby-identifier">exit</span>
      <span class="ruby-keyword kw">end</span>
    <span class="ruby-keyword kw">end</span>

    <span class="ruby-identifier">opts</span>.<span class="ruby-identifier">parse!</span>(<span class="ruby-constant">ARGV</span>)
    <span class="ruby-identifier">raise</span>(<span class="ruby-constant">OptionParser</span><span class="ruby-operator">::</span><span class="ruby-constant">MissingArgument</span>.<span class="ruby-identifier">new</span>(<span class="ruby-value str">&quot;-t&quot;</span>)) <span class="ruby-keyword kw">if</span> <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">dbtype</span> <span class="ruby-operator">==</span> <span class="ruby-keyword kw">nil</span>
    <span class="ruby-identifier">raise</span>(<span class="ruby-constant">OptionParser</span><span class="ruby-operator">::</span><span class="ruby-constant">ParseError</span>, <span class="ruby-value str">&quot;Must specify input file!&quot;</span>) <span class="ruby-keyword kw">if</span> <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ifile</span>.<span class="ruby-identifier">nil?</span>
    <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ofile</span> = <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ifile</span>.<span class="ruby-identifier">dup</span> <span class="ruby-keyword kw">if</span> <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ofile</span>.<span class="ruby-identifier">nil?</span>
    <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ofile</span> <span class="ruby-operator">&lt;&lt;</span> <span class="ruby-value str">&quot;.gdbm&quot;</span> <span class="ruby-keyword kw">if</span> <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">dbtype</span> <span class="ruby-operator">==</span> <span class="ruby-identifier">:gdbm</span>
    <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ofile</span> <span class="ruby-operator">&lt;&lt;</span> <span class="ruby-value str">&quot;.sqlite&quot;</span> <span class="ruby-keyword kw">if</span> <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">dbtype</span> <span class="ruby-operator">==</span> <span class="ruby-identifier">:sqlite</span>
    <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ofile</span> <span class="ruby-operator">&lt;&lt;</span> <span class="ruby-value str">&quot;.sqlite3&quot;</span> <span class="ruby-keyword kw">if</span> <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">dbtype</span> <span class="ruby-operator">==</span> <span class="ruby-identifier">:sqlite3</span>
    <span class="ruby-identifier">myopts</span>.<span class="ruby-identifier">ifile</span> <span class="ruby-operator">&lt;&lt;</span> <span class="ruby-value str">&quot;.yaml&quot;</span>

    <span class="ruby-keyword kw">return</span> <span class="ruby-identifier">myopts</span>
  <span class="ruby-keyword kw">rescue</span> <span class="ruby-constant">OptionParser</span><span class="ruby-operator">::</span><span class="ruby-constant">ParseError</span>
    <span class="ruby-identifier">puts</span> <span class="ruby-node">&quot;ERROR - #{$!}&quot;</span>
    <span class="ruby-identifier">puts</span> <span class="ruby-value str">&quot;For help...&quot;</span>
    <span class="ruby-identifier">puts</span> <span class="ruby-node">&quot; ruby #{$0} --help&quot;</span>
    <span class="ruby-identifier">exit</span>
  <span class="ruby-keyword kw">end</span>
</pre>
          </div>
        </div>
      </div>

      <div id="method-M000037" class="method-detail">
        <a name="M000037"></a>

        <div class="method-heading">
          <a href="#M000037" class="method-signature">
          <span class="method-name">run</span><span class="method-args">()</span>
          </a>
        </div>
      
        <div class="method-description">
          <p>
Launches the loader
</p>
          <p><a class="source-toggle" href="#"
            onclick="toggleCode('M000037-source');return false;">[Source]</a></p>
          <div class="method-source-code" id="M000037-source">
<pre>
<span class="ruby-comment cmt"># File dbload.rb, line 116</span>
  <span class="ruby-keyword kw">def</span> <span class="ruby-identifier">run</span>

    <span class="ruby-constant">YAML</span><span class="ruby-operator">::</span><span class="ruby-identifier">load_file</span>(<span class="ruby-ivar">@opts</span>.<span class="ruby-identifier">ifile</span>).<span class="ruby-identifier">each</span> <span class="ruby-keyword kw">do</span> <span class="ruby-operator">|</span><span class="ruby-identifier">o</span><span class="ruby-operator">|</span>
      <span class="ruby-ivar">@dbtop</span> = <span class="ruby-identifier">o</span>.<span class="ruby-identifier">id</span> <span class="ruby-keyword kw">if</span> <span class="ruby-identifier">o</span>.<span class="ruby-identifier">id</span> <span class="ruby-operator">&gt;</span> <span class="ruby-ivar">@dbtop</span>
      <span class="ruby-ivar">@db</span>[<span class="ruby-identifier">o</span>.<span class="ruby-identifier">id</span>]=<span class="ruby-identifier">o</span>
      <span class="ruby-ivar">@count</span> <span class="ruby-operator">+=</span> <span class="ruby-value">1</span>
    <span class="ruby-keyword kw">end</span>

    <span class="ruby-keyword kw">case</span> <span class="ruby-ivar">@opts</span>.<span class="ruby-identifier">dbtype</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:sdbm</span>
      <span class="ruby-constant">SDBM</span>.<span class="ruby-identifier">open</span>(<span class="ruby-ivar">@opts</span>.<span class="ruby-identifier">ofile</span>, <span class="ruby-value">0666</span>) <span class="ruby-keyword kw">do</span> <span class="ruby-operator">|</span><span class="ruby-identifier">db</span><span class="ruby-operator">|</span>
        <span class="ruby-ivar">@db</span>.<span class="ruby-identifier">each</span> {<span class="ruby-operator">|</span><span class="ruby-identifier">k</span>,<span class="ruby-identifier">v</span><span class="ruby-operator">|</span> <span class="ruby-identifier">db</span>[<span class="ruby-identifier">k</span>.<span class="ruby-identifier">to_s</span>] = <span class="ruby-constant">Utility</span>.<span class="ruby-identifier">encode</span>(<span class="ruby-identifier">v</span>)}
      <span class="ruby-keyword kw">end</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:gdbm</span>
      <span class="ruby-constant">GDBM</span>.<span class="ruby-identifier">open</span>(<span class="ruby-ivar">@opts</span>.<span class="ruby-identifier">ofile</span>, <span class="ruby-value">0666</span>) <span class="ruby-keyword kw">do</span> <span class="ruby-operator">|</span><span class="ruby-identifier">db</span><span class="ruby-operator">|</span>
        <span class="ruby-ivar">@db</span>.<span class="ruby-identifier">each</span> {<span class="ruby-operator">|</span><span class="ruby-identifier">k</span>,<span class="ruby-identifier">v</span><span class="ruby-operator">|</span> <span class="ruby-identifier">db</span>[<span class="ruby-identifier">k</span>.<span class="ruby-identifier">to_s</span>] = <span class="ruby-constant">Utility</span>.<span class="ruby-identifier">encode</span>(<span class="ruby-identifier">v</span>)}
      <span class="ruby-keyword kw">end</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:dbm</span>
      <span class="ruby-constant">DBM</span>.<span class="ruby-identifier">open</span>(<span class="ruby-ivar">@opts</span>.<span class="ruby-identifier">ofile</span>, <span class="ruby-value">0666</span>) <span class="ruby-keyword kw">do</span> <span class="ruby-operator">|</span><span class="ruby-identifier">db</span><span class="ruby-operator">|</span>
        <span class="ruby-ivar">@db</span>.<span class="ruby-identifier">each</span> {<span class="ruby-operator">|</span><span class="ruby-identifier">k</span>,<span class="ruby-identifier">v</span><span class="ruby-operator">|</span> <span class="ruby-identifier">db</span>[<span class="ruby-identifier">k</span>.<span class="ruby-identifier">to_s</span>] = <span class="ruby-constant">Utility</span>.<span class="ruby-identifier">encode</span>(<span class="ruby-identifier">v</span>)}
      <span class="ruby-keyword kw">end</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:sqlite</span>
      <span class="ruby-identifier">db</span> = <span class="ruby-constant">SQLite</span><span class="ruby-operator">::</span><span class="ruby-constant">Database</span>.<span class="ruby-identifier">open</span>(<span class="ruby-ivar">@opts</span>.<span class="ruby-identifier">ofile</span>)
      <span class="ruby-keyword kw">begin</span>
        <span class="ruby-identifier">db</span>.<span class="ruby-identifier">execute</span>(<span class="ruby-value str">&quot;drop table tmud;&quot;</span>)
      <span class="ruby-keyword kw">rescue</span>
      <span class="ruby-keyword kw">end</span>
      <span class="ruby-identifier">db</span>.<span class="ruby-identifier">execute</span>(<span class="ruby-value str">&quot;create table tmud (id integer primary key, data text);&quot;</span>)
      <span class="ruby-ivar">@db</span>.<span class="ruby-identifier">each</span> {<span class="ruby-operator">|</span><span class="ruby-identifier">k</span>,<span class="ruby-identifier">v</span><span class="ruby-operator">|</span> <span class="ruby-identifier">db</span>[<span class="ruby-identifier">k</span>] = <span class="ruby-constant">Utility</span>.<span class="ruby-identifier">encode</span>(<span class="ruby-identifier">v</span>)}
      <span class="ruby-identifier">db</span>.<span class="ruby-identifier">close</span>
    <span class="ruby-keyword kw">when</span> <span class="ruby-identifier">:sqlite3</span>
      <span class="ruby-identifier">db</span> = <span class="ruby-constant">SQLite3</span><span class="ruby-operator">::</span><span class="ruby-constant">Database</span>.<span class="ruby-identifier">open</span>(<span class="ruby-ivar">@opts</span>.<span class="ruby-identifier">ofile</span>)
      <span class="ruby-keyword kw">begin</span>
        <span class="ruby-identifier">db</span>.<span class="ruby-identifier">execute</span>(<span class="ruby-value str">&quot;drop table tmud;&quot;</span>)
      <span class="ruby-keyword kw">rescue</span> <span class="ruby-constant">Exception</span>
      <span class="ruby-keyword kw">end</span>
      <span class="ruby-identifier">db</span>.<span class="ruby-identifier">execute</span>(<span class="ruby-value str">&quot;create table tmud (id integer primary key, data text);&quot;</span>)
      <span class="ruby-ivar">@db</span>.<span class="ruby-identifier">each</span> {<span class="ruby-operator">|</span><span class="ruby-identifier">k</span>,<span class="ruby-identifier">v</span><span class="ruby-operator">|</span> <span class="ruby-identifier">db</span>[<span class="ruby-identifier">k</span>] = <span class="ruby-constant">Utility</span>.<span class="ruby-identifier">encode</span>(<span class="ruby-identifier">v</span>)}
      <span class="ruby-identifier">db</span>.<span class="ruby-identifier">close</span>
    <span class="ruby-keyword kw">end</span>

    <span class="ruby-identifier">puts</span> <span class="ruby-node">&quot;Highest object in use   : #{@dbtop}&quot;</span>
    <span class="ruby-identifier">puts</span> <span class="ruby-node">&quot;Count of objects dumped : #{@count}&quot;</span>
  <span class="ruby-keyword kw">end</span>
</pre>
          </div>
        </div>
      </div>


    </div>


  </div>


<div id="validator-badges">
  <p><small><a href="http://validator.w3.org/check/referer">[Validate]</a></small></p>
</div>

</body>
</html>