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.200 by root, Mon Jan 29 17:57:22 2007 UTC vs.
Revision 1.210 by root, Mon Feb 12 01:32:49 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";
… …
1079=cut 1090=cut
1080 1091
1081sub maps($) { 1092sub maps($) {
1082 my ($pl) = @_; 1093 my ($pl) = @_;
1083 1094
1095 $pl = ref $pl ? $pl->ob->name : $pl;
1096
1084 my $files = aio_readdir playerdir $pl 1097 my $files = aio_readdir playerdir $pl
1085 or return; 1098 or return;
1086 1099
1087 my @paths; 1100 my @paths;
1088 1101
1089 for (@$files) { 1102 for (@$files) {
1090 utf8::decode $_; 1103 utf8::decode $_;
1091 next if /\.(?:pl|pst)$/; 1104 next if /\.(?:pl|pst)$/;
1092 next unless /^$PATH_SEP/o; 1105 next unless /^$PATH_SEP/o;
1093 1106
1094 push @paths, cf::map::normalise "~" . $pl->ob->name . "/" . $_; 1107 push @paths, cf::map::normalise "~$pl/$_";
1095 } 1108 }
1096 1109
1097 \@paths 1110 \@paths
1098} 1111}
1099 1112
… …
1596 return if $self->players; 1609 return if $self->players;
1597 return if $self->isa ("ext::map_per_player");#d# 1610 return if $self->isa ("ext::map_per_player");#d#
1598 1611
1599 warn "resetting map ", $self->path;#d# 1612 warn "resetting map ", $self->path;#d#
1600 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
1601 delete $cf::MAP{$self->path}; 1624 delete $cf::MAP{$self->path};
1602 1625
1603 $self->in_memory (cf::MAP_SWAPPED);
1604 $self->clear; 1626 $self->clear;
1605 1627
1606 $_->clear_links_to ($self) for values %cf::MAP; 1628 $_->clear_links_to ($self) for values %cf::MAP;
1607 1629
1608 $self->unlink_save; 1630 $self->unlink_save;
… …
2001 # get handled first 2023 # get handled first
2002 my $queue = delete $ns->{query_queue} 2024 my $queue = delete $ns->{query_queue}
2003 or return; # be conservative, not sure how that can happen, but we saw a crash here 2025 or return; # be conservative, not sure how that can happen, but we saw a crash here
2004 2026
2005 (shift @$queue)->[1]->($msg); 2027 (shift @$queue)->[1]->($msg);
2028 return unless $ns->valid; # temporary(?) workaround for callback destroying socket
2006 2029
2007 push @{ $ns->{query_queue} }, @$queue; 2030 push @{ $ns->{query_queue} }, @$queue;
2008 2031
2009 if (@{ $ns->{query_queue} } == @$queue) { 2032 if (@{ $ns->{query_queue} } == @$queue) {
2010 if (@$queue) { 2033 if (@$queue) {
… …
2171The 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
2172using it, it should be unique. 2195using it, it should be unique.
2173 2196
2174=over 4 2197=over 4
2175 2198
2176=item $hashref = cf::db_get $family
2177
2178Return a hashref for use by the extension C<$family>, which can be
2179modified. After modifications, you have to call C<cf::db_dirty> or
2180C<cf::db_sync>.
2181
2182=item $value = cf::db_get $family => $key 2199=item $value = cf::db_get $family => $key
2183 2200
2184Returns a single value from the database 2201Returns a single value from the database.
2185
2186=item cf::db_put $family => $hashref
2187
2188Stores the given family hashref into the database. Updates are delayed, if
2189you want the data to be synced to disk immediately, use C<cf::db_sync>.
2190 2202
2191=item cf::db_put $family => $key => $value 2203=item cf::db_put $family => $key => $value
2192 2204
2193Stores the given C<$value> in the family hash. Updates are delayed, if you 2205Stores the given C<$value> in the family.
2194want the data to be synced to disk immediately, use C<cf::db_sync>.
2195
2196=item cf::db_dirty
2197
2198Marks the database as dirty, to be updated at a later time.
2199
2200=item cf::db_sync
2201
2202Immediately write the database to disk I<if it is dirty>.
2203 2206
2204=cut 2207=cut
2205 2208
2206our $DB; 2209our $DB;
2207 2210
2208{ 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
2209 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;
2210 2230
2211 sub db_load() { 2231 cf::db_put (board => data => $pst->{board});
2212 $DB = stat $path ? Storable::retrieve $path : { }; 2232 cf::db_put (guildrules => data => $pst->{guildrules});
2213 } 2233 cf::db_put (rent => balance => $pst->{rent}{balance});
2234 BDB::db_env_txn_checkpoint $DB_ENV;
2214 2235
2215 my $pid; 2236 unlink $path;
2216
2217 sub db_save() {
2218 waitpid $pid, 0 if $pid;
2219 if (0 == ($pid = fork)) {
2220 $DB->{_meta}{version} = 1;
2221 Storable::nstore $DB, "$path~";
2222 rename "$path~", $path;
2223 cf::_exit 0 if defined $pid;
2224 } 2237 };
2225 }
2226
2227 my $dirty;
2228
2229 sub db_sync() {
2230 db_save if $dirty;
2231 undef $dirty;
2232 }
2233
2234 my $idle = Event->idle (
2235 reentrant => 0,
2236 min => 10,
2237 max => 20,
2238 repeat => 0,
2239 data => WF_AUTOCANCEL,
2240 cb => \&db_sync,
2241 );
2242
2243 sub db_dirty() {
2244 $dirty = 1;
2245 $idle->start;
2246 }
2247
2248 sub db_get($;$) {
2249 @_ >= 2
2250 ? $DB->{$_[0]}{$_[1]}
2251 : ($DB->{$_[0]} ||= { })
2252 }
2253
2254 sub db_put($$;$) {
2255 if (@_ >= 3) {
2256 $DB->{$_[0]}{$_[1]} = $_[2];
2257 } else {
2258 $DB->{$_[0]} = $_[1];
2259 }
2260 db_dirty;
2261 }
2262
2263 cf::global->attach (
2264 prio => 10000,
2265 on_cleanup => sub {
2266 db_sync;
2267 }, 2238 }
2268 ); 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 { };
2269} 2256}
2270 2257
2271############################################################################# 2258#############################################################################
2272# the server's main() 2259# the server's init and main functions
2260
2261sub load_resources {
2262 load_regions sprintf "%s/%s/regions", cf::datadir, cf::mapdir
2263 or die "unable to load regions file\n";#d#
2264}
2273 2265
2274sub cfg_load { 2266sub cfg_load {
2275 open my $fh, "<:utf8", cf::confdir . "/config" 2267 open my $fh, "<:utf8", cf::confdir . "/config"
2276 or return; 2268 or return;
2277 2269
… …
2288 $CFG{mlockall} ? eval "mlockall()" : eval "munlockall()" 2280 $CFG{mlockall} ? eval "mlockall()" : eval "munlockall()"
2289 and die "WARNING: m(un)lockall failed: $!\n"; 2281 and die "WARNING: m(un)lockall failed: $!\n";
2290 }; 2282 };
2291 warn $@ if $@; 2283 warn $@ if $@;
2292 } 2284 }
2285}
2286
2287sub init {
2288 load_resources;
2293} 2289}
2294 2290
2295sub main { 2291sub main {
2296 # we must not ever block the main coroutine 2292 # we must not ever block the main coroutine
2297 local $Coro::idle = sub { 2293 local $Coro::idle = sub {
… …
2300 Event::one_event; 2296 Event::one_event;
2301 })->prio (Coro::PRIO_MAX); 2297 })->prio (Coro::PRIO_MAX);
2302 }; 2298 };
2303 2299
2304 cfg_load; 2300 cfg_load;
2305 db_load; 2301 db_init;
2306 load_extensions; 2302 load_extensions;
2307 2303
2308 $TICK_WATCHER->start; 2304 $TICK_WATCHER->start;
2309 Event::loop; 2305 Event::loop;
2310} 2306}
… …
2349 my $map = $cf::MAP{$path} or next; 2345 my $map = $cf::MAP{$path} or next;
2350 $map->valid or next; 2346 $map->valid or next;
2351 $map->save; 2347 $map->save;
2352 } 2348 }
2353 warn "end emergency map save\n"; 2349 warn "end emergency map save\n";
2350
2351 warn "begin emergency database checkpoint\n";
2352 BDB::db_env_txn_checkpoint $DB_ENV;
2353 warn "end emergency database checkpoint\n";
2354 }; 2354 };
2355 2355
2356 warn "leave emergency perl save\n"; 2356 warn "leave emergency perl save\n";
2357} 2357}
2358 2358
… …
2371 cf::emergency_save; 2371 cf::emergency_save;
2372 2372
2373 eval { 2373 eval {
2374 # if anything goes wrong in here, we should simply crash as we already saved 2374 # if anything goes wrong in here, we should simply crash as we already saved
2375 2375
2376 warn "syncing database to disk";
2377 cf::db_sync;
2378
2379 warn "cancelling all WF_AUTOCANCEL watchers"; 2376 warn "cancelling all WF_AUTOCANCEL watchers";
2380 for (Event::all_watchers) { 2377 for (Event::all_watchers) {
2381 $_->cancel if $_->data & WF_AUTOCANCEL; 2378 $_->cancel if $_->data & WF_AUTOCANCEL;
2382 } 2379 }
2383 2380
2381 warn "syncing database to disk";
2382 BDB::db_env_txn_checkpoint $DB_ENV, 0, 0, 0, sub { };
2383
2384 warn "flushing outstanding aio requests"; 2384 warn "flushing outstanding aio requests";
2385 for (;;) { 2385 for (;;) {
2386 BDB::flush;
2386 IO::AIO::flush; 2387 IO::AIO::flush;
2387 Coro::cede; 2388 Coro::cede;
2388 last unless IO::AIO::nreqs; 2389 last unless IO::AIO::nreqs || BDB::nreqs;
2389 warn "iterate..."; 2390 warn "iterate...";
2390 } 2391 }
2391 2392
2392 warn "cancelling all extension coros"; 2393 warn "cancelling all extension coros";
2393 $_->cancel for values %EXT_CORO; 2394 $_->cancel for values %EXT_CORO;
… …
2448 require cf; 2449 require cf;
2449 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt 2450 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt
2450 2451
2451 warn "loading config and database again"; 2452 warn "loading config and database again";
2452 cf::cfg_load; 2453 cf::cfg_load;
2453 cf::db_load;
2454 2454
2455 warn "loading extensions"; 2455 warn "loading extensions";
2456 cf::load_extensions; 2456 cf::load_extensions;
2457 2457
2458 warn "reattaching attachments to objects/players"; 2458 warn "reattaching attachments to objects/players";
2459 _global_reattach; 2459 _global_reattach;
2460 warn "reattaching attachments to maps"; 2460 warn "reattaching attachments to maps";
2461 reattach $_ for values %MAP; 2461 reattach $_ for values %MAP;
2462
2463 warn "loading reloadable resources";
2464 load_resources;
2462 2465
2463 warn "restarting server ticker"; 2466 warn "restarting server ticker";
2464 2467
2465 $TICK_WATCHER->start; 2468 $TICK_WATCHER->start;
2466 }; 2469 };
… …
2535 $TICK_WATCHER->at ($NEXT_TICK); 2538 $TICK_WATCHER->at ($NEXT_TICK);
2536 $TICK_WATCHER->start; 2539 $TICK_WATCHER->start;
2537 }, 2540 },
2538); 2541);
2539 2542
2543{
2544 BDB::max_poll_time $TICK * 0.1;
2545 $BDB_POLL_WATCHER = Event->io (
2546 reentrant => 0,
2547 fd => BDB::poll_fileno,
2548 poll => 'r',
2549 prio => 0,
2550 data => WF_AUTOCANCEL,
2551 cb => \&BDB::poll_cb,
2552 );
2553 BDB::min_parallel 8;
2554
2555 BDB::set_sync_prepare {
2556 my $status;
2557 my $current = $Coro::current;
2558 (
2559 sub {
2560 $status = $!;
2561 $current->ready; undef $current;
2562 },
2563 sub {
2564 Coro::schedule while defined $current;
2565 $! = $status;
2566 },
2567 )
2568 };
2569
2570 unless ($DB_ENV) {
2571 $DB_ENV = BDB::db_env_create;
2572
2573 cf::sync_job {
2574 eval {
2575 BDB::db_env_open
2576 $DB_ENV,
2577 $BDB_ENV_DIR,
2578 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN
2579 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE,
2580 0666;
2581
2582 cf::cleanup "db_env_open($BDB_ENV_DIR): $!" if $!;
2583
2584 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1);
2585 $DB_ENV->set_lk_detect;
2586 };
2587
2588 cf::cleanup "db_env_open(db): $@" if $@;
2589 };
2590 }
2591}
2592
2593{
2594 IO::AIO::min_parallel 8;
2595
2596 undef $Coro::AIO::WATCHER;
2540IO::AIO::max_poll_time $TICK * 0.1; 2597 IO::AIO::max_poll_time $TICK * 0.1;
2541
2542undef $Coro::AIO::WATCHER;
2543$AIO_POLL_WATCHER = Event->io ( 2598 $AIO_POLL_WATCHER = Event->io (
2544 reentrant => 0, 2599 reentrant => 0,
2545 fd => IO::AIO::poll_fileno, 2600 fd => IO::AIO::poll_fileno,
2546 poll => 'r', 2601 poll => 'r',
2547 prio => 6, 2602 prio => 6,
2548 data => WF_AUTOCANCEL, 2603 data => WF_AUTOCANCEL,
2549 cb => \&IO::AIO::poll_cb, 2604 cb => \&IO::AIO::poll_cb,
2550); 2605 );
2606}
2551 2607
2552$WRITE_RUNTIME_WATCHER = Event->timer ( 2608$WRITE_RUNTIME_WATCHER = Event->timer (
2553 reentrant => 0, 2609 reentrant => 0,
2554 data => WF_AUTOCANCEL, 2610 data => WF_AUTOCANCEL,
2555 after => 1, 2611 after => 1,

Diff Legend

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