src/AltSpider/AltModel.cs

   1 using System;
   2 using System.Collections.Generic;
   3 using System.Data;
   4 using System.Data.Common;
   5 using System.Diagnostics;
   6
   7 using MySql.Data.MySqlClient;
   8
   9 namespace AltSpider
  10 {
  11     public class AltModel : IDisposable
  12     {
  13         private readonly string _connectionString;
  14         private readonly DbConnection _connection;
  15
  16         public AltModel()
  17         {
  18             DbConnectionStringBuilder dbcsb
  19                 = new DbConnectionStringBuilder
  20                       {
  21                           { "Data Source", "localhost" },
  22                           { "User ID", "root" },
  23                           { "Database", "share_index" },
  24                           { "CharSet", "utf8" }
  25                       };
  26
  27             _connectionString = dbcsb.ConnectionString;
  28
  29             _connection = new MySqlConnection( _connectionString );
  30             _connection.Open();
  31         }
  32
  33         /// <summary>
  34         /// Saves given path to the database.
  35         /// </summary>
  36         /// <param name="path"></param>
  37         /// <returns>Leaf node ID</returns>
  38         /// <exception cref="ApplicationException">Thrown if saving failed.</exception>
  39         public int SavePath( string path )
  40         {
  41             IEnumerable<string> getTokens = GetTokens( path );
  42
  43             int parentId = 1;
  44
  45             foreach ( string token in getTokens )
  46             {
  47                 parentId = SaveNode( token, parentId );
  48                 if ( parentId == -1 )
  49                 {
  50                     throw new ApplicationException( String.Format( "Saving node {0} of {1} failed.", token, path ) );
  51                 }
  52             }
  53
  54             return parentId;
  55         }
  56
  57         /// <summary>
  58         /// Adds new node to the if does not exist in the database otherwise, loads the node.
  59         /// </summary>
  60         /// <param name="nodeName"></param>
  61         /// <param name="parentNodeId"></param>
  62         /// <returns>Child node ID</returns>
  63         private int SaveNode( string nodeName, int parentNodeId )
  64         {
  65             DbCommand cmd = _connection.CreateCommand();
  66             cmd.CommandText = "proc_save_node";
  67             cmd.CommandType = CommandType.StoredProcedure;
  68
  69             DbParameter nameParam = cmd.CreateParameter();
  70             nameParam.Direction = ParameterDirection.Input;
  71             nameParam.ParameterName = "$name";
  72             nameParam.Value = nodeName;
  73
  74             DbParameter parendIdParam = cmd.CreateParameter();
  75             parendIdParam.Direction = ParameterDirection.Input;
  76             parendIdParam.ParameterName = "$parent_id";
  77             parendIdParam.Value = parentNodeId;
  78
  79             DbParameter childIdParam = cmd.CreateParameter();
  80             childIdParam.Direction = ParameterDirection.Output;
  81             childIdParam.ParameterName = "$child_node_id";
  82
  83             cmd.Parameters.AddRange( new[] { nameParam, parendIdParam, childIdParam } );
  84
  85             cmd.ExecuteNonQuery();
  86
  87             int childNodeId = -1;
  88             if ( childIdParam.Value != null && childIdParam.Value != DBNull.Value )
  89             {
  90                 childNodeId = (int) childIdParam.Value;
  91             }
  92
  93             cmd.Dispose();
  94
  95             return childNodeId;
  96         }
  97
  98         /// <summary>
  99         /// Split given path into separate tokens (directory and file names).
 100         /// </summary>
 101         /// <param name="path"></param>
 102         /// <returns></returns>
 103         private static IEnumerable<string> GetTokens( string path )
 104         {
 105             string[] patrs = path.Substring( 1 ).Split( new[] { '/', '\\' }, StringSplitOptions.RemoveEmptyEntries );
 106             patrs[ 0 ] = path[ 0 ] + patrs[ 0 ];
 107             return patrs;
 108         }
 109
 110         public void Dispose()
 111         {
 112             if ( _connection.State != ConnectionState.Closed )
 113             {
 114                 try
 115                 {
 116                     _connection.Close();
 117                 }
 118                 catch ( DbException ex )
 119                 {
 120                     Trace.WriteLine( ex.Message, "Closing connection" );
 121                 }
 122             }
 123         }
 124
 125         /// <summary>
 126         /// Save file info into the database.
 127         /// </summary>
 128         /// <param name="nodeId">File node id</param>
 129         /// <param name="size">File size in bytes</param>
 130         /// <param name="ctime">File creation time</param>
 131         /// <param name="mtime">File's last modification time</param>
 132         /// <param name="atime">File's last access time</param>
 133         public void SaveFileInfo( int nodeId, long size, DateTime ctime, DateTime mtime, DateTime atime )
 134         {
 135             DbCommand cmd = _connection.CreateCommand();
 136             cmd.CommandText = "proc_save_file_info";
 137             cmd.CommandType = CommandType.StoredProcedure;
 138
 139             DbParameter idParam = cmd.CreateParameter();
 140             idParam.Direction = ParameterDirection.Input;
 141             idParam.ParameterName = "$id";
 142             idParam.Value = nodeId;
 143
 144             DbParameter sizeParam = cmd.CreateParameter();
 145             sizeParam.Direction = ParameterDirection.Input;
 146             sizeParam.ParameterName = "$size";
 147             sizeParam.Value = size;
 148
 149             DbParameter ctimeParam = cmd.CreateParameter();
 150             ctimeParam.Direction = ParameterDirection.Input;
 151             ctimeParam.ParameterName = "$ctime";
 152             ctimeParam.Value = ctime;
 153
 154             DbParameter mtimeParam = cmd.CreateParameter();
 155             mtimeParam.Direction = ParameterDirection.Input;
 156             mtimeParam.ParameterName = "$mtime";
 157             mtimeParam.Value = mtime;
 158
 159             DbParameter atimeParam = cmd.CreateParameter();
 160             atimeParam.Direction = ParameterDirection.Input;
 161             atimeParam.ParameterName = "$atime";
 162             atimeParam.Value = atime;
 163
 164             cmd.Parameters.AddRange( new[] { idParam, sizeParam, ctimeParam, mtimeParam, atimeParam } );
 165
 166             cmd.ExecuteNonQuery();
 167             cmd.Dispose();
 168         }
 169
 170         /// <summary>
 171         /// Get list of shares stored in the database.
 172         /// </summary>
 173         /// <returns>List of shares</returns>
 174         public List<string> GetShares()
 175         {
 176             List<string> shares = new List<string>();
 177
 178             DbCommand cmd = _connection.CreateCommand();
 179             cmd.CommandText = "SELECT path FROM view_shares";
 180             cmd.CommandType = CommandType.Text;
 181
 182             DbDataReader reader = cmd.ExecuteReader();
 183             if ( reader.HasRows )
 184             {
 185                 while ( reader.Read() )
 186                 {
 187                     shares.Add( reader[ "path" ] as string );
 188                 }
 189             }
 190             reader.Close();
 191             cmd.Dispose();
 192
 193             return shares;
 194         }
 195     }
 196 }