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.209 by root, Mon Feb 12 01:25:55 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
… …
2182The 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
2183using it, it should be unique. 2186using it, it should be unique.
2184 2187
2185=over 4 2188=over 4
2186 2189
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 2190=item $value = cf::db_get $family => $key
2194 2191
2195Returns a single value from the database 2192Returns 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 2193
2202=item cf::db_put $family => $key => $value 2194=item cf::db_put $family => $key => $value
2203 2195
2204Stores the given C<$value> in the family hash. Updates are delayed, if you 2196Stores 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 2197
2215=cut 2198=cut
2216 2199
2217our $DB; 2200our $DB;
2218 2201
2219{ 2202unless ($DB) {
2203 $DB = BDB::db_create $DB_ENV;
2204
2205 cf::sync_job {
2206 eval {
2207 $DB->set_flags (BDB::CHKSUM);
2208
2209 BDB::db_open $DB, undef, "db", undef, BDB::BTREE,
2210 BDB::CREATE | BDB::AUTO_COMMIT, 0666;
2211 cf::cleanup "db_open(db): $!" if $!;
2212 };
2213 cf::cleanup "db_open(db): $@" if $@;
2214 };
2215
2220 my $path = cf::localdir . "/database.pst"; 2216 my $path = cf::localdir . "/database.pst";
2217 if (stat $path) {
2218 cf::sync_job {
2219 my $pst = Storable::retrieve $path;
2221 2220
2222 sub db_load() { 2221 cf::db_put (board => data => $pst->{board});
2223 $DB = stat $path ? Storable::retrieve $path : { }; 2222 cf::db_put (guildrules => data => $pst->{guildrules});
2224 } 2223 cf::db_put (rent => balance => $pst->{rent}{balance});
2224 BDB::db_env_txn_checkpoint $DB_ENV;
2225 2225
2226 my $pid; 2226 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 } 2227 };
2236 } 2228 }
2229}
2237 2230
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($;$) { 2231sub db_get($$) {
2260 @_ >= 2 2232 my $key = "$_[0]/$_[1]";
2261 ? $DB->{$_[0]}{$_[1]}
2262 : ($DB->{$_[0]} ||= { })
2263 }
2264 2233
2234 cf::sync_job {
2235 BDB::db_get $DB, undef, $key, my $data;
2236
2237 $! ? ()
2238 : Compress::LZF::sthaw $data
2239 }
2240}
2241
2265 sub db_put($$;$) { 2242sub db_put($$$) {
2266 if (@_ >= 3) { 2243 BDB::dbreq_pri 4;
2267 $DB->{$_[0]}{$_[1]} = $_[2]; 2244 BDB::db_put $DB, undef, "$_[0]/$_[1]", Compress::LZF::sfreeze_cr $_[2], 0, sub { };
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 },
2279 );
2280} 2245}
2281 2246
2282############################################################################# 2247#############################################################################
2283# the server's init and main functions 2248# the server's init and main functions
2284 2249
… …
2320 Event::one_event; 2285 Event::one_event;
2321 })->prio (Coro::PRIO_MAX); 2286 })->prio (Coro::PRIO_MAX);
2322 }; 2287 };
2323 2288
2324 cfg_load; 2289 cfg_load;
2325 db_load;
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";
2377} 2345}
2378 2346
… …
2391 cf::emergency_save; 2359 cf::emergency_save;
2392 2360
2393 eval { 2361 eval {
2394 # if anything goes wrong in here, we should simply crash as we already saved 2362 # if anything goes wrong in here, we should simply crash as we already saved
2395 2363
2396 warn "syncing database to disk";
2397 cf::db_sync;
2398
2399 warn "cancelling all WF_AUTOCANCEL watchers"; 2364 warn "cancelling all WF_AUTOCANCEL watchers";
2400 for (Event::all_watchers) { 2365 for (Event::all_watchers) {
2401 $_->cancel if $_->data & WF_AUTOCANCEL; 2366 $_->cancel if $_->data & WF_AUTOCANCEL;
2402 } 2367 }
2403 2368
2369 warn "syncing database to disk";
2370 BDB::db_env_txn_checkpoint $DB_ENV, 0, 0, 0, sub { };
2371
2404 warn "flushing outstanding aio requests"; 2372 warn "flushing outstanding aio requests";
2405 for (;;) { 2373 for (;;) {
2374 BDB::flush;
2406 IO::AIO::flush; 2375 IO::AIO::flush;
2407 Coro::cede; 2376 Coro::cede;
2408 last unless IO::AIO::nreqs; 2377 last unless IO::AIO::nreqs || BDB::nreqs;
2409 warn "iterate..."; 2378 warn "iterate...";
2410 } 2379 }
2411 2380
2412 warn "cancelling all extension coros"; 2381 warn "cancelling all extension coros";
2413 $_->cancel for values %EXT_CORO; 2382 $_->cancel for values %EXT_CORO;
… …
2468 require cf; 2437 require cf;
2469 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt 2438 cf::_connect_to_perl; # nominally unnecessary, but cannot hurt
2470 2439
2471 warn "loading config and database again"; 2440 warn "loading config and database again";
2472 cf::cfg_load; 2441 cf::cfg_load;
2473 cf::db_load;
2474 2442
2475 warn "loading extensions"; 2443 warn "loading extensions";
2476 cf::load_extensions; 2444 cf::load_extensions;
2477 2445
2478 warn "reattaching attachments to objects/players"; 2446 warn "reattaching attachments to objects/players";
… …
2589 2557
2590 unless ($DB_ENV) { 2558 unless ($DB_ENV) {
2591 $DB_ENV = BDB::db_env_create; 2559 $DB_ENV = BDB::db_env_create;
2592 2560
2593 cf::sync_job { 2561 cf::sync_job {
2562 eval {
2594 BDB::db_env_open 2563 BDB::db_env_open
2595 $DB_ENV, 2564 $DB_ENV,
2596 $BDB_ENV_DIR, 2565 $BDB_ENV_DIR,
2597 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN 2566 BDB::INIT_LOCK | BDB::INIT_LOG | BDB::INIT_MPOOL | BDB::INIT_TXN
2598 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE, 2567 | BDB::RECOVER | BDB::REGISTER | BDB::USE_ENVIRON | BDB::CREATE,
2599 0666; 2568 0666;
2600 2569
2601 cf::cleanup "$!" if $!; 2570 cf::cleanup "db_env_open($BDB_ENV_DIR): $!" if $!;
2602 2571
2603 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1); 2572 $DB_ENV->set_flags (BDB::AUTO_COMMIT | BDB::REGION_INIT | BDB::TXN_NOSYNC, 1);
2604 $DB_ENV->set_lk_detect; 2573 $DB_ENV->set_lk_detect;
2574 };
2575
2576 cf::cleanup "db_env_open(db): $@" if $@;
2605 }; 2577 };
2606 } 2578 }
2607} 2579}
2608 2580
2609{ 2581{

Diff Legend

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