user-tools/LevelDB

   1 #!/usr/bin/env perl
   2
   3 use Tie::LevelDB;
   4 use Data::Dumper;
   5 use feature 'switch';
   6
   7
   8 if(scalar @ARGV == 0 or $ARGV[0] =~ /^-?-[h?]/i)
   9 {
  10         print "LevelDB - Commandline Interface for Google's leveldb key-value storage,
  11 based on Tie::LeveDB(3pm)
  12 Usage: LevelDB <leveldb-directory> <command> [<key> [<value>]]
  13 Commands:
  14   check             exit zero if <leveldb-directory> is a valid leveldb
  15   open              try to open the database, exit non-zero if failed
  16
  17   keys              enumerate all keys line-by-line
  18   show              display all keys and values in KEY<TAB>VALUE<LF> format
  19   get KEY           print the value of KEY
  20
  21   set KEY VAL       set the value of KEY to VAL unless already equals to VAL
  22   default KEY VAL   set the value of KEY unless KEY is already set
  23   replace KEY VAL   set the value of KEY unless KEY does not exist
  24   alter KEY VAL     same as 'set' but force writes
  25   del KEY           remove KEY
  26
  27   merge [FILE]      import all key-value pairs from FILE or STDIN in KEY<TAB>VALUE<LF> format
  28   merge-default [FILE]  same as 'merge' but sets only unset keys
  29   merge-replace [FILE]  same as 'merge' but sets only already existing keys
  30 ";
  31         exit 1;
  32 }
  33
  34
  35 if($ARGV[1] eq 'check')
  36 {
  37         if(-f $ARGV[0]."/LOG" and -f $ARGV[0]."/CURRENT")
  38         {
  39                 eval { new Tie::LevelDB::DB($ARGV[0]); 1; };
  40                 if($@)
  41                 {
  42                         if($@ =~ /^IO error: lock/)
  43                         {
  44                                 exit 0;
  45                         }
  46                         die $@;
  47                 }
  48                 exit 0;
  49         }
  50         exit 1;
  51 }
  52
  53 sub keyexists
  54 {
  55         return !! grep {$_ eq $_[0]} keys %DB;
  56 }
  57
  58 if($ARGV[1] =~ /^(keys|show|merge|merge-default|merge-replace|get|set|alter|replace|default|del|open)$/)
  59 {
  60         tie %DB, 'Tie::LevelDB', $ARGV[0];
  61
  62         given($ARGV[1])
  63         {
  64                 when('keys')
  65                 {
  66                         print "$_\n" for sort {lc $a cmp lc $b} keys %DB;
  67                 }
  68                 when('show')
  69                 {
  70                         print "$_\t$DB{$_}\n" for sort {lc $a cmp lc $b} keys %DB;
  71                 }
  72                 when(/^merge/)
  73                 {
  74                         if($ARGV[2] ne '')
  75                         {
  76                                 open $fh, '<', $ARGV[2] or die $!;
  77                         }
  78                         else
  79                         {
  80                                 $fh = \*STDIN;
  81                         }
  82
  83                         while(<$fh>)
  84                         {
  85                                 if(my ($key, $value) = /^(.+?)\t(.*)$/)
  86                                 {
  87                                         given($ARGV[1])
  88                                         {
  89                                                 when('merge')
  90                                                 {
  91                                                         if(not keyexists($key) or $DB{$key} ne $value)
  92                                                         {
  93                                                                 $DB{$key} = $value;
  94                                                         }
  95                                                 }
  96                                                 when('merge-default')
  97                                                 {
  98                                                         if(not keyexists($key))
  99                                                         {
 100                                                                 $DB{$key} = $value;
 101                                                         }
 102                                                 }
 103                                                 when('merge-replace')
 104                                                 {
 105                                                         if(keyexists($key) and $DB{$key} ne $value)
 106                                                         {
 107                                                                 $DB{$key} = $value;
 108                                                         }
 109                                                 }
 110                                         }
 111                                 }
 112                                 else
 113                                 {
 114                                         warn "Invalid input: $_";
 115                                 }
 116                         }
 117                 }
 118                 when('get')
 119                 {
 120                         print $DB{$ARGV[2]};
 121                 }
 122                 when('alter')
 123                 {
 124                         $DB{$ARGV[2]} = $ARGV[3];
 125                 }
 126                 when('set')
 127                 {
 128                         if(not keyexists($ARGV[2]) or $DB{$ARGV[2]} ne $ARGV[3])
 129                         {
 130                                 $DB{$ARGV[2]} = $ARGV[3];
 131                         }
 132                 }
 133                 when('replace')
 134                 {
 135                         if(keyexists($ARGV[2]))
 136                         {
 137                                 $DB{$ARGV[2]} = $ARGV[3];
 138                         }
 139                 }
 140                 when('default')
 141                 {
 142                         if(not keyexists($ARGV[2]))
 143                         {
 144                                 $DB{$ARGV[2]} = $ARGV[3];
 145                         }
 146                 }
 147                 when('del')
 148                 {
 149                         delete $DB{$ARGV[2]};
 150                 }
 151                 when('open')
 152                 {
 153                         1;
 154                 }
 155         }
 156 }
 157 else
 158 {
 159         warn "Invalid command: $ARGV[1]\n";
 160         exit 1;
 161 }
 162
 163 __END__
 164
 165 =pod
 166
 167 =head1 NAME
 168
 169 LevelDB - Commandline interface for Google's leveldb key-value storage
 170
 171 =cut