… | |
… | |
15 | use Coro::Timer; |
15 | use Coro::Timer; |
16 | use Coro::Signal; |
16 | use Coro::Signal; |
17 | use Coro::Semaphore; |
17 | use Coro::Semaphore; |
18 | use Coro::AIO; |
18 | use Coro::AIO; |
19 | |
19 | |
|
|
20 | use BDB (); |
20 | use Data::Dumper; |
21 | use Data::Dumper; |
21 | use Digest::MD5; |
22 | use Digest::MD5; |
22 | use Fcntl; |
23 | use Fcntl; |
23 | use IO::AIO 2.32 (); |
24 | use IO::AIO 2.32 (); |
24 | use YAML::Syck (); |
25 | use YAML::Syck (); |
25 | use Time::HiRes; |
26 | use Time::HiRes; |
|
|
27 | use Compress::LZF; |
|
|
28 | |
|
|
29 | Compress::LZF::sfreeze_cr { }; # prime Compress::LZF so it does not use require later |
26 | |
30 | |
27 | use Event; $Event::Eval = 1; # no idea why this is required, but it is |
31 | use Event; $Event::Eval = 1; # no idea why this is required, but it is |
28 | |
32 | |
29 | sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload |
33 | sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload |
30 | |
34 | |
… | |
… | |
48 | our $TICK_WATCHER; |
52 | our $TICK_WATCHER; |
49 | our $AIO_POLL_WATCHER; |
53 | our $AIO_POLL_WATCHER; |
50 | our $WRITE_RUNTIME_WATCHER; |
54 | our $WRITE_RUNTIME_WATCHER; |
51 | our $NEXT_TICK; |
55 | our $NEXT_TICK; |
52 | our $NOW; |
56 | our $NOW; |
53 | our $USE_FSYNC; # use fsync to write maps - default off |
57 | our $USE_FSYNC = 1; # use fsync to write maps - default off |
|
|
58 | |
|
|
59 | our $BDB_POLL_WATCHER; |
|
|
60 | our $DB_ENV; |
54 | |
61 | |
55 | our %CFG; |
62 | our %CFG; |
56 | |
63 | |
57 | our $UPTIME; $UPTIME ||= time; |
64 | our $UPTIME; $UPTIME ||= time; |
58 | our $RUNTIME; |
65 | our $RUNTIME; |
59 | |
66 | |
60 | our %PLAYER; # all users |
67 | our %PLAYER; # all users |
61 | our %MAP; # all maps |
68 | our %MAP; # all maps |
62 | our $LINK_MAP; # the special {link} map, which is always available |
69 | our $LINK_MAP; # the special {link} map, which is always available |
63 | our $RANDOM_MAPS = cf::localdir . "/random"; |
70 | our $RANDOM_MAPS = cf::localdir . "/random"; |
|
|
71 | our $BDB_ENV_DIR = cf::localdir . "/db"; |
64 | |
72 | |
65 | our $WAIT_FOR_TICK; $WAIT_FOR_TICK ||= new Coro::Signal; |
73 | our $WAIT_FOR_TICK; $WAIT_FOR_TICK ||= new Coro::Signal; |
66 | our $WAIT_FOR_TICK_ONE; $WAIT_FOR_TICK_ONE ||= new Coro::Signal; |
74 | our $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 ∕ |
… | |
… | |
81 | mkdir cf::localdir; |
89 | mkdir cf::localdir; |
82 | mkdir cf::localdir . "/" . cf::playerdir; |
90 | mkdir cf::localdir . "/" . cf::playerdir; |
83 | mkdir cf::localdir . "/" . cf::tmpdir; |
91 | mkdir cf::localdir . "/" . cf::tmpdir; |
84 | mkdir cf::localdir . "/" . cf::uniquedir; |
92 | mkdir cf::localdir . "/" . cf::uniquedir; |
85 | mkdir $RANDOM_MAPS; |
93 | mkdir $RANDOM_MAPS; |
|
|
94 | mkdir $BDB_ENV_DIR; |
86 | |
95 | |
87 | our $EMERGENCY_POSITION; |
96 | our $EMERGENCY_POSITION; |
88 | |
97 | |
89 | sub cf::map::normalise; |
98 | sub 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; |
… | |
… | |
2176 | The parameter C<$family> should best start with the name of the extension |
2194 | The parameter C<$family> should best start with the name of the extension |
2177 | using it, it should be unique. |
2195 | using it, it should be unique. |
2178 | |
2196 | |
2179 | =over 4 |
2197 | =over 4 |
2180 | |
2198 | |
2181 | =item $hashref = cf::db_get $family |
|
|
2182 | |
|
|
2183 | Return a hashref for use by the extension C<$family>, which can be |
|
|
2184 | modified. After modifications, you have to call C<cf::db_dirty> or |
|
|
2185 | C<cf::db_sync>. |
|
|
2186 | |
|
|
2187 | =item $value = cf::db_get $family => $key |
2199 | =item $value = cf::db_get $family => $key |
2188 | |
2200 | |
2189 | Returns a single value from the database |
2201 | Returns a single value from the database. |
2190 | |
|
|
2191 | =item cf::db_put $family => $hashref |
|
|
2192 | |
|
|
2193 | Stores the given family hashref into the database. Updates are delayed, if |
|
|
2194 | you 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 | |
2198 | Stores the given C<$value> in the family hash. Updates are delayed, if you |
2205 | Stores the given C<$value> in the family. |
2199 | want the data to be synced to disk immediately, use C<cf::db_sync>. |
|
|
2200 | |
|
|
2201 | =item cf::db_dirty |
|
|
2202 | |
|
|
2203 | Marks the database as dirty, to be updated at a later time. |
|
|
2204 | |
|
|
2205 | =item cf::db_sync |
|
|
2206 | |
|
|
2207 | Immediately write the database to disk I<if it is dirty>. |
|
|
2208 | |
2206 | |
2209 | =cut |
2207 | =cut |
2210 | |
2208 | |
2211 | our $DB; |
2209 | our $DB; |
2212 | |
2210 | |
2213 | { |
2211 | sub 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 | |
|
|
2242 | sub 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 | |
|
|
2253 | sub 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; |
2323 | Event::loop; |
2305 | Event::loop; |
2324 | } |
2306 | } |
… | |
… | |
2363 | my $map = $cf::MAP{$path} or next; |
2345 | my $map = $cf::MAP{$path} or next; |
2364 | $map->valid or next; |
2346 | $map->valid or next; |
2365 | $map->save; |
2347 | $map->save; |
2366 | } |
2348 | } |
2367 | 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"; |
2368 | }; |
2354 | }; |
2369 | |
2355 | |
2370 | warn "leave emergency perl save\n"; |
2356 | warn "leave emergency perl save\n"; |
2371 | } |
2357 | } |
2372 | |
2358 | |
… | |
… | |
2385 | cf::emergency_save; |
2371 | cf::emergency_save; |
2386 | |
2372 | |
2387 | eval { |
2373 | eval { |
2388 | # 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 |
2389 | |
2375 | |
2390 | warn "syncing database to disk"; |
|
|
2391 | cf::db_sync; |
|
|
2392 | |
|
|
2393 | warn "cancelling all WF_AUTOCANCEL watchers"; |
2376 | warn "cancelling all WF_AUTOCANCEL watchers"; |
2394 | for (Event::all_watchers) { |
2377 | for (Event::all_watchers) { |
2395 | $_->cancel if $_->data & WF_AUTOCANCEL; |
2378 | $_->cancel if $_->data & WF_AUTOCANCEL; |
2396 | } |
2379 | } |
2397 | |
2380 | |
|
|
2381 | warn "syncing database to disk"; |
|
|
2382 | BDB::db_env_txn_checkpoint $DB_ENV, 0, 0, 0, sub { }; |
|
|
2383 | |
2398 | warn "flushing outstanding aio requests"; |
2384 | warn "flushing outstanding aio requests"; |
2399 | for (;;) { |
2385 | for (;;) { |
|
|
2386 | BDB::flush; |
2400 | IO::AIO::flush; |
2387 | IO::AIO::flush; |
2401 | Coro::cede; |
2388 | Coro::cede; |
2402 | last unless IO::AIO::nreqs; |
2389 | last unless IO::AIO::nreqs || BDB::nreqs; |
2403 | warn "iterate..."; |
2390 | warn "iterate..."; |
2404 | } |
2391 | } |
2405 | |
2392 | |
2406 | warn "cancelling all extension coros"; |
2393 | warn "cancelling all extension coros"; |
2407 | $_->cancel for values %EXT_CORO; |
2394 | $_->cancel for values %EXT_CORO; |
… | |
… | |
2462 | require cf; |
2449 | require cf; |
2463 | cf::_connect_to_perl; # nominally unnecessary, but cannot hurt |
2450 | cf::_connect_to_perl; # nominally unnecessary, but cannot hurt |
2464 | |
2451 | |
2465 | warn "loading config and database again"; |
2452 | warn "loading config and database again"; |
2466 | cf::cfg_load; |
2453 | cf::cfg_load; |
2467 | cf::db_load; |
|
|
2468 | |
2454 | |
2469 | warn "loading extensions"; |
2455 | warn "loading extensions"; |
2470 | cf::load_extensions; |
2456 | cf::load_extensions; |
2471 | |
2457 | |
2472 | warn "reattaching attachments to objects/players"; |
2458 | warn "reattaching attachments to objects/players"; |
… | |
… | |
2552 | $TICK_WATCHER->at ($NEXT_TICK); |
2538 | $TICK_WATCHER->at ($NEXT_TICK); |
2553 | $TICK_WATCHER->start; |
2539 | $TICK_WATCHER->start; |
2554 | }, |
2540 | }, |
2555 | ); |
2541 | ); |
2556 | |
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; |
2557 | IO::AIO::max_poll_time $TICK * 0.1; |
2597 | IO::AIO::max_poll_time $TICK * 0.1; |
2558 | |
|
|
2559 | undef $Coro::AIO::WATCHER; |
|
|
2560 | $AIO_POLL_WATCHER = Event->io ( |
2598 | $AIO_POLL_WATCHER = Event->io ( |
2561 | reentrant => 0, |
2599 | reentrant => 0, |
2562 | fd => IO::AIO::poll_fileno, |
2600 | fd => IO::AIO::poll_fileno, |
2563 | poll => 'r', |
2601 | poll => 'r', |
2564 | prio => 6, |
2602 | prio => 6, |
2565 | data => WF_AUTOCANCEL, |
2603 | data => WF_AUTOCANCEL, |
2566 | cb => \&IO::AIO::poll_cb, |
2604 | cb => \&IO::AIO::poll_cb, |
2567 | ); |
2605 | ); |
|
|
2606 | } |
2568 | |
2607 | |
2569 | $WRITE_RUNTIME_WATCHER = Event->timer ( |
2608 | $WRITE_RUNTIME_WATCHER = Event->timer ( |
2570 | reentrant => 0, |
2609 | reentrant => 0, |
2571 | data => WF_AUTOCANCEL, |
2610 | data => WF_AUTOCANCEL, |
2572 | after => 1, |
2611 | after => 1, |