… | |
… | |
22 | use Digest::MD5; |
22 | use Digest::MD5; |
23 | use Fcntl; |
23 | use Fcntl; |
24 | use IO::AIO 2.32 (); |
24 | use IO::AIO 2.32 (); |
25 | use YAML::Syck (); |
25 | use YAML::Syck (); |
26 | 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 |
27 | |
30 | |
28 | 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 |
29 | |
32 | |
30 | sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload |
33 | sub WF_AUTOCANCEL () { 1 } # automatically cancel this watcher on reload |
31 | |
34 | |
… | |
… | |
353 | |
356 | |
354 | $coro |
357 | $coro |
355 | } |
358 | } |
356 | |
359 | |
357 | sub write_runtime { |
360 | sub write_runtime { |
|
|
361 | $TICK_WATCHER->is_active or cf::cleanup "mainloop frozen but runtime active", 1; |
|
|
362 | |
358 | my $runtime = cf::localdir . "/runtime"; |
363 | my $runtime = cf::localdir . "/runtime"; |
359 | |
364 | |
360 | my $fh = aio_open "$runtime~", O_WRONLY | O_CREAT, 0644 |
365 | my $fh = aio_open "$runtime~", O_WRONLY | O_CREAT, 0644 |
361 | or return; |
366 | or return; |
362 | |
367 | |
… | |
… | |
1606 | return if $self->players; |
1611 | return if $self->players; |
1607 | return if $self->isa ("ext::map_per_player");#d# |
1612 | return if $self->isa ("ext::map_per_player");#d# |
1608 | |
1613 | |
1609 | warn "resetting map ", $self->path;#d# |
1614 | warn "resetting map ", $self->path;#d# |
1610 | |
1615 | |
|
|
1616 | $self->in_memory (cf::MAP_SWAPPED); |
|
|
1617 | |
|
|
1618 | # need to save uniques path |
|
|
1619 | unless ($self->{deny_save}) { |
|
|
1620 | my $uniq = $self->uniq_path; utf8::encode $uniq; |
|
|
1621 | |
|
|
1622 | $self->_save_objects ($uniq, cf::IO_UNIQUES) |
|
|
1623 | if $uniq; |
|
|
1624 | } |
|
|
1625 | |
1611 | delete $cf::MAP{$self->path}; |
1626 | delete $cf::MAP{$self->path}; |
1612 | |
1627 | |
1613 | $self->in_memory (cf::MAP_SWAPPED); |
|
|
1614 | $self->clear; |
1628 | $self->clear; |
1615 | |
1629 | |
1616 | $_->clear_links_to ($self) for values %cf::MAP; |
1630 | $_->clear_links_to ($self) for values %cf::MAP; |
1617 | |
1631 | |
1618 | $self->unlink_save; |
1632 | $self->unlink_save; |
… | |
… | |
2182 | The parameter C<$family> should best start with the name of the extension |
2196 | The parameter C<$family> should best start with the name of the extension |
2183 | using it, it should be unique. |
2197 | using it, it should be unique. |
2184 | |
2198 | |
2185 | =over 4 |
2199 | =over 4 |
2186 | |
2200 | |
2187 | =item $hashref = cf::db_get $family |
|
|
2188 | |
|
|
2189 | Return a hashref for use by the extension C<$family>, which can be |
|
|
2190 | modified. After modifications, you have to call C<cf::db_dirty> or |
|
|
2191 | C<cf::db_sync>. |
|
|
2192 | |
|
|
2193 | =item $value = cf::db_get $family => $key |
2201 | =item $value = cf::db_get $family => $key |
2194 | |
2202 | |
2195 | Returns a single value from the database |
2203 | Returns a single value from the database. |
2196 | |
|
|
2197 | =item cf::db_put $family => $hashref |
|
|
2198 | |
|
|
2199 | Stores the given family hashref into the database. Updates are delayed, if |
|
|
2200 | you want the data to be synced to disk immediately, use C<cf::db_sync>. |
|
|
2201 | |
2204 | |
2202 | =item cf::db_put $family => $key => $value |
2205 | =item cf::db_put $family => $key => $value |
2203 | |
2206 | |
2204 | Stores the given C<$value> in the family hash. Updates are delayed, if you |
2207 | Stores the given C<$value> in the family. |
2205 | want the data to be synced to disk immediately, use C<cf::db_sync>. |
|
|
2206 | |
|
|
2207 | =item cf::db_dirty |
|
|
2208 | |
|
|
2209 | Marks the database as dirty, to be updated at a later time. |
|
|
2210 | |
|
|
2211 | =item cf::db_sync |
|
|
2212 | |
|
|
2213 | Immediately write the database to disk I<if it is dirty>. |
|
|
2214 | |
2208 | |
2215 | =cut |
2209 | =cut |
2216 | |
2210 | |
2217 | our $DB; |
2211 | our $DB; |
2218 | |
2212 | |
2219 | { |
2213 | sub db_init { |
|
|
2214 | unless ($DB) { |
|
|
2215 | $DB = BDB::db_create $DB_ENV; |
|
|
2216 | |
|
|
2217 | cf::sync_job { |
|
|
2218 | eval { |
|
|
2219 | $DB->set_flags (BDB::CHKSUM); |
|
|
2220 | |
|
|
2221 | BDB::db_open $DB, undef, "db", undef, BDB::BTREE, |
|
|
2222 | BDB::CREATE | BDB::AUTO_COMMIT, 0666; |
|
|
2223 | cf::cleanup "db_open(db): $!" if $!; |
|
|
2224 | }; |
|
|
2225 | cf::cleanup "db_open(db): $@" if $@; |
|
|
2226 | }; |
|
|
2227 | |
2220 | my $path = cf::localdir . "/database.pst"; |
2228 | my $path = cf::localdir . "/database.pst"; |
|
|
2229 | if (stat $path) { |
|
|
2230 | cf::sync_job { |
|
|
2231 | my $pst = Storable::retrieve $path; |
2221 | |
2232 | |
2222 | sub db_load() { |
2233 | cf::db_put (board => data => $pst->{board}); |
2223 | $DB = stat $path ? Storable::retrieve $path : { }; |
2234 | cf::db_put (guildrules => data => $pst->{guildrules}); |
2224 | } |
2235 | cf::db_put (rent => balance => $pst->{rent}{balance}); |
|
|
2236 | BDB::db_env_txn_checkpoint $DB_ENV; |
2225 | |
2237 | |
2226 | my $pid; |
2238 | 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 | } |
2239 | }; |
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 | }, |
2240 | } |
2279 | ); |
2241 | } |
|
|
2242 | } |
|
|
2243 | |
|
|
2244 | sub db_get($$) { |
|
|
2245 | my $key = "$_[0]/$_[1]"; |
|
|
2246 | |
|
|
2247 | cf::sync_job { |
|
|
2248 | BDB::db_get $DB, undef, $key, my $data; |
|
|
2249 | |
|
|
2250 | $! ? () |
|
|
2251 | : Compress::LZF::sthaw $data |
|
|
2252 | } |
|
|
2253 | } |
|
|
2254 | |
|
|
2255 | sub db_put($$$) { |
|
|
2256 | BDB::dbreq_pri 4; |
|
|
2257 | BDB::db_put $DB, undef, "$_[0]/$_[1]", Compress::LZF::sfreeze_cr $_[2], 0, sub { }; |
2280 | } |
2258 | } |
2281 | |
2259 | |
2282 | ############################################################################# |
2260 | ############################################################################# |
2283 | # the server's init and main functions |
2261 | # the server's init and main functions |
2284 | |
2262 | |
… | |
… | |
2320 | Event::one_event; |
2298 | Event::one_event; |
2321 | })->prio (Coro::PRIO_MAX); |
2299 | })->prio (Coro::PRIO_MAX); |
2322 | }; |
2300 | }; |
2323 | |
2301 | |
2324 | cfg_load; |
2302 | cfg_load; |
2325 | db_load; |
2303 | db_init; |
2326 | load_extensions; |
2304 | load_extensions; |
2327 | |
2305 | |
2328 | $TICK_WATCHER->start; |
2306 | $TICK_WATCHER->start; |
2329 | Event::loop; |
2307 | Event::loop; |
2330 | } |
2308 | } |
… | |
… | |
2369 | my $map = $cf::MAP{$path} or next; |
2347 | my $map = $cf::MAP{$path} or next; |
2370 | $map->valid or next; |
2348 | $map->valid or next; |
2371 | $map->save; |
2349 | $map->save; |
2372 | } |
2350 | } |
2373 | warn "end emergency map save\n"; |
2351 | warn "end emergency map save\n"; |
|
|
2352 | |
|
|
2353 | warn "begin emergency database checkpoint\n"; |
|
|
2354 | BDB::db_env_txn_checkpoint $DB_ENV; |
|
|
2355 | warn "end emergency database checkpoint\n"; |
2374 | }; |
2356 | }; |
2375 | |
2357 | |
2376 | warn "leave emergency perl save\n"; |
2358 | warn "leave emergency perl save\n"; |
|
|
2359 | } |
|
|
2360 | |
|
|
2361 | sub post_cleanup { |
|
|
2362 | my ($make_core) = @_; |
|
|
2363 | |
|
|
2364 | warn Carp::longmess "post_cleanup backtrace" |
|
|
2365 | if $make_core; |
2377 | } |
2366 | } |
2378 | |
2367 | |
2379 | sub reload() { |
2368 | sub reload() { |
2380 | # can/must only be called in main |
2369 | # can/must only be called in main |
2381 | if ($Coro::current != $Coro::main) { |
2370 | if ($Coro::current != $Coro::main) { |
… | |
… | |
2391 | cf::emergency_save; |
2380 | cf::emergency_save; |
2392 | |
2381 | |
2393 | eval { |
2382 | eval { |
2394 | # if anything goes wrong in here, we should simply crash as we already saved |
2383 | # if anything goes wrong in here, we should simply crash as we already saved |
2395 | |
2384 | |
2396 | warn "syncing database to disk"; |
|
|
2397 | cf::db_sync; |
|
|
2398 | |
|
|
2399 | warn "cancelling all WF_AUTOCANCEL watchers"; |
2385 | warn "cancelling all WF_AUTOCANCEL watchers"; |
2400 | for (Event::all_watchers) { |
2386 | for (Event::all_watchers) { |
2401 | $_->cancel if $_->data & WF_AUTOCANCEL; |
2387 | $_->cancel if $_->data & WF_AUTOCANCEL; |
2402 | } |
2388 | } |
2403 | |
2389 | |
|
|
2390 | warn "syncing database to disk"; |
|
|
2391 | BDB::db_env_txn_checkpoint $DB_ENV, 0, 0, 0, sub { }; |
|
|
2392 | |
2404 | warn "flushing outstanding aio requests"; |
2393 | warn "flushing outstanding aio requests"; |
2405 | for (;;) { |
2394 | for (;;) { |
|
|
2395 | BDB::flush; |
2406 | IO::AIO::flush; |
2396 | IO::AIO::flush; |
2407 | Coro::cede; |
2397 | Coro::cede; |
2408 | last unless IO::AIO::nreqs; |
2398 | last unless IO::AIO::nreqs || BDB::nreqs; |
2409 | warn "iterate..."; |
2399 | warn "iterate..."; |
2410 | } |
2400 | } |
2411 | |
2401 | |
2412 | warn "cancelling all extension coros"; |
2402 | warn "cancelling all extension coros"; |
2413 | $_->cancel for values %EXT_CORO; |
2403 | $_->cancel for values %EXT_CORO; |
… | |
… | |
2468 | require cf; |
2458 | require cf; |
2469 | cf::_connect_to_perl; # nominally unnecessary, but cannot hurt |
2459 | cf::_connect_to_perl; # nominally unnecessary, but cannot hurt |
2470 | |
2460 | |
2471 | warn "loading config and database again"; |
2461 | warn "loading config and database again"; |
2472 | cf::cfg_load; |
2462 | cf::cfg_load; |
2473 | cf::db_load; |
|
|
2474 | |
2463 | |
2475 | warn "loading extensions"; |
2464 | warn "loading extensions"; |
2476 | cf::load_extensions; |
2465 | cf::load_extensions; |
2477 | |
2466 | |
2478 | warn "reattaching attachments to objects/players"; |
2467 | warn "reattaching attachments to objects/players"; |
… | |
… | |
2589 | |
2578 | |
2590 | unless ($DB_ENV) { |
2579 | unless ($DB_ENV) { |
2591 | $DB_ENV = BDB::db_env_create; |
2580 | $DB_ENV = BDB::db_env_create; |
2592 | |
2581 | |
2593 | cf::sync_job { |
2582 | cf::sync_job { |
|
|
2583 | eval { |
2594 | BDB::db_env_open |
2584 | BDB::db_env_open |
2595 | $DB_ENV, |
2585 | $DB_ENV, |
2596 | $BDB_ENV_DIR, |
2586 | $BDB_ENV_DIR, |
2597 | BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN |
2587 | BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN |
2598 | | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE, |
2588 | | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE, |
2599 | 0666; |
2589 | 0666; |
2600 | |
2590 | |
2601 | cf::cleanup "$!" if $!; |
2591 | cf::cleanup "db_env_open($BDB_ENV_DIR): $!" if $!; |
2602 | |
2592 | |
2603 | $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1); |
2593 | $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1); |
2604 | $DB_ENV->set_lk_detect; |
2594 | $DB_ENV->set_lk_detect; |
|
|
2595 | }; |
|
|
2596 | |
|
|
2597 | cf::cleanup "db_env_open(db): $@" if $@; |
2605 | }; |
2598 | }; |
2606 | } |
2599 | } |
2607 | } |
2600 | } |
2608 | |
2601 | |
2609 | { |
2602 | { |