bludgeon/SanityCheck.cs

   1
   2 using System;
   3 using System.Collections;
   4
   5 using Beagle.Util;
   6
   7 namespace Bludgeon {
   8
   9         public class SanityCheck {
  10
  11                 private SanityCheck () { }
  12
  13                 static public bool CheckQuery (Beagle.Query query, FileModel root)
  14                 {
  15                         Hashtable matching_path_hash;
  16                         matching_path_hash = new Hashtable ();
  17
  18                         foreach (Uri uri in QueryFu.GetUris (query))
  19                                 matching_path_hash [uri.LocalPath] = null;
  20
  21                         bool success;
  22                         success = true;
  23
  24                         foreach (FileModel file in root.GetMatchingDescendants (query)) {
  25                                 if (! matching_path_hash.Contains (file.FullName)) {
  26                                         Log.Failure ("Missing match {0}", file.FullName);
  27                                         success = false;
  28                                 }
  29                                 matching_path_hash.Remove (file.FullName);
  30                         }
  31
  32                         foreach (string path in matching_path_hash.Keys) {
  33                                 Log.Failure ("Unexpected match {0}", path);
  34                                 success = false;
  35                         }
  36
  37                         return success;
  38                 }
  39
  40                 static public bool VerifyIndex (FileModel root)
  41                 {
  42                         Log.Info ("Verifying index for {0}", root.FullName);
  43
  44                         bool success;
  45                         success = true;
  46
  47                         for (int i = 0; i < Token.Count; ++i) {
  48                                 Beagle.Query query;
  49                                 query = QueryFu.NewTokenQuery (i);
  50
  51                                 if (! CheckQuery (query, root)) {
  52                                         Log.Spew ("Failed query is:");
  53                                         QueryFu.SpewQuery (query);
  54                                         success = false;
  55                                 }
  56                         }
  57
  58                         if (success)
  59                                 Log.Info ("Index successfully verified for {0}", root.FullName);
  60                         else
  61                                 Log.Info ("Verification failed for {0}", root.FullName);
  62
  63                         return success;
  64                 }
  65
  66                 static public bool TestRandomQueries (FileModel root,
  67                                                       double    minutes_to_run)
  68                 {
  69                         Log.Info ("Running random queries on {0} for {1:0.0} minutes", root.FullName, minutes_to_run);
  70
  71                         bool success;
  72                         success = true;
  73
  74                         Stopwatch sw;
  75                         sw = new Stopwatch ();
  76                         sw.Start ();
  77
  78                         int count = 0;
  79                         while (true) {
  80                                 if ((count % 100 == 0) && sw.ElapsedTime > minutes_to_run * 60)
  81                                         break;
  82                                 ++count;
  83
  84                                 Beagle.Query query;
  85                                 query = QueryFu.NewRandomQuery ();
  86
  87                                 if (! CheckQuery (query, root)) {
  88                                         Log.Spew ("Failed query is:");
  89                                         QueryFu.SpewQuery (query);
  90                                         success = false;
  91                                         break;
  92                                 }
  93                         }
  94
  95                         // In case we ended early
  96                         minutes_to_run = sw.ElapsedTime / 60;
  97
  98                         Log.Spew ("Ran {0} queries in {1:0.0} minutes ({2:0.0} queries/s)",
  99                                   count, minutes_to_run, count / (minutes_to_run * 60));
 100
 101                         return success;
 102                 }
 103
 104         }
 105 }