ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/deliantra/Deliantra-Client/DC/DB.pm
(Generate patch)

Comparing deliantra/Deliantra-Client/DC/DB.pm (file contents):
Revision 1.11 by root, Sat Jul 14 12:05:53 2007 UTC vs.
Revision 1.23 by root, Fri Nov 16 04:54:25 2007 UTC

16 16
17use strict; 17use strict;
18use utf8; 18use utf8;
19 19
20use Carp (); 20use Carp ();
21use AnyEvent ();
22use Storable (); 21use Storable ();
23use Config; 22use Config;
24 23
25use CFPlus; 24use CFPlus;
26 25
27our $DB_HOME = "$Crossfire::VARDIR/cfplus-$BerkeleyDB::db_version-$Config{archname}"; 26our $DB_HOME = "$Crossfire::VARDIR/cfplus-$BerkeleyDB::db_version-$Config{archname}";
28 27
29sub path_of_res($) { 28sub path_of_res($) {
29 utf8::downgrade $_[0]; # bug in unpack "H*"
30 "$DB_HOME/res-data-$_[0]" 30 "$DB_HOME/res-data-" . unpack "H*", $_[0]
31} 31}
32 32
33sub sync { 33sub sync {
34 # for debugging 34 # for debugging
35 #CFPlus::DB::Server::req (sync => sub { }); 35 #CFPlus::DB::Server::req (sync => sub { });
50 50
51sub unlink($$) { 51sub unlink($$) {
52 CFPlus::DB::Server::req (unlink => @_); 52 CFPlus::DB::Server::req (unlink => @_);
53} 53}
54 54
55sub read_file($$) {
56 CFPlus::DB::Server::req (read_file => @_);
57}
58
55sub write_file($$$) { 59sub write_file($$$) {
56 CFPlus::DB::Server::req (write_file => @_); 60 CFPlus::DB::Server::req (write_file => @_);
57} 61}
58 62
59sub prefetch_file($$$) { 63sub prefetch_file($$$) {
60 CFPlus::DB::Server::req (prefetch_file => @_); 64 CFPlus::DB::Server::req (prefetch_file => @_);
61} 65}
62 66
67sub logprint($$$) {
68 CFPlus::DB::Server::req (logprint => @_);
69}
70
63our $tilemap; 71our $tilemap;
64 72
65sub get_tile_id_sync($) { 73sub get_tile_id_sync($) {
66 my ($hash) = @_; 74 my ($name) = @_;
67 75
68 # fetch the full face table first 76 # fetch the full face table first
69 unless ($tilemap) { 77 unless ($tilemap) {
70 CFPlus::DB::Server::req (table => facemap => sub { $tilemap = $_[0] }); 78 CFPlus::DB::Server::req (table => facemap => sub {
79 $tilemap = $_[0];
80 delete $tilemap->{id};
81 my %maptile = reverse %$tilemap;#d#
82 if ((scalar keys %$tilemap) != (scalar keys %maptile)) {#d#
83 $tilemap = { };#d#
84 CFPlus::error "FATAL: facemap is not a 1:1 mapping, please report this and delete your $DB_HOME directory!\n";#d#
85 }#d#
86 });
71 sync; 87 sync;
72 } 88 }
73 89
74 $tilemap->{$hash} ||= do { 90 $tilemap->{$name} ||= do {
75 my $id; 91 my $id;
76 CFPlus::DB::Server::req (get_tile_id => $hash, sub { $id = $_[0] }); 92 CFPlus::DB::Server::req (get_tile_id => $name, sub { $id = $_[0] });
77 sync; 93 sync;
78 $id 94 $id
79 } 95 }
80} 96}
81 97
82package CFPlus::DB::Server; 98package CFPlus::DB::Server;
83 99
84use strict; 100use strict;
85 101
102use EV ();
86use Fcntl; 103use Fcntl;
87use BerkeleyDB; 104use BerkeleyDB;
88 105
89our $DB_ENV; 106our $DB_ENV;
90our $DB_STATE; 107our $DB_STATE;
101 -Cachesize => 8_000_000, 118 -Cachesize => 8_000_000,
102 -ErrFile => "$DB_HOME/errorlog.txt", 119 -ErrFile => "$DB_HOME/errorlog.txt",
103# -ErrPrefix => "DATABASE", 120# -ErrPrefix => "DATABASE",
104 -Verbose => 1, 121 -Verbose => 1,
105 -Flags => DB_CREATE | DB_RECOVER | DB_INIT_MPOOL | DB_INIT_LOCK | DB_INIT_TXN | $recover, 122 -Flags => DB_CREATE | DB_RECOVER | DB_INIT_MPOOL | DB_INIT_LOCK | DB_INIT_TXN | $recover,
106 -SetFlags => DB_AUTO_COMMIT | DB_LOG_AUTOREMOVE, 123 -SetFlags => DB_AUTO_COMMIT | DB_LOG_AUTOREMOVE | DB_TXN_WRITE_NOSYNC,
107 or die "unable to create/open database home $DB_HOME: $BerkeleyDB::Error"; 124 or die "unable to create/open database home $DB_HOME: $BerkeleyDB::Error";
108 125
109 1 126 1
110} 127}
111 128
119 -Env => $DB_ENV, 136 -Env => $DB_ENV,
120 -Filename => $table, 137 -Filename => $table,
121# -Filename => "database", 138# -Filename => "database",
122# -Subname => $table, 139# -Subname => $table,
123 -Property => DB_CHKSUM, 140 -Property => DB_CHKSUM,
124 -Flags => DB_CREATE | DB_UPGRADE, 141 -Flags => DB_AUTO_COMMIT | DB_CREATE | DB_UPGRADE,
125 or die "unable to create/open database table $_[0]: $BerkeleyDB::Error" 142 or die "unable to create/open database table $_[0]: $BerkeleyDB::Error"
126 } 143 }
127} 144}
128
129our $SYNC_INTERVAL = 60;
130 145
131our %CB; 146our %CB;
132our $FH; 147our $FH;
133our $ID = "aaa0"; 148our $ID = "aaa0";
134our ($fh_r_watcher, $fh_w_watcher); 149our ($fh_r_watcher, $fh_w_watcher);
135our $sync_timer; 150our $sync_timer;
136our $write_buf; 151our $write_buf;
137our $read_buf; 152our $read_buf;
138 153
154our $SYNC = EV::timer_ns 0, 60, sub {
155 $_[0]->stop;
156 CFPlus::DB::Server::req (sync => sub { });
157};
158
139sub fh_write { 159sub fh_write {
140 my $len = syswrite $FH, $write_buf; 160 my $len = syswrite $FH, $write_buf;
141 161
142 substr $write_buf, 0, $len, ""; 162 substr $write_buf, 0, $len, "";
143 163
144 undef $fh_w_watcher 164 $fh_w_watcher->stop
145 unless length $write_buf; 165 unless length $write_buf;
146} 166}
147 167
148sub fh_read { 168sub fh_read {
149 my $status = sysread $FH, $read_buf, 16384, length $read_buf; 169 my $status = sysread $FH, $read_buf, 16384, length $read_buf;
185 205
186 my $id = ++$ID; 206 my $id = ++$ID;
187 $write_buf .= pack "N/a*", Storable::freeze [$id, $type, @args]; 207 $write_buf .= pack "N/a*", Storable::freeze [$id, $type, @args];
188 $CB{$id} = $cb; 208 $CB{$id} = $cb;
189 209
190 $fh_w_watcher = AnyEvent->io (fh => $FH, poll => 'w', cb => \&fh_write); 210 $fh_w_watcher->start;
191} 211 $SYNC->again unless $SYNC->is_active;
192
193sub sync_tick {
194 req "sync", sub { };
195 $sync_timer = AnyEvent->timer (after => $SYNC_INTERVAL, cb => \&sync_tick);
196} 212}
197 213
198sub do_sync { 214sub do_sync {
199 $DB_ENV->txn_checkpoint (0, 0, 0); 215 $DB_ENV->txn_checkpoint (0, 0, 0);
200 () 216 ()
240 256
241 \%kv 257 \%kv
242} 258}
243 259
244sub do_get_tile_id { 260sub do_get_tile_id {
245 my ($hash) = @_; 261 my ($name) = @_;
246 262
247 my $id; 263 my $id;
248 my $table = table "facemap"; 264 my $table = table "facemap";
249 265
250 return $id 266 return $id
251 if $table->db_get ($hash, $id) == 0; 267 if $table->db_get ($name, $id) == 0;
252 268
253 for (1..100) { 269 for (1..100) {
254 my $txn = $DB_ENV->txn_begin; 270 my $txn = $DB_ENV->txn_begin;
255 my $status = $table->db_get (id => $id); 271 my $status = $table->db_get (id => $id);
256 if ($status == 0 || $status == BerkeleyDB::DB_NOTFOUND) { 272 if ($status == 0 || $status == BerkeleyDB::DB_NOTFOUND) {
257 $id = ($id || 64) + 1; 273 $id = ($id || 64) + 1;
258 if ($table->db_put (id => $id) == 0 274 if ($table->db_put (id => $id) == 0
259 && $table->db_put ($hash => $id) == 0) { 275 && $table->db_put ($name => $id) == 0) {
260 $txn->txn_commit; 276 $txn->txn_commit;
261 277
262 return $id; 278 return $id;
263 } 279 }
264 } 280 }
265 $txn->txn_abort; 281 $txn->txn_abort;
282 select undef, undef, undef, 0.01 * rand;
266 } 283 }
267 284
268 die "maximum number of transaction retries reached - database problems?"; 285 die "maximum number of transaction retries reached - database problems?";
269} 286}
270 287
271sub do_unlink { 288sub do_unlink {
272 unlink $_[0]; 289 unlink $_[0];
273} 290}
274 291
292sub do_read_file {
293 my ($path) = @_;
294
295 utf8::downgrade $path;
296 open my $fh, "<:raw", $path
297 or return;
298 sysread $fh, my $buf, -s $fh;
299
300 $buf
301}
302
275sub do_write_file { 303sub do_write_file {
276 my ($file, $data) = @_; 304 my ($path, $data) = @_;
277 305
278 utf8::downgrade $file; 306 utf8::downgrade $path;
279 utf8::downgrade $data; 307 utf8::downgrade $data;
280 open my $fh, ">:raw", $file 308 open my $fh, ">:raw", $path
281 or return; 309 or return;
282 print $fh $data; 310 syswrite $fh, $data;
283 close $fh; 311 close $fh;
284 312
285 1 313 1
286} 314}
287 315
288sub do_prefetch_file { 316sub do_prefetch_file {
289 my ($file, $size) = @_; 317 my ($path, $size) = @_;
290 318
291 utf8::downgrade $file; 319 utf8::downgrade $path;
292 open my $fh, "<:raw", $file 320 open my $fh, "<:raw", $path
293 or return; 321 or return;
294 sysread $fh, my $buf, $size; 322 sysread $fh, my $buf, $size;
295 323
296 1 324 1
325}
326
327our %LOG_FH;
328
329sub do_logprint {
330 my ($path, $line) = @_;
331
332 $LOG_FH{$path} ||= do {
333 open my $fh, ">>:utf8", $path
334 or warn "Couldn't open logfile $path: $!";
335
336 $fh->autoflush (1);
337
338 $fh
339 };
340
341 my ($sec, $min, $hour, $mday, $mon, $year) = localtime time;
342
343 my $ts = sprintf "%04d-%02d-%02d %02d:%02d:%02d",
344 $year + 1900, $mon + 1, $mday, $hour, $min, $sec;
345
346 print { $LOG_FH{$path} } "$ts $line\n"
297} 347}
298 348
299sub run { 349sub run {
300 ($FH, my $fh) = CFPlus::socketpipe; 350 ($FH, my $fh) = CFPlus::socketpipe;
301 351
303 my $oldfh = select $fh; $| = 1; select $oldfh; 353 my $oldfh = select $fh; $| = 1; select $oldfh;
304 354
305 my $pid = fork; 355 my $pid = fork;
306 356
307 if (defined $pid && !$pid) { 357 if (defined $pid && !$pid) {
358 local $SIG{QUIT};
308 local $SIG{__DIE__}; 359 local $SIG{__DIE__};
360 local $SIG{__WARN__};
309 eval { 361 eval {
310 close $FH; 362 close $FH;
311 363
312 unless (eval { open_db }) { 364 unless (eval { open_db }) {
313 eval { File::Path::rmtree $DB_HOME }; 365 eval { File::Path::rmtree $DB_HOME };
326 my ($id, $type, @args) = @$req; 378 my ($id, $type, @args) = @$req;
327 my $cb = CFPlus::DB::Server->can ("do_$type") 379 my $cb = CFPlus::DB::Server->can ("do_$type")
328 or die "$type: unknown database request type\n"; 380 or die "$type: unknown database request type\n";
329 my $res = pack "N/a*", Storable::freeze [$id, $cb->(@args)]; 381 my $res = pack "N/a*", Storable::freeze [$id, $cb->(@args)];
330 (syswrite $fh, $res) == length $res 382 (syswrite $fh, $res) == length $res
331 or die; 383 or die "DB::write: $!";
332 } 384 }
333 }; 385 };
334 386
335 my $error = $@; 387 my $error = $@;
336 388
337 eval { 389 eval {
390 $DB_ENV->txn_checkpoint (0, 0, 0);
391
338 undef %DB_TABLE; 392 undef %DB_TABLE;
339 undef $DB_ENV; 393 undef $DB_ENV;
340 394
341 Storable::store_fd [die => $error], $fh; 395 Storable::store_fd [die => $error], $fh;
342 }; 396 };
347 close $fh; 401 close $fh;
348 CFPlus::fh_nonblocking $FH, 1; 402 CFPlus::fh_nonblocking $FH, 1;
349 403
350 $CB{die} = sub { die shift }; 404 $CB{die} = sub { die shift };
351 405
352 $fh_r_watcher = AnyEvent->io (fh => $FH, poll => 'r', cb => \&fh_read); 406 $fh_r_watcher = EV::io $FH, EV::READ , \&fh_read;
353 407 $fh_w_watcher = EV::io $FH, EV::WRITE, \&fh_write;
354 sync_tick; 408 $SYNC->again unless $SYNC->is_active;
355} 409}
356 410
357sub stop { 411sub stop {
358 close $FH; 412 close $FH;
359} 413}

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines