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.201 by root, Mon Jan 29 18:15:23 2007 UTC vs.
Revision 1.208 by root, Mon Feb 12 00:27:20 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
… …
358 # 60 allows for the watchdog to kill the server. 368 # 60 allows for the watchdog to kill the server.
359 369
360 (aio_write $fh, 0, (length $value), $value, 0) <= 0 370 (aio_write $fh, 0, (length $value), $value, 0) <= 0
361 and return; 371 and return;
362 372
373 # always fsync - this file is important
363 aio_fsync $fh 374 aio_fsync $fh
364 and return; 375 and return;
365 376
366 close $fh 377 close $fh
367 or return; 378 or return;
… …
752 $filename, length $$rdata, scalar @$objs; 763 $filename, length $$rdata, scalar @$objs;
753 764
754 if (my $fh = aio_open "$filename~", O_WRONLY | O_CREAT, 0600) { 765 if (my $fh = aio_open "$filename~", O_WRONLY | O_CREAT, 0600) {
755 chmod SAVE_MODE, $fh; 766 chmod SAVE_MODE, $fh;
756 aio_write $fh, 0, (length $$rdata), $$rdata, 0; 767 aio_write $fh, 0, (length $$rdata), $$rdata, 0;
757 aio_fsync $fh; 768 aio_fsync $fh if $cf::USE_FSYNC;
758 close $fh; 769 close $fh;
759 770
760 if (@$objs) { 771 if (@$objs) {
761 if (my $fh = aio_open "$filename.pst~", O_WRONLY | O_CREAT, 0600) { 772 if (my $fh = aio_open "$filename.pst~", O_WRONLY | O_CREAT, 0600) {
762 chmod SAVE_MODE, $fh; 773 chmod SAVE_MODE, $fh;
763 my $data = Storable::nfreeze { version => 1, objs => $objs }; 774 my $data = Storable::nfreeze { version => 1, objs => $objs };
764 aio_write $fh, 0, (length $data), $data, 0; 775 aio_write $fh, 0, (length $data), $data, 0;
765 aio_fsync $fh; 776 aio_fsync $fh if $cf::USE_FSYNC;
766 close $fh; 777 close $fh;
767 aio_rename "$filename.pst~", "$filename.pst"; 778 aio_rename "$filename.pst~", "$filename.pst";
768 } 779 }
769 } else { 780 } else {
770 aio_unlink "$filename.pst"; 781 aio_unlink "$filename.pst";
… …
2003 # get handled first 2014 # get handled first
2004 my $queue = delete $ns->{query_queue} 2015 my $queue = delete $ns->{query_queue}
2005 or return; # be conservative, not sure how that can happen, but we saw a crash here 2016 or return; # be conservative, not sure how that can happen, but we saw a crash here
2006 2017
2007 (shift @$queue)->[1]->($msg); 2018 (shift @$queue)->[1]->($msg);
2019 return unless $ns->valid; # temporary(?) workaround for callback destroying socket
2008 2020
2009 push @{ $ns->{query_queue} }, @$queue; 2021 push @{ $ns->{query_queue} }, @$queue;
2010 2022
2011 if (@{ $ns->{query_queue} } == @$queue) { 2023 if (@{ $ns->{query_queue} } == @$queue) {
2012 if (@$queue) { 2024 if (@$queue) {
… …
2173The parameter C<$family> should best start with the name of the extension 2185The parameter C<$family> should best start with the name of the extension
2174using it, it should be unique. 2186using it, it should be unique.
2175 2187
2176=over 4 2188=over 4
2177 2189
2178=item $hashref = cf::db_get $family
2179
2180Return a hashref for use by the extension C<$family>, which can be
2181modified. After modifications, you have to call C<cf::db_dirty> or
2182C<cf::db_sync>.
2183
2184=item $value = cf::db_get $family => $key 2190=item $value = cf::db_get $family => $key
2185 2191
2186Returns a single value from the database 2192Returns a single value from the database.
2187
2188=item cf::db_put $family => $hashref
2189
2190Stores the given family hashref into the database. Updates are delayed, if
2191you want the data to be synced to disk immediately, use C<cf::db_sync>.
2192 2193
2193=item cf::db_put $family => $key => $value 2194=item cf::db_put $family => $key => $value
2194 2195
2195Stores the given C<$value> in the family hash. Updates are delayed, if you 2196Stores the given C<$value> in the family.
2196want the data to be synced to disk immediately, use C<cf::db_sync>.
2197
2198=item cf::db_dirty
2199
2200Marks the database as dirty, to be updated at a later time.
2201
2202=item cf::db_sync
2203
2204Immediately write the database to disk I<if it is dirty>.
2205 2197
2206=cut 2198=cut
2207 2199
2208our $DB; 2200our $DB;
2209 2201
2210{ 2202sub db_load() {
2203 unless ($DB) {
2204 $DB = BDB::db_create $DB_ENV;
2205
2206 cf::sync_job {
2207 eval {
2208 $DB->set_flags (BDB::CHKSUM);
2209
2210 BDB::db_open $DB, undef, "db", undef, BDB::BTREE,
2211 BDB::CREATE | BDB::AUTO_COMMIT, 0666;
2212 cf::cleanup "db_open(db): $!" if $!;
2213 };
2214 cf::cleanup "db_open(db): $@" if $@;
2215 };
2216
2211 my $path = cf::localdir . "/database.pst"; 2217 my $path = cf::localdir . "/database.pst";
2218 if (stat $path) {
2219 cf::sync_job {
2220 my $pst = Storable::retrieve $path;
2212 2221
2213 sub db_load() { 2222 cf::db_put (board => data => $pst->{board});
2214 $DB = stat $path ? Storable::retrieve $path : { }; 2223 cf::db_put (guildrules => data => $pst->{guildrules});
2215 } 2224 cf::db_put (rent => balance => $pst->{rent}{balance});
2225 BDB::db_env_txn_checkpoint $DB_ENV;
2216 2226
2217 my $pid; 2227 unlink $path;
2218
2219 sub db_save() {
2220 waitpid $pid, 0 if $pid;
2221 if (0 == ($pid = fork)) {
2222 $DB->{_meta}{version} = 1;
2223 Storable::nstore $DB, "$path~";
2224 rename "$path~", $path;
2225 cf::_exit 0 if defined $pid;
2226 } 2228 };
2227 }
2228
2229 my $dirty;
2230
2231 sub db_sync() {
2232 db_save if $dirty;
2233 undef $dirty;
2234 }
2235
2236 my $idle = Event->idle (
2237 reentrant => 0,
2238 min => 10,
2239 max => 20,
2240 repeat => 0,
2241 data => WF_AUTOCANCEL,
2242 cb => \&db_sync,
2243 );
2244
2245 sub db_dirty() {
2246 $dirty = 1;
2247 $idle->start;
2248 }
2249
2250 sub db_get($;$) {
2251 @_ >= 2
2252 ? $DB->{$_[0]}{$_[1]}
2253 : ($DB->{$_[0]} ||= { })
2254 }
2255
2256 sub db_put($$;$) {
2257 if (@_ >= 3) {
2258 $DB->{$_[0]}{$_[1]} = $_[2];
2259 } else {
2260 $DB->{$_[0]} = $_[1];
2261 }
2262 db_dirty;
2263 }
2264
2265 cf::global->attach (
2266 prio => 10000,
2267 on_cleanup => sub {
2268 db_sync;
2269 }, 2229 }
2270 ); 2230 }
2231}
2232
2233sub db_get($$) {
2234 my $key = "$_[0]/$_[1]";
2235
2236 cf::sync_job {
2237 BDB::db_get $DB, undef, $key, my $data;
2238
2239 $! ? ()
2240 : Compress::LZF::sthaw $data
2241 }
2242}
2243
2244sub db_put($$$) {
2245 BDB::dbreq_pri 4;
2246 BDB::db_put $DB, undef, "$_[0]/$_[1]", Compress::LZF::sfreeze_cr $_[2], 0, sub { };
2271} 2247}
2272 2248
2273############################################################################# 2249#############################################################################
2274# the server's main() 2250# the server's init and main functions
2251
2252sub load_resources {
2253 load_regions sprintf "%s/%s/regions", cf::datadir, cf::mapdir
2254 or die "unable to load regions file\n";#d#
2255}
2275 2256
2276sub cfg_load { 2257sub cfg_load {
2277 open my $fh, "<:utf8", cf::confdir . "/config" 2258 open my $fh, "<:utf8", cf::confdir . "/config"
2278 or return; 2259 or return;
2279 2260
… …
2290 $CFG{mlockall} ? eval "mlockall()" : eval "munlockall()" 2271 $CFG{mlockall} ? eval "mlockall()" : eval "munlockall()"
2291 and die "WARNING: m(un)lockall failed: $!\n"; 2272 and die "WARNING: m(un)lockall failed: $!\n";
2292 }; 2273 };
2293 warn $@ if $@; 2274 warn $@ if $@;
2294 } 2275 }
2276}
2277
2278sub init {
2279 load_resources;
2295} 2280}
2296 2281
2297sub main { 2282sub main {
2298 # we must not ever block the main coroutine 2283 # we must not ever block the main coroutine
2299 local $Coro::idle = sub { 2284 local $Coro::idle = sub {
… …
2351 my $map = $cf::MAP{$path} or next; 2336 my $map = $cf::MAP{$path} or next;
2352 $map->valid or next; 2337 $map->valid or next;
2353 $map->save; 2338 $map->save;
2354 } 2339 }
2355 warn "end emergency map save\n"; 2340 warn "end emergency map save\n";
2341
2342 warn "begin emergency database checkpoint\n";
2343 BDB::db_env_txn_checkpoint $DB_ENV;
2344 warn "end emergency database checkpoint\n";
2356 }; 2345 };
2357 2346
2358 warn "leave emergency perl save\n"; 2347 warn "leave emergency perl save\n";
2359} 2348}
2360 2349
… …
2373 cf::emergency_save; 2362 cf::emergency_save;
2374 2363
2375 eval { 2364 eval {
2376 # if anything goes wrong in here, we should simply crash as we already saved 2365 # if anything goes wrong in here, we should simply crash as we already saved
2377 2366
2378 warn "syncing database to disk";
2379 cf::db_sync;
2380
2381 warn "cancelling all WF_AUTOCANCEL watchers"; 2367 warn "cancelling all WF_AUTOCANCEL watchers";
2382 for (Event::all_watchers) { 2368 for (Event::all_watchers) {
2383 $_->cancel if $_->data & WF_AUTOCANCEL; 2369 $_->cancel if $_->data & WF_AUTOCANCEL;
2384 } 2370 }
2385 2371
2372 warn "syncing database to disk";
2373 BDB::db_env_txn_checkpoint $DB_ENV, 0, 0, 0, sub { };
2374
2386 warn "flushing outstanding aio requests"; 2375 warn "flushing outstanding aio requests";
2387 for (;;) { 2376 for (;;) {
2377 BDB::flush;
2388 IO::AIO::flush; 2378 IO::AIO::flush;
2389 Coro::cede; 2379 Coro::cede;
2390 last unless IO::AIO::nreqs; 2380 last unless IO::AIO::nreqs || BDB::nreqs;
2391 warn "iterate..."; 2381 warn "iterate...";
2392 } 2382 }
2393 2383
2394 warn "cancelling all extension coros"; 2384 warn "cancelling all extension coros";
2395 $_->cancel for values %EXT_CORO; 2385 $_->cancel for values %EXT_CORO;
… …
2459 2449
2460 warn "reattaching attachments to objects/players"; 2450 warn "reattaching attachments to objects/players";
2461 _global_reattach; 2451 _global_reattach;
2462 warn "reattaching attachments to maps"; 2452 warn "reattaching attachments to maps";
2463 reattach $_ for values %MAP; 2453 reattach $_ for values %MAP;
2454
2455 warn "loading reloadable resources";
2456 load_resources;
2464 2457
2465 warn "restarting server ticker"; 2458 warn "restarting server ticker";
2466 2459
2467 $TICK_WATCHER->start; 2460 $TICK_WATCHER->start;
2468 }; 2461 };
… …
2537 $TICK_WATCHER->at ($NEXT_TICK); 2530 $TICK_WATCHER->at ($NEXT_TICK);
2538 $TICK_WATCHER->start; 2531 $TICK_WATCHER->start;
2539 }, 2532 },
2540); 2533);
2541 2534
2535{
2536 BDB::max_poll_time $TICK * 0.1;
2537 $BDB_POLL_WATCHER = Event->io (
2538 reentrant => 0,
2539 fd => BDB::poll_fileno,
2540 poll => 'r',
2541 prio => 0,
2542 data => WF_AUTOCANCEL,
2543 cb => \&BDB::poll_cb,
2544 );
2545 BDB::min_parallel 8;
2546
2547 BDB::set_sync_prepare {
2548 my $status;
2549 my $current = $Coro::current;
2550 (
2551 sub {
2552 $status = $!;
2553 $current->ready; undef $current;
2554 },
2555 sub {
2556 Coro::schedule while defined $current;
2557 $! = $status;
2558 },
2559 )
2560 };
2561
2562 unless ($DB_ENV) {
2563 $DB_ENV = BDB::db_env_create;
2564
2565 cf::sync_job {
2566 eval {
2567 BDB::db_env_open
2568 $DB_ENV,
2569 $BDB_ENV_DIR,
2570 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN
2571 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE,
2572 0666;
2573
2574 cf::cleanup "db_env_open($BDB_ENV_DIR): $!" if $!;
2575
2576 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1);
2577 $DB_ENV->set_lk_detect;
2578 };
2579
2580 cf::cleanup "db_env_open(db): $@" if $@;
2581 };
2582 }
2583}
2584
2585{
2586 IO::AIO::min_parallel 8;
2587
2588 undef $Coro::AIO::WATCHER;
2542IO::AIO::max_poll_time $TICK * 0.1; 2589 IO::AIO::max_poll_time $TICK * 0.1;
2543
2544undef $Coro::AIO::WATCHER;
2545$AIO_POLL_WATCHER = Event->io ( 2590 $AIO_POLL_WATCHER = Event->io (
2546 reentrant => 0, 2591 reentrant => 0,
2547 fd => IO::AIO::poll_fileno, 2592 fd => IO::AIO::poll_fileno,
2548 poll => 'r', 2593 poll => 'r',
2549 prio => 6, 2594 prio => 6,
2550 data => WF_AUTOCANCEL, 2595 data => WF_AUTOCANCEL,
2551 cb => \&IO::AIO::poll_cb, 2596 cb => \&IO::AIO::poll_cb,
2552); 2597 );
2598}
2553 2599
2554$WRITE_RUNTIME_WATCHER = Event->timer ( 2600$WRITE_RUNTIME_WATCHER = Event->timer (
2555 reentrant => 0, 2601 reentrant => 0,
2556 data => WF_AUTOCANCEL, 2602 data => WF_AUTOCANCEL,
2557 after => 1, 2603 after => 1,

Diff Legend

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