ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/deliantra/server/lib/cf.pm
(Generate patch)

Comparing deliantra/server/lib/cf.pm (file contents):
Revision 1.202 by root, Mon Jan 29 20:16:28 2007 UTC vs.
Revision 1.211 by root, Tue Feb 13 16:23:32 2007 UTC

… …
15use Coro::Timer; 15use Coro::Timer;
16use Coro::Signal; 16use Coro::Signal;
17use Coro::Semaphore; 17use Coro::Semaphore;
18use Coro::AIO; 18use Coro::AIO;
19 19
20use BDB ();
20use Data::Dumper; 21use Data::Dumper;
21use Digest::MD5; 22use Digest::MD5;
22use Fcntl; 23use Fcntl;
23use IO::AIO 2.32 (); 24use IO::AIO 2.32 ();
24use YAML::Syck (); 25use YAML::Syck ();
25use Time::HiRes; 26use Time::HiRes;
27use Compress::LZF;
28
29Compress::LZF::sfreeze_cr { }; # prime Compress::LZF so it does not use require later
26 30
27use Event; $Event::Eval = 1; # no idea why this is required, but it is 31use Event; $Event::Eval = 1; # no idea why this is required, but it is
28 32
29sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload 33sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload
30 34
… …
48our $TICK_WATCHER; 52our $TICK_WATCHER;
49our $AIO_POLL_WATCHER; 53our $AIO_POLL_WATCHER;
50our $WRITE_RUNTIME_WATCHER; 54our $WRITE_RUNTIME_WATCHER;
51our $NEXT_TICK; 55our $NEXT_TICK;
52our $NOW; 56our $NOW;
57our $USE_FSYNC = 1; # use fsync to write maps - default off
58
59our $BDB_POLL_WATCHER;
60our $DB_ENV;
53 61
54our %CFG; 62our %CFG;
55 63
56our $UPTIME; $UPTIME ||= time; 64our $UPTIME; $UPTIME ||= time;
57our $RUNTIME; 65our $RUNTIME;
58 66
59our %PLAYER; # all users 67our %PLAYER; # all users
60our %MAP; # all maps 68our %MAP; # all maps
61our $LINK_MAP; # the special {link} map, which is always available 69our $LINK_MAP; # the special {link} map, which is always available
62our $RANDOM_MAPS = cf::localdir . "/random"; 70our $RANDOM_MAPS = cf::localdir . "/random";
71our $BDB_ENV_DIR = cf::localdir . "/db";
63 72
64our $WAIT_FOR_TICK; $WAIT_FOR_TICK ||= new Coro::Signal; 73our $WAIT_FOR_TICK; $WAIT_FOR_TICK ||= new Coro::Signal;
65our $WAIT_FOR_TICK_ONE; $WAIT_FOR_TICK_ONE ||= new Coro::Signal; 74our $WAIT_FOR_TICK_ONE; $WAIT_FOR_TICK_ONE ||= new Coro::Signal;
66 75
67# used to convert map paths into valid unix filenames by replacing / by ∕ 76# used to convert map paths into valid unix filenames by replacing / by ∕
… …
80mkdir cf::localdir; 89mkdir cf::localdir;
81mkdir cf::localdir . "/" . cf::playerdir; 90mkdir cf::localdir . "/" . cf::playerdir;
82mkdir cf::localdir . "/" . cf::tmpdir; 91mkdir cf::localdir . "/" . cf::tmpdir;
83mkdir cf::localdir . "/" . cf::uniquedir; 92mkdir cf::localdir . "/" . cf::uniquedir;
84mkdir $RANDOM_MAPS; 93mkdir $RANDOM_MAPS;
94mkdir $BDB_ENV_DIR;
85 95
86our $EMERGENCY_POSITION; 96our $EMERGENCY_POSITION;
87 97
88sub cf::map::normalise; 98sub cf::map::normalise;
89 99
… …
346 356
347 $coro 357 $coro
348} 358}
349 359
350sub write_runtime { 360sub write_runtime {
361 $TICK_WATCHER->is_active or cf::cleanup "mainloop frozen but runtime active", 1;
362
351 my $runtime = cf::localdir . "/runtime"; 363 my $runtime = cf::localdir . "/runtime";
352 364
353 my $fh = aio_open "$runtime~", O_WRONLY | O_CREAT, 0644 365 my $fh = aio_open "$runtime~", O_WRONLY | O_CREAT, 0644
354 or return; 366 or return;
355 367
… …
358 # 60 allows for the watchdog to kill the server. 370 # 60 allows for the watchdog to kill the server.
359 371
360 (aio_write $fh, 0, (length $value), $value, 0) <= 0 372 (aio_write $fh, 0, (length $value), $value, 0) <= 0
361 and return; 373 and return;
362 374
375 # always fsync - this file is important
363 aio_fsync $fh 376 aio_fsync $fh
364 and return; 377 and return;
365 378
366 close $fh 379 close $fh
367 or return; 380 or return;
… …
752 $filename, length $$rdata, scalar @$objs; 765 $filename, length $$rdata, scalar @$objs;
753 766
754 if (my $fh = aio_open "$filename~", O_WRONLY | O_CREAT, 0600) { 767 if (my $fh = aio_open "$filename~", O_WRONLY | O_CREAT, 0600) {
755 chmod SAVE_MODE, $fh; 768 chmod SAVE_MODE, $fh;
756 aio_write $fh, 0, (length $$rdata), $$rdata, 0; 769 aio_write $fh, 0, (length $$rdata), $$rdata, 0;
757 aio_fsync $fh; 770 aio_fsync $fh if $cf::USE_FSYNC;
758 close $fh; 771 close $fh;
759 772
760 if (@$objs) { 773 if (@$objs) {
761 if (my $fh = aio_open "$filename.pst~", O_WRONLY | O_CREAT, 0600) { 774 if (my $fh = aio_open "$filename.pst~", O_WRONLY | O_CREAT, 0600) {
762 chmod SAVE_MODE, $fh; 775 chmod SAVE_MODE, $fh;
763 my $data = Storable::nfreeze { version => 1, objs => $objs }; 776 my $data = Storable::nfreeze { version => 1, objs => $objs };
764 aio_write $fh, 0, (length $data), $data, 0; 777 aio_write $fh, 0, (length $data), $data, 0;
765 aio_fsync $fh; 778 aio_fsync $fh if $cf::USE_FSYNC;
766 close $fh; 779 close $fh;
767 aio_rename "$filename.pst~", "$filename.pst"; 780 aio_rename "$filename.pst~", "$filename.pst";
768 } 781 }
769 } else { 782 } else {
770 aio_unlink "$filename.pst"; 783 aio_unlink "$filename.pst";
… …
1598 return if $self->players; 1611 return if $self->players;
1599 return if $self->isa ("ext::map_per_player");#d# 1612 return if $self->isa ("ext::map_per_player");#d#
1600 1613
1601 warn "resetting map ", $self->path;#d# 1614 warn "resetting map ", $self->path;#d#
1602 1615
1616 $self->in_memory (cf::MAP_SWAPPED);
1617
1618 # need to save uniques path
1619 unless ($self->{deny_save}) {
1620 my $uniq = $self->uniq_path; utf8::encode $uniq;
1621
1622 $self->_save_objects ($uniq, cf::IO_UNIQUES)
1623 if $uniq;
1624 }
1625
1603 delete $cf::MAP{$self->path}; 1626 delete $cf::MAP{$self->path};
1604 1627
1605 $self->in_memory (cf::MAP_SWAPPED);
1606 $self->clear; 1628 $self->clear;
1607 1629
1608 $_->clear_links_to ($self) for values %cf::MAP; 1630 $_->clear_links_to ($self) for values %cf::MAP;
1609 1631
1610 $self->unlink_save; 1632 $self->unlink_save;
… …
2174The parameter C<$family> should best start with the name of the extension 2196The parameter C<$family> should best start with the name of the extension
2175using it, it should be unique. 2197using it, it should be unique.
2176 2198
2177=over 4 2199=over 4
2178 2200
2179=item $hashref = cf::db_get $family
2180
2181Return a hashref for use by the extension C<$family>, which can be
2182modified. After modifications, you have to call C<cf::db_dirty> or
2183C<cf::db_sync>.
2184
2185=item $value = cf::db_get $family => $key 2201=item $value = cf::db_get $family => $key
2186 2202
2187Returns a single value from the database 2203Returns a single value from the database.
2188
2189=item cf::db_put $family => $hashref
2190
2191Stores the given family hashref into the database. Updates are delayed, if
2192you want the data to be synced to disk immediately, use C<cf::db_sync>.
2193 2204
2194=item cf::db_put $family => $key => $value 2205=item cf::db_put $family => $key => $value
2195 2206
2196Stores the given C<$value> in the family hash. Updates are delayed, if you 2207Stores the given C<$value> in the family.
2197want the data to be synced to disk immediately, use C<cf::db_sync>.
2198
2199=item cf::db_dirty
2200
2201Marks the database as dirty, to be updated at a later time.
2202
2203=item cf::db_sync
2204
2205Immediately write the database to disk I<if it is dirty>.
2206 2208
2207=cut 2209=cut
2208 2210
2209our $DB; 2211our $DB;
2210 2212
2211{ 2213sub db_init {
2214 unless ($DB) {
2215 $DB = BDB::db_create $DB_ENV;
2216
2217 cf::sync_job {
2218 eval {
2219 $DB->set_flags (BDB::CHKSUM);
2220
2221 BDB::db_open $DB, undef, "db", undef, BDB::BTREE,
2222 BDB::CREATE | BDB::AUTO_COMMIT, 0666;
2223 cf::cleanup "db_open(db): $!" if $!;
2224 };
2225 cf::cleanup "db_open(db): $@" if $@;
2226 };
2227
2212 my $path = cf::localdir . "/database.pst"; 2228 my $path = cf::localdir . "/database.pst";
2229 if (stat $path) {
2230 cf::sync_job {
2231 my $pst = Storable::retrieve $path;
2213 2232
2214 sub db_load() { 2233 cf::db_put (board => data => $pst->{board});
2215 $DB = stat $path ? Storable::retrieve $path : { }; 2234 cf::db_put (guildrules => data => $pst->{guildrules});
2216 } 2235 cf::db_put (rent => balance => $pst->{rent}{balance});
2236 BDB::db_env_txn_checkpoint $DB_ENV;
2217 2237
2218 my $pid; 2238 unlink $path;
2219
2220 sub db_save() {
2221 waitpid $pid, 0 if $pid;
2222 if (0 == ($pid = fork)) {
2223 $DB->{_meta}{version} = 1;
2224 Storable::nstore $DB, "$path~";
2225 rename "$path~", $path;
2226 cf::_exit 0 if defined $pid;
2227 } 2239 };
2228 }
2229
2230 my $dirty;
2231
2232 sub db_sync() {
2233 db_save if $dirty;
2234 undef $dirty;
2235 }
2236
2237 my $idle = Event->idle (
2238 reentrant => 0,
2239 min => 10,
2240 max => 20,
2241 repeat => 0,
2242 data => WF_AUTOCANCEL,
2243 cb => \&db_sync,
2244 );
2245
2246 sub db_dirty() {
2247 $dirty = 1;
2248 $idle->start;
2249 }
2250
2251 sub db_get($;$) {
2252 @_ >= 2
2253 ? $DB->{$_[0]}{$_[1]}
2254 : ($DB->{$_[0]} ||= { })
2255 }
2256
2257 sub db_put($$;$) {
2258 if (@_ >= 3) {
2259 $DB->{$_[0]}{$_[1]} = $_[2];
2260 } else {
2261 $DB->{$_[0]} = $_[1];
2262 }
2263 db_dirty;
2264 }
2265
2266 cf::global->attach (
2267 prio => 10000,
2268 on_cleanup => sub {
2269 db_sync;
2270 }, 2240 }
2271 ); 2241 }
2242}
2243
2244sub db_get($$) {
2245 my $key = "$_[0]/$_[1]";
2246
2247 cf::sync_job {
2248 BDB::db_get $DB, undef, $key, my $data;
2249
2250 $! ? ()
2251 : Compress::LZF::sthaw $data
2252 }
2253}
2254
2255sub db_put($$$) {
2256 BDB::dbreq_pri 4;
2257 BDB::db_put $DB, undef, "$_[0]/$_[1]", Compress::LZF::sfreeze_cr $_[2], 0, sub { };
2272} 2258}
2273 2259
2274############################################################################# 2260#############################################################################
2275# the server's main() 2261# the server's init and main functions
2262
2263sub load_resources {
2264 load_regions sprintf "%s/%s/regions", cf::datadir, cf::mapdir
2265 or die "unable to load regions file\n";#d#
2266}
2276 2267
2277sub cfg_load { 2268sub cfg_load {
2278 open my $fh, "<:utf8", cf::confdir . "/config" 2269 open my $fh, "<:utf8", cf::confdir . "/config"
2279 or return; 2270 or return;
2280 2271
… …
2291 $CFG{mlockall} ? eval "mlockall()" : eval "munlockall()" 2282 $CFG{mlockall} ? eval "mlockall()" : eval "munlockall()"
2292 and die "WARNING: m(un)lockall failed: $!\n"; 2283 and die "WARNING: m(un)lockall failed: $!\n";
2293 }; 2284 };
2294 warn $@ if $@; 2285 warn $@ if $@;
2295 } 2286 }
2287}
2288
2289sub init {
2290 load_resources;
2296} 2291}
2297 2292
2298sub main { 2293sub main {
2299 # we must not ever block the main coroutine 2294 # we must not ever block the main coroutine
2300 local $Coro::idle = sub { 2295 local $Coro::idle = sub {
… …
2303 Event::one_event; 2298 Event::one_event;
2304 })->prio (Coro::PRIO_MAX); 2299 })->prio (Coro::PRIO_MAX);
2305 }; 2300 };
2306 2301
2307 cfg_load; 2302 cfg_load;
2308 db_load; 2303 db_init;
2309 load_extensions; 2304 load_extensions;
2310 2305
2311 $TICK_WATCHER->start; 2306 $TICK_WATCHER->start;
2312 Event::loop; 2307 Event::loop;
2313} 2308}
… …
2352 my $map = $cf::MAP{$path} or next; 2347 my $map = $cf::MAP{$path} or next;
2353 $map->valid or next; 2348 $map->valid or next;
2354 $map->save; 2349 $map->save;
2355 } 2350 }
2356 warn "end emergency map save\n"; 2351 warn "end emergency map save\n";
2352
2353 warn "begin emergency database checkpoint\n";
2354 BDB::db_env_txn_checkpoint $DB_ENV;
2355 warn "end emergency database checkpoint\n";
2357 }; 2356 };
2358 2357
2359 warn "leave emergency perl save\n"; 2358 warn "leave emergency perl save\n";
2359}
2360
2361sub post_cleanup {
2362 my ($make_core) = @_;
2363
2364 warn Carp::longmess "post_cleanup backtrace"
2365 if $make_core;
2360} 2366}
2361 2367
2362sub reload() { 2368sub reload() {
2363 # can/must only be called in main 2369 # can/must only be called in main
2364 if ($Coro::current != $Coro::main) { 2370 if ($Coro::current != $Coro::main) {
… …
2374 cf::emergency_save; 2380 cf::emergency_save;
2375 2381
2376 eval { 2382 eval {
2377 # if anything goes wrong in here, we should simply crash as we already saved 2383 # if anything goes wrong in here, we should simply crash as we already saved
2378 2384
2379 warn "syncing database to disk";
2380 cf::db_sync;
2381
2382 warn "cancelling all WF_AUTOCANCEL watchers"; 2385 warn "cancelling all WF_AUTOCANCEL watchers";
2383 for (Event::all_watchers) { 2386 for (Event::all_watchers) {
2384 $_->cancel if $_->data & WF_AUTOCANCEL; 2387 $_->cancel if $_->data & WF_AUTOCANCEL;
2385 } 2388 }
2386 2389
2390 warn "syncing database to disk";
2391 BDB::db_env_txn_checkpoint $DB_ENV, 0, 0, 0, sub { };
2392
2387 warn "flushing outstanding aio requests"; 2393 warn "flushing outstanding aio requests";
2388 for (;;) { 2394 for (;;) {
2395 BDB::flush;
2389 IO::AIO::flush; 2396 IO::AIO::flush;
2390 Coro::cede; 2397 Coro::cede;
2391 last unless IO::AIO::nreqs; 2398 last unless IO::AIO::nreqs || BDB::nreqs;
2392 warn "iterate..."; 2399 warn "iterate...";
2393 } 2400 }
2394 2401
2395 warn "cancelling all extension coros"; 2402 warn "cancelling all extension coros";
2396 $_->cancel for values %EXT_CORO; 2403 $_->cancel for values %EXT_CORO;
… …
2451 require cf; 2458 require cf;
2452 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt 2459 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt
2453 2460
2454 warn "loading config and database again"; 2461 warn "loading config and database again";
2455 cf::cfg_load; 2462 cf::cfg_load;
2456 cf::db_load;
2457 2463
2458 warn "loading extensions"; 2464 warn "loading extensions";
2459 cf::load_extensions; 2465 cf::load_extensions;
2460 2466
2461 warn "reattaching attachments to objects/players"; 2467 warn "reattaching attachments to objects/players";
2462 _global_reattach; 2468 _global_reattach;
2463 warn "reattaching attachments to maps"; 2469 warn "reattaching attachments to maps";
2464 reattach $_ for values %MAP; 2470 reattach $_ for values %MAP;
2471
2472 warn "loading reloadable resources";
2473 load_resources;
2465 2474
2466 warn "restarting server ticker"; 2475 warn "restarting server ticker";
2467 2476
2468 $TICK_WATCHER->start; 2477 $TICK_WATCHER->start;
2469 }; 2478 };
… …
2538 $TICK_WATCHER->at ($NEXT_TICK); 2547 $TICK_WATCHER->at ($NEXT_TICK);
2539 $TICK_WATCHER->start; 2548 $TICK_WATCHER->start;
2540 }, 2549 },
2541); 2550);
2542 2551
2552{
2553 BDB::max_poll_time $TICK * 0.1;
2554 $BDB_POLL_WATCHER = Event->io (
2555 reentrant => 0,
2556 fd => BDB::poll_fileno,
2557 poll => 'r',
2558 prio => 0,
2559 data => WF_AUTOCANCEL,
2560 cb => \&BDB::poll_cb,
2561 );
2562 BDB::min_parallel 8;
2563
2564 BDB::set_sync_prepare {
2565 my $status;
2566 my $current = $Coro::current;
2567 (
2568 sub {
2569 $status = $!;
2570 $current->ready; undef $current;
2571 },
2572 sub {
2573 Coro::schedule while defined $current;
2574 $! = $status;
2575 },
2576 )
2577 };
2578
2579 unless ($DB_ENV) {
2580 $DB_ENV = BDB::db_env_create;
2581
2582 cf::sync_job {
2583 eval {
2584 BDB::db_env_open
2585 $DB_ENV,
2586 $BDB_ENV_DIR,
2587 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN
2588 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE,
2589 0666;
2590
2591 cf::cleanup "db_env_open($BDB_ENV_DIR): $!" if $!;
2592
2593 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1);
2594 $DB_ENV->set_lk_detect;
2595 };
2596
2597 cf::cleanup "db_env_open(db): $@" if $@;
2598 };
2599 }
2600}
2601
2602{
2603 IO::AIO::min_parallel 8;
2604
2605 undef $Coro::AIO::WATCHER;
2543IO::AIO::max_poll_time $TICK * 0.1; 2606 IO::AIO::max_poll_time $TICK * 0.1;
2544
2545undef $Coro::AIO::WATCHER;
2546$AIO_POLL_WATCHER = Event->io ( 2607 $AIO_POLL_WATCHER = Event->io (
2547 reentrant => 0, 2608 reentrant => 0,
2548 fd => IO::AIO::poll_fileno, 2609 fd => IO::AIO::poll_fileno,
2549 poll => 'r', 2610 poll => 'r',
2550 prio => 6, 2611 prio => 6,
2551 data => WF_AUTOCANCEL, 2612 data => WF_AUTOCANCEL,
2552 cb => \&IO::AIO::poll_cb, 2613 cb => \&IO::AIO::poll_cb,
2553); 2614 );
2615}
2554 2616
2555$WRITE_RUNTIME_WATCHER = Event->timer ( 2617$WRITE_RUNTIME_WATCHER = Event->timer (
2556 reentrant => 0, 2618 reentrant => 0,
2557 data => WF_AUTOCANCEL, 2619 data => WF_AUTOCANCEL,
2558 after => 1, 2620 after => 1,

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines