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.207 by root, Sun Feb 11 22:42:09 2007 UTC vs.
Revision 1.219 by root, Sat Feb 17 03:19:44 2007 UTC

22use Digest::MD5; 22use Digest::MD5;
23use Fcntl; 23use Fcntl;
24use IO::AIO 2.32 (); 24use IO::AIO 2.32 ();
25use YAML::Syck (); 25use YAML::Syck ();
26use Time::HiRes; 26use Time::HiRes;
27use Compress::LZF;
28
29Compress::LZF::sfreeze_cr { }; # prime Compress::LZF so it does not use require later
27 30
28use 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
29 32
30sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload 33sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload
31 34
46our $LIBDIR = datadir . "/ext"; 49our $LIBDIR = datadir . "/ext";
47 50
48our $TICK = MAX_TIME * 1e-6; 51our $TICK = MAX_TIME * 1e-6;
49our $TICK_WATCHER; 52our $TICK_WATCHER;
50our $AIO_POLL_WATCHER; 53our $AIO_POLL_WATCHER;
51our $WRITE_RUNTIME_WATCHER; 54our $NEXT_RUNTIME_WRITE; # when should the runtime file be written
52our $NEXT_TICK; 55our $NEXT_TICK;
53our $NOW; 56our $NOW;
54our $USE_FSYNC = 1; # use fsync to write maps - default off 57our $USE_FSYNC = 1; # use fsync to write maps - default off
55 58
56our $BDB_POLL_WATCHER; 59our $BDB_POLL_WATCHER;
270sub freeze_mainloop { 273sub freeze_mainloop {
271 return unless $TICK_WATCHER->is_active; 274 return unless $TICK_WATCHER->is_active;
272 275
273 my $guard = Coro::guard { 276 my $guard = Coro::guard {
274 $TICK_WATCHER->start; 277 $TICK_WATCHER->start;
275 $WRITE_RUNTIME_WATCHER->start;
276 }; 278 };
277 $WRITE_RUNTIME_WATCHER->stop;
278 $TICK_WATCHER->stop; 279 $TICK_WATCHER->stop;
279 $guard 280 $guard
280} 281}
281 282
282=item cf::async { BLOCK } 283=item cf::async { BLOCK }
353 354
354 $coro 355 $coro
355} 356}
356 357
357sub write_runtime { 358sub write_runtime {
359 my $guard = cf::lock_acquire "write_runtime";
360
358 my $runtime = cf::localdir . "/runtime"; 361 my $runtime = cf::localdir . "/runtime";
359 362
360 my $fh = aio_open "$runtime~", O_WRONLY | O_CREAT, 0644 363 my $fh = aio_open "$runtime~", O_WRONLY | O_CREAT, 0644
361 or return; 364 or return;
362 365
784 } 787 }
785 } else { 788 } else {
786 aio_unlink $filename; 789 aio_unlink $filename;
787 aio_unlink "$filename.pst"; 790 aio_unlink "$filename.pst";
788 } 791 }
789
790 #d##TODO# nuke non .map-files if exist
791 if ($filename =~ s/\.map$//) {
792 aio_unlink $filename;
793 aio_unlink "$filename.pst";
794 }
795 } 792 }
796} 793}
797 794
798sub object_freezer_as_string { 795sub object_freezer_as_string {
799 my ($rdata, $objs) = @_; 796 my ($rdata, $objs) = @_;
805 802
806sub object_thawer_load { 803sub object_thawer_load {
807 my ($filename) = @_; 804 my ($filename) = @_;
808 805
809 my ($data, $av); 806 my ($data, $av);
810
811 #d#TODO remove .map if file does not exist
812 aio_stat $filename and $filename =~ s/\.map$//;
813 807
814 (aio_load $filename, $data) >= 0 808 (aio_load $filename, $data) >= 0
815 or return; 809 or return;
816 810
817 unless (aio_stat "$filename.pst") { 811 unless (aio_stat "$filename.pst") {
1304 my ($self) = @_; 1298 my ($self) = @_;
1305 1299
1306 utf8::encode (my $save = $self->save_path); 1300 utf8::encode (my $save = $self->save_path);
1307 IO::AIO::aioreq_pri 4; Coro::AIO::aio_unlink $save; 1301 IO::AIO::aioreq_pri 4; Coro::AIO::aio_unlink $save;
1308 IO::AIO::aioreq_pri 4; Coro::AIO::aio_unlink "$save.pst"; 1302 IO::AIO::aioreq_pri 4; Coro::AIO::aio_unlink "$save.pst";
1309
1310 #d#TODO remove .map and also nuke
1311 $save =~ s/\.map// or return;#d#
1312 IO::AIO::aioreq_pri 4; Coro::AIO::aio_unlink $save;#d#
1313 IO::AIO::aioreq_pri 4; Coro::AIO::aio_unlink "$save.pst";#d#
1314} 1303}
1315 1304
1316sub load_header_from($) { 1305sub load_header_from($) {
1317 my ($self, $path) = @_; 1306 my ($self, $path) = @_;
1318 1307
1606 return if $self->players; 1595 return if $self->players;
1607 return if $self->isa ("ext::map_per_player");#d# 1596 return if $self->isa ("ext::map_per_player");#d#
1608 1597
1609 warn "resetting map ", $self->path;#d# 1598 warn "resetting map ", $self->path;#d#
1610 1599
1600 $self->in_memory (cf::MAP_SWAPPED);
1601
1602 # need to save uniques path
1603 unless ($self->{deny_save}) {
1604 my $uniq = $self->uniq_path; utf8::encode $uniq;
1605
1606 $self->_save_objects ($uniq, cf::IO_UNIQUES)
1607 if $uniq;
1608 }
1609
1611 delete $cf::MAP{$self->path}; 1610 delete $cf::MAP{$self->path};
1612 1611
1613 $self->in_memory (cf::MAP_SWAPPED);
1614 $self->clear; 1612 $self->clear;
1615 1613
1616 $_->clear_links_to ($self) for values %cf::MAP; 1614 $_->clear_links_to ($self) for values %cf::MAP;
1617 1615
1618 $self->unlink_save; 1616 $self->unlink_save;
1808 $map->load; 1806 $map->load;
1809 $map->load_diag; 1807 $map->load_diag;
1810 1808
1811 return unless $self->contr->active; 1809 return unless $self->contr->active;
1812 $self->activate_recursive; 1810 $self->activate_recursive;
1811
1812 local $self->{_prev_pos} = $link_pos; # ugly hack for rent.ext
1813 $self->enter_map ($map, $x, $y); 1813 $self->enter_map ($map, $x, $y);
1814} 1814}
1815 1815
1816cf::player->attach ( 1816cf::player->attach (
1817 on_logout => sub { 1817 on_logout => sub {
1830 # try to abort aborted map switching on player login :) 1830 # try to abort aborted map switching on player login :)
1831 # should happen only on crashes 1831 # should happen only on crashes
1832 if ($pl->ob->{_link_pos}) { 1832 if ($pl->ob->{_link_pos}) {
1833 $pl->ob->enter_link; 1833 $pl->ob->enter_link;
1834 (async { 1834 (async {
1835 # we need this sleep as the login has a concurrent enter_exit running
1836 # and this sleep increases chances of the player not ending up in scorn
1837 $pl->ob->reply (undef, 1835 $pl->ob->reply (undef,
1838 "There was an internal problem at your last logout, " 1836 "There was an internal problem at your last logout, "
1839 . "the server will try to bring you to your intended destination in a second.", 1837 . "the server will try to bring you to your intended destination in a second.",
1840 cf::NDI_RED); 1838 cf::NDI_RED);
1839 # we need this sleep as the login has a concurrent enter_exit running
1840 # and this sleep increases chances of the player not ending up in scorn
1841 Coro::Timer::sleep 1; 1841 Coro::Timer::sleep 1;
1842 $pl->ob->leave_link; 1842 $pl->ob->leave_link;
1843 })->prio (2); 1843 })->prio (2);
1844 } 1844 }
1845 }, 1845 },
2182The parameter C<$family> should best start with the name of the extension 2182The parameter C<$family> should best start with the name of the extension
2183using it, it should be unique. 2183using it, it should be unique.
2184 2184
2185=over 4 2185=over 4
2186 2186
2187=item $hashref = cf::db_get $family
2188
2189Return a hashref for use by the extension C<$family>, which can be
2190modified. After modifications, you have to call C<cf::db_dirty> or
2191C<cf::db_sync>.
2192
2193=item $value = cf::db_get $family => $key 2187=item $value = cf::db_get $family => $key
2194 2188
2195Returns a single value from the database 2189Returns a single value from the database.
2196
2197=item cf::db_put $family => $hashref
2198
2199Stores the given family hashref into the database. Updates are delayed, if
2200you want the data to be synced to disk immediately, use C<cf::db_sync>.
2201 2190
2202=item cf::db_put $family => $key => $value 2191=item cf::db_put $family => $key => $value
2203 2192
2204Stores the given C<$value> in the family hash. Updates are delayed, if you 2193Stores the given C<$value> in the family.
2205want the data to be synced to disk immediately, use C<cf::db_sync>.
2206
2207=item cf::db_dirty
2208
2209Marks the database as dirty, to be updated at a later time.
2210
2211=item cf::db_sync
2212
2213Immediately write the database to disk I<if it is dirty>.
2214 2194
2215=cut 2195=cut
2216 2196
2217our $DB; 2197our $DB;
2218 2198
2219{ 2199sub db_init {
2200 unless ($DB) {
2201 $DB = BDB::db_create $DB_ENV;
2202
2203 cf::sync_job {
2204 eval {
2205 $DB->set_flags (BDB::CHKSUM);
2206
2207 BDB::db_open $DB, undef, "db", undef, BDB::BTREE,
2208 BDB::CREATE | BDB::AUTO_COMMIT, 0666;
2209 cf::cleanup "db_open(db): $!" if $!;
2210 };
2211 cf::cleanup "db_open(db): $@" if $@;
2212 };
2213
2220 my $path = cf::localdir . "/database.pst"; 2214 my $path = cf::localdir . "/database.pst";
2215 if (stat $path) {
2216 cf::sync_job {
2217 my $pst = Storable::retrieve $path;
2221 2218
2222 sub db_load() { 2219 cf::db_put (board => data => $pst->{board});
2223 $DB = stat $path ? Storable::retrieve $path : { }; 2220 cf::db_put (guildrules => data => $pst->{guildrules});
2224 } 2221 cf::db_put (rent => balance => $pst->{rent}{balance});
2222 BDB::db_env_txn_checkpoint $DB_ENV;
2225 2223
2226 my $pid; 2224 unlink $path;
2227
2228 sub db_save() {
2229 waitpid $pid, 0 if $pid;
2230 if (0 == ($pid = fork)) {
2231 $DB->{_meta}{version} = 1;
2232 Storable::nstore $DB, "$path~";
2233 rename "$path~", $path;
2234 cf::_exit 0 if defined $pid;
2235 } 2225 };
2236 }
2237
2238 my $dirty;
2239
2240 sub db_sync() {
2241 db_save if $dirty;
2242 undef $dirty;
2243 }
2244
2245 my $idle = Event->idle (
2246 reentrant => 0,
2247 min => 10,
2248 max => 20,
2249 repeat => 0,
2250 data => WF_AUTOCANCEL,
2251 cb => \&db_sync,
2252 );
2253
2254 sub db_dirty() {
2255 $dirty = 1;
2256 $idle->start;
2257 }
2258
2259 sub db_get($;$) {
2260 @_ >= 2
2261 ? $DB->{$_[0]}{$_[1]}
2262 : ($DB->{$_[0]} ||= { })
2263 }
2264
2265 sub db_put($$;$) {
2266 if (@_ >= 3) {
2267 $DB->{$_[0]}{$_[1]} = $_[2];
2268 } else {
2269 $DB->{$_[0]} = $_[1];
2270 }
2271 db_dirty;
2272 }
2273
2274 cf::global->attach (
2275 prio => 10000,
2276 on_cleanup => sub {
2277 db_sync;
2278 }, 2226 }
2279 ); 2227 }
2228}
2229
2230sub db_get($$) {
2231 my $key = "$_[0]/$_[1]";
2232
2233 cf::sync_job {
2234 BDB::db_get $DB, undef, $key, my $data;
2235
2236 $! ? ()
2237 : Compress::LZF::sthaw $data
2238 }
2239}
2240
2241sub db_put($$$) {
2242 BDB::dbreq_pri 4;
2243 BDB::db_put $DB, undef, "$_[0]/$_[1]", Compress::LZF::sfreeze_cr $_[2], 0, sub { };
2280} 2244}
2281 2245
2282############################################################################# 2246#############################################################################
2283# the server's init and main functions 2247# the server's init and main functions
2284 2248
2285sub load_resources { 2249sub init_resources {
2286 load_regions sprintf "%s/%s/regions", cf::datadir, cf::mapdir 2250 load_resource_file sprintf "%s/%s/regions", cf::datadir, cf::mapdir
2287 or die "unable to load regions file\n";#d# 2251 or die "unable to load regions file\n";#d#
2288} 2252}
2289 2253
2290sub cfg_load { 2254sub cfg_load {
2291 open my $fh, "<:utf8", cf::confdir . "/config" 2255 open my $fh, "<:utf8", cf::confdir . "/config"
2307 warn $@ if $@; 2271 warn $@ if $@;
2308 } 2272 }
2309} 2273}
2310 2274
2311sub init { 2275sub init {
2312 load_resources; 2276 init_resources;
2313} 2277}
2314 2278
2315sub main { 2279sub main {
2316 # we must not ever block the main coroutine 2280 # we must not ever block the main coroutine
2317 local $Coro::idle = sub { 2281 local $Coro::idle = sub {
2320 Event::one_event; 2284 Event::one_event;
2321 })->prio (Coro::PRIO_MAX); 2285 })->prio (Coro::PRIO_MAX);
2322 }; 2286 };
2323 2287
2324 cfg_load; 2288 cfg_load;
2325 db_load; 2289 db_init;
2326 load_extensions; 2290 load_extensions;
2327 2291
2328 $TICK_WATCHER->start; 2292 $TICK_WATCHER->start;
2329 Event::loop; 2293 Event::loop;
2330} 2294}
2369 my $map = $cf::MAP{$path} or next; 2333 my $map = $cf::MAP{$path} or next;
2370 $map->valid or next; 2334 $map->valid or next;
2371 $map->save; 2335 $map->save;
2372 } 2336 }
2373 warn "end emergency map save\n"; 2337 warn "end emergency map save\n";
2338
2339 warn "begin emergency database checkpoint\n";
2340 BDB::db_env_txn_checkpoint $DB_ENV;
2341 warn "end emergency database checkpoint\n";
2374 }; 2342 };
2375 2343
2376 warn "leave emergency perl save\n"; 2344 warn "leave emergency perl save\n";
2345}
2346
2347sub post_cleanup {
2348 my ($make_core) = @_;
2349
2350 warn Carp::longmess "post_cleanup backtrace"
2351 if $make_core;
2377} 2352}
2378 2353
2379sub reload() { 2354sub reload() {
2380 # can/must only be called in main 2355 # can/must only be called in main
2381 if ($Coro::current != $Coro::main) { 2356 if ($Coro::current != $Coro::main) {
2383 return; 2358 return;
2384 } 2359 }
2385 2360
2386 warn "reloading..."; 2361 warn "reloading...";
2387 2362
2388 warn "cancelling server ticker"; 2363 warn "entering sync_job";
2389 $TICK_WATCHER->cancel;
2390 2364
2365 cf::sync_job {
2366 cf::write_runtime; # external watchdog should not bark
2391 cf::emergency_save; 2367 cf::emergency_save;
2368 cf::write_runtime; # external watchdog should not bark
2392 2369
2393 eval { 2370 warn "syncing database to disk";
2371 BDB::db_env_txn_checkpoint $DB_ENV;
2372
2394 # if anything goes wrong in here, we should simply crash as we already saved 2373 # if anything goes wrong in here, we should simply crash as we already saved
2395
2396 warn "syncing database to disk";
2397 cf::db_sync;
2398 2374
2399 warn "cancelling all WF_AUTOCANCEL watchers"; 2375 warn "cancelling all WF_AUTOCANCEL watchers";
2400 for (Event::all_watchers) { 2376 for (Event::all_watchers) {
2401 $_->cancel if $_->data & WF_AUTOCANCEL; 2377 $_->cancel if $_->data & WF_AUTOCANCEL;
2402 } 2378 }
2403 2379
2404 warn "flushing outstanding aio requests"; 2380 warn "flushing outstanding aio requests";
2405 for (;;) { 2381 for (;;) {
2382 BDB::flush;
2406 IO::AIO::flush; 2383 IO::AIO::flush;
2407 Coro::cede; 2384 Coro::cede;
2408 last unless IO::AIO::nreqs; 2385 last unless IO::AIO::nreqs || BDB::nreqs;
2409 warn "iterate..."; 2386 warn "iterate...";
2410 } 2387 }
2411 2388
2412 warn "cancelling all extension coros"; 2389 warn "cancelling all extension coros";
2413 $_->cancel for values %EXT_CORO; 2390 $_->cancel for values %EXT_CORO;
2468 require cf; 2445 require cf;
2469 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt 2446 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt
2470 2447
2471 warn "loading config and database again"; 2448 warn "loading config and database again";
2472 cf::cfg_load; 2449 cf::cfg_load;
2473 cf::db_load;
2474 2450
2475 warn "loading extensions"; 2451 warn "loading extensions";
2476 cf::load_extensions; 2452 cf::load_extensions;
2477 2453
2478 warn "reattaching attachments to objects/players"; 2454 warn "reattaching attachments to objects/players";
2479 _global_reattach; 2455 _global_reattach;
2480 warn "reattaching attachments to maps"; 2456 warn "reattaching attachments to maps";
2481 reattach $_ for values %MAP; 2457 reattach $_ for values %MAP;
2482 2458
2483 warn "loading reloadable resources"; 2459 warn "loading reloadable resources";
2484 load_resources; 2460 init_resources;
2485 2461
2486 warn "restarting server ticker"; 2462 warn "leaving sync_job";
2487 2463
2488 $TICK_WATCHER->start; 2464 1
2489 }; 2465 } or do {
2490
2491 if ($@) {
2492 warn $@; 2466 warn $@;
2493 warn "error while reloading, exiting."; 2467 warn "error while reloading, exiting.";
2494 exit 1; 2468 exit 1;
2495 } 2469 };
2496 2470
2497 warn "reloaded"; 2471 warn "reloaded";
2498}; 2472};
2499 2473
2500our $RELOAD_WATCHER; # used only during reload 2474our $RELOAD_WATCHER; # used only during reload
2541 2515
2542 cf::server_tick; # one server iteration 2516 cf::server_tick; # one server iteration
2543 $RUNTIME += $TICK; 2517 $RUNTIME += $TICK;
2544 $NEXT_TICK += $TICK; 2518 $NEXT_TICK += $TICK;
2545 2519
2520 if ($NOW >= $NEXT_RUNTIME_WRITE) {
2521 $NEXT_RUNTIME_WRITE = $NOW + 10;
2522 Coro::async_pool {
2523 write_runtime
2524 or warn "ERROR: unable to write runtime file: $!";
2525 };
2526 }
2527
2546 $WAIT_FOR_TICK->broadcast; 2528 $WAIT_FOR_TICK->broadcast;
2547 $WAIT_FOR_TICK_ONE->send if $WAIT_FOR_TICK_ONE->awaited; 2529 $WAIT_FOR_TICK_ONE->send if $WAIT_FOR_TICK_ONE->awaited;
2548 2530
2549 Event::sweep;
2550 Coro::cede_notself;
2551
2552# my $AFTER = Event::time; 2531# my $AFTER = Event::time;
2553# warn $AFTER - $NOW;#d# 2532# warn $AFTER - $NOW;#d#
2554 2533
2555 # if we are delayed by four ticks or more, skip them all 2534 # if we are delayed by four ticks or more, skip them all
2556 $NEXT_TICK = Event::time if Event::time >= $NEXT_TICK + $TICK * 4; 2535 $NEXT_TICK = Event::time if Event::time >= $NEXT_TICK + $TICK * 4;
2589 2568
2590 unless ($DB_ENV) { 2569 unless ($DB_ENV) {
2591 $DB_ENV = BDB::db_env_create; 2570 $DB_ENV = BDB::db_env_create;
2592 2571
2593 cf::sync_job { 2572 cf::sync_job {
2573 eval {
2594 BDB::db_env_open 2574 BDB::db_env_open
2595 $DB_ENV, 2575 $DB_ENV,
2596 $BDB_ENV_DIR, 2576 $BDB_ENV_DIR,
2597 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN 2577 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN
2598 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE, 2578 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE,
2599 0666; 2579 0666;
2600 2580
2601 cf::cleanup "$!" if $!; 2581 cf::cleanup "db_env_open($BDB_ENV_DIR): $!" if $!;
2602 2582
2603 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1); 2583 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1);
2604 $DB_ENV->set_lk_detect; 2584 $DB_ENV->set_lk_detect;
2585 };
2586
2587 cf::cleanup "db_env_open(db): $@" if $@;
2605 }; 2588 };
2606 } 2589 }
2607} 2590}
2608 2591
2609{ 2592{
2611 2594
2612 undef $Coro::AIO::WATCHER; 2595 undef $Coro::AIO::WATCHER;
2613 IO::AIO::max_poll_time $TICK * 0.1; 2596 IO::AIO::max_poll_time $TICK * 0.1;
2614 $AIO_POLL_WATCHER = Event->io ( 2597 $AIO_POLL_WATCHER = Event->io (
2615 reentrant => 0, 2598 reentrant => 0,
2599 data => WF_AUTOCANCEL,
2616 fd => IO::AIO::poll_fileno, 2600 fd => IO::AIO::poll_fileno,
2617 poll => 'r', 2601 poll => 'r',
2618 prio => 6, 2602 prio => 6,
2619 data => WF_AUTOCANCEL,
2620 cb => \&IO::AIO::poll_cb, 2603 cb => \&IO::AIO::poll_cb,
2621 ); 2604 );
2622} 2605}
2623 2606
2624$WRITE_RUNTIME_WATCHER = Event->timer (
2625 reentrant => 0,
2626 data => WF_AUTOCANCEL,
2627 after => 1,
2628 interval => 10,
2629 prio => 6, # keep it lowest so it acts like a watchdog
2630 cb => Coro::unblock_sub {
2631 write_runtime
2632 or warn "ERROR: unable to write runtime file: $!";
2633 },
2634);
2635
2636END { cf::emergency_save } 2607END { cf::emergency_save }
2637 2608
26381 26091
2639 2610

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines