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.204 by root, Fri Feb 2 19:59:22 2007 UTC vs.
Revision 1.213 by root, Tue Feb 13 20:01:04 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;
53our $USE_FSYNC; # use fsync to write maps - default off 57our $USE_FSYNC = 1; # use fsync to write maps - default off
58
59our $BDB_POLL_WATCHER;
60our $DB_ENV;
54 61
55our %CFG; 62our %CFG;
56 63
57our $UPTIME; $UPTIME ||= time; 64our $UPTIME; $UPTIME ||= time;
58our $RUNTIME; 65our $RUNTIME;
59 66
60our %PLAYER; # all users 67our %PLAYER; # all users
61our %MAP; # all maps 68our %MAP; # all maps
62our $LINK_MAP; # the special {link} map, which is always available 69our $LINK_MAP; # the special {link} map, which is always available
63our $RANDOM_MAPS = cf::localdir . "/random"; 70our $RANDOM_MAPS = cf::localdir . "/random";
71our $BDB_ENV_DIR = cf::localdir . "/db";
64 72
65our $WAIT_FOR_TICK; $WAIT_FOR_TICK ||= new Coro::Signal; 73our $WAIT_FOR_TICK; $WAIT_FOR_TICK ||= new Coro::Signal;
66our $WAIT_FOR_TICK_ONE; $WAIT_FOR_TICK_ONE ||= new Coro::Signal; 74our $WAIT_FOR_TICK_ONE; $WAIT_FOR_TICK_ONE ||= new Coro::Signal;
67 75
68# 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 ∕
81mkdir cf::localdir; 89mkdir cf::localdir;
82mkdir cf::localdir . "/" . cf::playerdir; 90mkdir cf::localdir . "/" . cf::playerdir;
83mkdir cf::localdir . "/" . cf::tmpdir; 91mkdir cf::localdir . "/" . cf::tmpdir;
84mkdir cf::localdir . "/" . cf::uniquedir; 92mkdir cf::localdir . "/" . cf::uniquedir;
85mkdir $RANDOM_MAPS; 93mkdir $RANDOM_MAPS;
94mkdir $BDB_ENV_DIR;
86 95
87our $EMERGENCY_POSITION; 96our $EMERGENCY_POSITION;
88 97
89sub cf::map::normalise; 98sub cf::map::normalise;
90 99
1600 return if $self->players; 1609 return if $self->players;
1601 return if $self->isa ("ext::map_per_player");#d# 1610 return if $self->isa ("ext::map_per_player");#d#
1602 1611
1603 warn "resetting map ", $self->path;#d# 1612 warn "resetting map ", $self->path;#d#
1604 1613
1614 $self->in_memory (cf::MAP_SWAPPED);
1615
1616 # need to save uniques path
1617 unless ($self->{deny_save}) {
1618 my $uniq = $self->uniq_path; utf8::encode $uniq;
1619
1620 $self->_save_objects ($uniq, cf::IO_UNIQUES)
1621 if $uniq;
1622 }
1623
1605 delete $cf::MAP{$self->path}; 1624 delete $cf::MAP{$self->path};
1606 1625
1607 $self->in_memory (cf::MAP_SWAPPED);
1608 $self->clear; 1626 $self->clear;
1609 1627
1610 $_->clear_links_to ($self) for values %cf::MAP; 1628 $_->clear_links_to ($self) for values %cf::MAP;
1611 1629
1612 $self->unlink_save; 1630 $self->unlink_save;
2176The parameter C<$family> should best start with the name of the extension 2194The parameter C<$family> should best start with the name of the extension
2177using it, it should be unique. 2195using it, it should be unique.
2178 2196
2179=over 4 2197=over 4
2180 2198
2181=item $hashref = cf::db_get $family
2182
2183Return a hashref for use by the extension C<$family>, which can be
2184modified. After modifications, you have to call C<cf::db_dirty> or
2185C<cf::db_sync>.
2186
2187=item $value = cf::db_get $family => $key 2199=item $value = cf::db_get $family => $key
2188 2200
2189Returns a single value from the database 2201Returns a single value from the database.
2190
2191=item cf::db_put $family => $hashref
2192
2193Stores the given family hashref into the database. Updates are delayed, if
2194you want the data to be synced to disk immediately, use C<cf::db_sync>.
2195 2202
2196=item cf::db_put $family => $key => $value 2203=item cf::db_put $family => $key => $value
2197 2204
2198Stores the given C<$value> in the family hash. Updates are delayed, if you 2205Stores the given C<$value> in the family.
2199want the data to be synced to disk immediately, use C<cf::db_sync>.
2200
2201=item cf::db_dirty
2202
2203Marks the database as dirty, to be updated at a later time.
2204
2205=item cf::db_sync
2206
2207Immediately write the database to disk I<if it is dirty>.
2208 2206
2209=cut 2207=cut
2210 2208
2211our $DB; 2209our $DB;
2212 2210
2213{ 2211sub db_init {
2212 unless ($DB) {
2213 $DB = BDB::db_create $DB_ENV;
2214
2215 cf::sync_job {
2216 eval {
2217 $DB->set_flags (BDB::CHKSUM);
2218
2219 BDB::db_open $DB, undef, "db", undef, BDB::BTREE,
2220 BDB::CREATE | BDB::AUTO_COMMIT, 0666;
2221 cf::cleanup "db_open(db): $!" if $!;
2222 };
2223 cf::cleanup "db_open(db): $@" if $@;
2224 };
2225
2214 my $path = cf::localdir . "/database.pst"; 2226 my $path = cf::localdir . "/database.pst";
2227 if (stat $path) {
2228 cf::sync_job {
2229 my $pst = Storable::retrieve $path;
2215 2230
2216 sub db_load() { 2231 cf::db_put (board => data => $pst->{board});
2217 $DB = stat $path ? Storable::retrieve $path : { }; 2232 cf::db_put (guildrules => data => $pst->{guildrules});
2218 } 2233 cf::db_put (rent => balance => $pst->{rent}{balance});
2234 BDB::db_env_txn_checkpoint $DB_ENV;
2219 2235
2220 my $pid; 2236 unlink $path;
2221
2222 sub db_save() {
2223 waitpid $pid, 0 if $pid;
2224 if (0 == ($pid = fork)) {
2225 $DB->{_meta}{version} = 1;
2226 Storable::nstore $DB, "$path~";
2227 rename "$path~", $path;
2228 cf::_exit 0 if defined $pid;
2229 } 2237 };
2230 }
2231
2232 my $dirty;
2233
2234 sub db_sync() {
2235 db_save if $dirty;
2236 undef $dirty;
2237 }
2238
2239 my $idle = Event->idle (
2240 reentrant => 0,
2241 min => 10,
2242 max => 20,
2243 repeat => 0,
2244 data => WF_AUTOCANCEL,
2245 cb => \&db_sync,
2246 );
2247
2248 sub db_dirty() {
2249 $dirty = 1;
2250 $idle->start;
2251 }
2252
2253 sub db_get($;$) {
2254 @_ >= 2
2255 ? $DB->{$_[0]}{$_[1]}
2256 : ($DB->{$_[0]} ||= { })
2257 }
2258
2259 sub db_put($$;$) {
2260 if (@_ >= 3) {
2261 $DB->{$_[0]}{$_[1]} = $_[2];
2262 } else {
2263 $DB->{$_[0]} = $_[1];
2264 }
2265 db_dirty;
2266 }
2267
2268 cf::global->attach (
2269 prio => 10000,
2270 on_cleanup => sub {
2271 db_sync;
2272 }, 2238 }
2273 ); 2239 }
2240}
2241
2242sub db_get($$) {
2243 my $key = "$_[0]/$_[1]";
2244
2245 cf::sync_job {
2246 BDB::db_get $DB, undef, $key, my $data;
2247
2248 $! ? ()
2249 : Compress::LZF::sthaw $data
2250 }
2251}
2252
2253sub db_put($$$) {
2254 BDB::dbreq_pri 4;
2255 BDB::db_put $DB, undef, "$_[0]/$_[1]", Compress::LZF::sfreeze_cr $_[2], 0, sub { };
2274} 2256}
2275 2257
2276############################################################################# 2258#############################################################################
2277# the server's init and main functions 2259# the server's init and main functions
2278 2260
2314 Event::one_event; 2296 Event::one_event;
2315 })->prio (Coro::PRIO_MAX); 2297 })->prio (Coro::PRIO_MAX);
2316 }; 2298 };
2317 2299
2318 cfg_load; 2300 cfg_load;
2319 db_load; 2301 db_init;
2320 load_extensions; 2302 load_extensions;
2321 2303
2322 $TICK_WATCHER->start; 2304 $TICK_WATCHER->start;
2305 $WRITE_RUNTIME_WATCHER->start;
2323 Event::loop; 2306 Event::loop;
2324} 2307}
2325 2308
2326############################################################################# 2309#############################################################################
2327# initialisation and cleanup 2310# initialisation and cleanup
2363 my $map = $cf::MAP{$path} or next; 2346 my $map = $cf::MAP{$path} or next;
2364 $map->valid or next; 2347 $map->valid or next;
2365 $map->save; 2348 $map->save;
2366 } 2349 }
2367 warn "end emergency map save\n"; 2350 warn "end emergency map save\n";
2351
2352 warn "begin emergency database checkpoint\n";
2353 BDB::db_env_txn_checkpoint $DB_ENV;
2354 warn "end emergency database checkpoint\n";
2368 }; 2355 };
2369 2356
2370 warn "leave emergency perl save\n"; 2357 warn "leave emergency perl save\n";
2358}
2359
2360sub post_cleanup {
2361 my ($make_core) = @_;
2362
2363 warn Carp::longmess "post_cleanup backtrace"
2364 if $make_core;
2371} 2365}
2372 2366
2373sub reload() { 2367sub reload() {
2374 # can/must only be called in main 2368 # can/must only be called in main
2375 if ($Coro::current != $Coro::main) { 2369 if ($Coro::current != $Coro::main) {
2377 return; 2371 return;
2378 } 2372 }
2379 2373
2380 warn "reloading..."; 2374 warn "reloading...";
2381 2375
2382 warn "cancelling server ticker"; 2376 warn "entering sync_job";
2383 $TICK_WATCHER->cancel;
2384 2377
2378 cf::sync_job {
2379 cf::write_runtime;
2385 cf::emergency_save; 2380 cf::emergency_save;
2381 cf::write_runtime;
2386 2382
2387 eval { 2383 warn "syncing database to disk";
2384 BDB::db_env_txn_checkpoint $DB_ENV;
2385
2388 # if anything goes wrong in here, we should simply crash as we already saved 2386 # if anything goes wrong in here, we should simply crash as we already saved
2389
2390 warn "syncing database to disk";
2391 cf::db_sync;
2392 2387
2393 warn "cancelling all WF_AUTOCANCEL watchers"; 2388 warn "cancelling all WF_AUTOCANCEL watchers";
2394 for (Event::all_watchers) { 2389 for (Event::all_watchers) {
2395 $_->cancel if $_->data & WF_AUTOCANCEL; 2390 $_->cancel if $_->data & WF_AUTOCANCEL;
2396 } 2391 }
2397 2392
2398 warn "flushing outstanding aio requests"; 2393 warn "flushing outstanding aio requests";
2399 for (;;) { 2394 for (;;) {
2395 BDB::flush;
2400 IO::AIO::flush; 2396 IO::AIO::flush;
2401 Coro::cede; 2397 Coro::cede;
2402 last unless IO::AIO::nreqs; 2398 last unless IO::AIO::nreqs || BDB::nreqs;
2403 warn "iterate..."; 2399 warn "iterate...";
2404 } 2400 }
2405 2401
2406 warn "cancelling all extension coros"; 2402 warn "cancelling all extension coros";
2407 $_->cancel for values %EXT_CORO; 2403 $_->cancel for values %EXT_CORO;
2462 require cf; 2458 require cf;
2463 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt 2459 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt
2464 2460
2465 warn "loading config and database again"; 2461 warn "loading config and database again";
2466 cf::cfg_load; 2462 cf::cfg_load;
2467 cf::db_load;
2468 2463
2469 warn "loading extensions"; 2464 warn "loading extensions";
2470 cf::load_extensions; 2465 cf::load_extensions;
2471 2466
2472 warn "reattaching attachments to objects/players"; 2467 warn "reattaching attachments to objects/players";
2475 reattach $_ for values %MAP; 2470 reattach $_ for values %MAP;
2476 2471
2477 warn "loading reloadable resources"; 2472 warn "loading reloadable resources";
2478 load_resources; 2473 load_resources;
2479 2474
2480 warn "restarting server ticker"; 2475 warn "leaving sync_job";
2481 2476
2482 $TICK_WATCHER->start; 2477 1
2483 }; 2478 } or do {
2484
2485 if ($@) {
2486 warn $@; 2479 warn $@;
2487 warn "error while reloading, exiting."; 2480 warn "error while reloading, exiting.";
2488 exit 1; 2481 exit 1;
2489 } 2482 };
2490 2483
2491 warn "reloaded"; 2484 warn "reloaded";
2492}; 2485};
2493 2486
2494our $RELOAD_WATCHER; # used only during reload 2487our $RELOAD_WATCHER; # used only during reload
2538 $NEXT_TICK += $TICK; 2531 $NEXT_TICK += $TICK;
2539 2532
2540 $WAIT_FOR_TICK->broadcast; 2533 $WAIT_FOR_TICK->broadcast;
2541 $WAIT_FOR_TICK_ONE->send if $WAIT_FOR_TICK_ONE->awaited; 2534 $WAIT_FOR_TICK_ONE->send if $WAIT_FOR_TICK_ONE->awaited;
2542 2535
2543 Event::sweep;
2544 Coro::cede_notself;
2545
2546# my $AFTER = Event::time; 2536# my $AFTER = Event::time;
2547# warn $AFTER - $NOW;#d# 2537# warn $AFTER - $NOW;#d#
2548 2538
2549 # if we are delayed by four ticks or more, skip them all 2539 # if we are delayed by four ticks or more, skip them all
2550 $NEXT_TICK = Event::time if Event::time >= $NEXT_TICK + $TICK * 4; 2540 $NEXT_TICK = Event::time if Event::time >= $NEXT_TICK + $TICK * 4;
2552 $TICK_WATCHER->at ($NEXT_TICK); 2542 $TICK_WATCHER->at ($NEXT_TICK);
2553 $TICK_WATCHER->start; 2543 $TICK_WATCHER->start;
2554 }, 2544 },
2555); 2545);
2556 2546
2547{
2548 BDB::max_poll_time $TICK * 0.1;
2549 $BDB_POLL_WATCHER = Event->io (
2550 reentrant => 0,
2551 fd => BDB::poll_fileno,
2552 poll => 'r',
2553 prio => 0,
2554 data => WF_AUTOCANCEL,
2555 cb => \&BDB::poll_cb,
2556 );
2557 BDB::min_parallel 8;
2558
2559 BDB::set_sync_prepare {
2560 my $status;
2561 my $current = $Coro::current;
2562 (
2563 sub {
2564 $status = $!;
2565 $current->ready; undef $current;
2566 },
2567 sub {
2568 Coro::schedule while defined $current;
2569 $! = $status;
2570 },
2571 )
2572 };
2573
2574 unless ($DB_ENV) {
2575 $DB_ENV = BDB::db_env_create;
2576
2577 cf::sync_job {
2578 eval {
2579 BDB::db_env_open
2580 $DB_ENV,
2581 $BDB_ENV_DIR,
2582 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN
2583 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE,
2584 0666;
2585
2586 cf::cleanup "db_env_open($BDB_ENV_DIR): $!" if $!;
2587
2588 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1);
2589 $DB_ENV->set_lk_detect;
2590 };
2591
2592 cf::cleanup "db_env_open(db): $@" if $@;
2593 };
2594 }
2595}
2596
2597{
2598 IO::AIO::min_parallel 8;
2599
2600 undef $Coro::AIO::WATCHER;
2557IO::AIO::max_poll_time $TICK * 0.1; 2601 IO::AIO::max_poll_time $TICK * 0.1;
2558
2559undef $Coro::AIO::WATCHER;
2560$AIO_POLL_WATCHER = Event->io ( 2602 $AIO_POLL_WATCHER = Event->io (
2561 reentrant => 0, 2603 reentrant => 0,
2562 fd => IO::AIO::poll_fileno, 2604 fd => IO::AIO::poll_fileno,
2563 poll => 'r', 2605 poll => 'r',
2564 prio => 6, 2606 prio => 6,
2565 data => WF_AUTOCANCEL, 2607 data => WF_AUTOCANCEL,
2566 cb => \&IO::AIO::poll_cb, 2608 cb => \&IO::AIO::poll_cb,
2567); 2609 );
2610}
2568 2611
2569$WRITE_RUNTIME_WATCHER = Event->timer ( 2612$WRITE_RUNTIME_WATCHER = Event->timer (
2570 reentrant => 0, 2613 reentrant => 0,
2571 data => WF_AUTOCANCEL, 2614 data => WF_AUTOCANCEL,
2615 parked => 1,
2572 after => 1, 2616 after => 1,
2573 interval => 10, 2617 interval => 10,
2574 prio => 6, # keep it lowest so it acts like a watchdog 2618 prio => 6, # keep it lowest so it acts like a watchdog
2575 cb => Coro::unblock_sub { 2619 cb => sub {
2620 $TICK_WATCHER->is_active or cf::cleanup "mainloop frozen but runtime active", 1;
2621
2622 Coro::async_pool {
2576 write_runtime 2623 write_runtime
2577 or warn "ERROR: unable to write runtime file: $!"; 2624 or warn "ERROR: unable to write runtime file: $!";
2625 };
2578 }, 2626 },
2579); 2627);
2580 2628
2581END { cf::emergency_save } 2629END { cf::emergency_save }
2582 2630

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines