]> git.ipfire.org Git - thirdparty/squid.git/blame - src/store_dir.cc
Renamed squid.h to squid-old.h and config.h to squid.h
[thirdparty/squid.git] / src / store_dir.cc
CommitLineData
f1dc9b30 1
2/*
262a0e14 3 * $Id$
f1dc9b30 4 *
5 * DEBUG: section 47 Store Directory Routines
6 * AUTHOR: Duane Wessels
7 *
2b6662ba 8 * SQUID Web Proxy Cache http://www.squid-cache.org/
e25c139f 9 * ----------------------------------------------------------
f1dc9b30 10 *
2b6662ba 11 * Squid is the result of efforts by numerous individuals from
12 * the Internet community; see the CONTRIBUTORS file for full
13 * details. Many organizations have provided support for Squid's
14 * development; see the SPONSORS file for full details. Squid is
15 * Copyrighted (C) 2001 by the Regents of the University of
16 * California; see the COPYRIGHT file for full details. Squid
17 * incorporates software developed and/or copyrighted by other
18 * sources; see the CREDITS file for full details.
f1dc9b30 19 *
20 * This program is free software; you can redistribute it and/or modify
21 * it under the terms of the GNU General Public License as published by
22 * the Free Software Foundation; either version 2 of the License, or
23 * (at your option) any later version.
26ac0430 24 *
f1dc9b30 25 * This program is distributed in the hope that it will be useful,
26 * but WITHOUT ANY WARRANTY; without even the implied warranty of
27 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
28 * GNU General Public License for more details.
26ac0430 29 *
f1dc9b30 30 * You should have received a copy of the GNU General Public License
31 * along with this program; if not, write to the Free Software
cbdec147 32 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
e25c139f 33 *
f1dc9b30 34 */
35
f7f3304a 36#include "squid-old.h"
e6ccf245 37#include "Store.h"
528b2c61 38#include "MemObject.h"
9487bae9
AR
39#include "MemStore.h"
40#include "mem_node.h"
a98bcbee 41#include "SquidMath.h"
985c86bc 42#include "SquidTime.h"
d3b3ab85 43#include "SwapDir.h"
4b981814 44#include "swap_log_op.h"
85407535 45
c0db87f2 46#if HAVE_STATVFS
47#if HAVE_SYS_STATVFS_H
48#include <sys/statvfs.h>
49#endif
ec15e022 50#endif /* HAVE_STATVFS */
51/* statfs() needs <sys/param.h> and <sys/mount.h> on BSD systems */
52#if HAVE_SYS_PARAM_H
53#include <sys/param.h>
203526a1 54#endif
ec15e022 55#if HAVE_SYS_MOUNT_H
56#include <sys/mount.h>
57#endif
58/* Windows and Linux use sys/vfs.h */
6c86a065 59#if HAVE_SYS_VFS_H
60#include <sys/vfs.h>
61#endif
c0db87f2 62
c8f4eac4 63#include "StoreHashIndex.h"
64
65a53c8e 65static STDIRSELECT storeDirSelectSwapDirRoundRobin;
66static STDIRSELECT storeDirSelectSwapDirLeastLoad;
c8f4eac4 67
b07b21cc 68/*
69 * store_dirs_rebuilding is initialized to _1_ as a hack so that
70 * storeDirWriteCleanLogs() doesn't try to do anything unless _all_
71 * cache_dirs have been read. For example, without this hack, Squid
72 * will try to write clean log files if -kparse fails (becasue it
73 * calls fatal()).
74 */
75int StoreController::store_dirs_rebuilding = 1;
bef81ea5 76
c8f4eac4 77StoreController::StoreController() : swapDir (new StoreHashIndex())
9199139f 78 , memStore(NULL)
c8f4eac4 79{}
80
81StoreController::~StoreController()
9487bae9
AR
82{
83 delete memStore;
84}
65a53c8e 85
86/*
87 * This function pointer is set according to 'store_dir_select_algorithm'
88 * in squid.conf.
89 */
90STDIRSELECT *storeDirSelectSwapDir = storeDirSelectSwapDirLeastLoad;
d141c677 91
9838d6c8 92void
c8f4eac4 93StoreController::init()
596dddc1 94{
57af1e3f 95 if (Config.memShared && IamWorkerProcess()) {
60be8b2d
AR
96 memStore = new MemStore;
97 memStore->init();
98 }
9487bae9 99
c8f4eac4 100 swapDir->init();
62e76326 101
65a53c8e 102 if (0 == strcasecmp(Config.store_dir_select_algorithm, "round-robin")) {
62e76326 103 storeDirSelectSwapDir = storeDirSelectSwapDirRoundRobin;
bf8fe701 104 debugs(47, 1, "Using Round Robin store dir selection");
65a53c8e 105 } else {
62e76326 106 storeDirSelectSwapDir = storeDirSelectSwapDirLeastLoad;
bf8fe701 107 debugs(47, 1, "Using Least Load store dir selection");
65a53c8e 108 }
85407535 109}
110
111void
c8f4eac4 112StoreController::createOneStore(Store &aStore)
596dddc1 113{
62e76326 114 /*
154c7949 115 * On Windows, fork() is not available.
116 * The following is a workaround for create store directories sequentially
117 * when running on native Windows port.
118 */
1191b93b 119#if !_SQUID_MSWIN_
62e76326 120
154c7949 121 if (fork())
62e76326 122 return;
123
099a1791 124#endif
62e76326 125
c8f4eac4 126 aStore.create();
62e76326 127
1191b93b 128#if !_SQUID_MSWIN_
62e76326 129
154c7949 130 exit(0);
62e76326 131
099a1791 132#endif
154c7949 133}
134
135void
c8f4eac4 136StoreController::create()
154c7949 137{
c8f4eac4 138 swapDir->create();
62e76326 139
1191b93b 140#if !_SQUID_MSWIN_
62e76326 141
8a1c8f2c 142 pid_t pid;
62e76326 143
b2c141d4 144 do {
62e76326 145 int status;
1191b93b 146#if _SQUID_NEXT_
62e76326 147
148 pid = wait3(&status, WNOHANG, NULL);
b2c141d4 149#else
62e76326 150
151 pid = waitpid(-1, &status, 0);
b2c141d4 152#endif
62e76326 153
b2c141d4 154 } while (pid > 0 || (pid < 0 && errno == EINTR));
62e76326 155
099a1791 156#endif
596dddc1 157}
158
a8a33c46 159/**
cd748f27 160 * Determine whether the given directory can handle this object
161 * size
162 *
163 * Note: if the object size is -1, then the only swapdirs that
a8a33c46 164 * will return true here are ones that have min and max unset,
cd748f27 165 * ie any-sized-object swapdirs. This is a good thing.
166 */
c8f4eac4 167bool
3e62bd58 168SwapDir::objectSizeIsAcceptable(int64_t objsize) const
cd748f27 169{
a8a33c46 170 // If the swapdir has no range limits, then it definitely can
b6662ffd 171 if (min_objsize <= 0 && max_objsize == -1)
c8f4eac4 172 return true;
d68f43a0 173
174 /*
a8a33c46
A
175 * If the object size is -1 and the storedir has limits we
176 * can't store it there.
d68f43a0 177 */
a8a33c46 178 if (objsize == -1)
c8f4eac4 179 return false;
d68f43a0 180
a8a33c46 181 // Else, make sure that the object size will fit.
b475997c
AJ
182 if (max_objsize == -1 && min_objsize <= objsize)
183 return true;
184 else
185 return min_objsize <= objsize && max_objsize > objsize;
cd748f27 186}
187
188
d141c677 189/*
190 * This new selection scheme simply does round-robin on all SwapDirs.
8e8d4f30 191 * A SwapDir is skipped if it is over the max_size (100%) limit, or
192 * overloaded.
d141c677 193 */
194static int
8e8d4f30 195storeDirSelectSwapDirRoundRobin(const StoreEntry * e)
d141c677 196{
197 static int dirn = 0;
198 int i;
8e8d4f30 199 int load;
c8f4eac4 200 RefCount<SwapDir> sd;
62e76326 201
aa1a691e
AR
202 // e->objectLen() is negative at this point when we are still STORE_PENDING
203 ssize_t objsize = e->mem_obj->expectedReplySize();
246e6cc1
AJ
204 if (objsize != -1)
205 objsize += e->mem_obj->swap_hdr_sz;
206
ada9124c 207 for (i = 0; i < Config.cacheSwap.n_configured; i++) {
62e76326 208 if (++dirn >= Config.cacheSwap.n_configured)
209 dirn = 0;
210
c8f4eac4 211 sd = dynamic_cast<SwapDir *>(INDEXSD(dirn));
62e76326 212
aa1a691e 213 if (!sd->canStore(*e, objsize, load))
62e76326 214 continue;
215
62e76326 216 if (load < 0 || load > 1000) {
217 continue;
218 }
219
220 return dirn;
d141c677 221 }
62e76326 222
8e8d4f30 223 return -1;
d141c677 224}
960a01e3 225
a2899918 226/*
cd748f27 227 * Spread load across all of the store directories
228 *
229 * Note: We should modify this later on to prefer sticking objects
230 * in the *tightest fit* swapdir to conserve space, along with the
26ac0430 231 * actual swapdir usage. But for now, this hack will do while
cd748f27 232 * testing, so you should order your swapdirs in the config file
233 * from smallest maxobjsize to unlimited (-1) maxobjsize.
234 *
235 * We also have to choose nleast == nconf since we need to consider
236 * ALL swapdirs, regardless of state. Again, this is a hack while
237 * we sort out the real usefulness of this algorithm.
a2899918 238 */
65a53c8e 239static int
240storeDirSelectSwapDirLeastLoad(const StoreEntry * e)
a2899918 241{
cc34568d 242 int64_t most_free = 0;
8e8d4f30 243 ssize_t least_objsize = -1;
244 int least_load = INT_MAX;
cd748f27 245 int load;
246 int dirn = -1;
247 int i;
c8f4eac4 248 RefCount<SwapDir> SD;
cd748f27 249
aa1a691e
AR
250 // e->objectLen() is negative at this point when we are still STORE_PENDING
251 ssize_t objsize = e->mem_obj->expectedReplySize();
62e76326 252
cd748f27 253 if (objsize != -1)
62e76326 254 objsize += e->mem_obj->swap_hdr_sz;
255
cd748f27 256 for (i = 0; i < Config.cacheSwap.n_configured; i++) {
c8f4eac4 257 SD = dynamic_cast<SwapDir *>(INDEXSD(i));
62e76326 258 SD->flags.selected = 0;
62e76326 259
aa1a691e 260 if (!SD->canStore(*e, objsize, load))
62e76326 261 continue;
262
aa1a691e 263 if (load < 0 || load > 1000)
62e76326 264 continue;
265
266 if (load > least_load)
267 continue;
268
cc34568d 269 const int64_t cur_free = SD->maxSize() - SD->currentSize();
62e76326 270
271 /* If the load is equal, then look in more details */
272 if (load == least_load) {
273 /* closest max_objsize fit */
274
275 if (least_objsize != -1)
276 if (SD->max_objsize > least_objsize || SD->max_objsize == -1)
277 continue;
278
279 /* most free */
280 if (cur_free < most_free)
281 continue;
282 }
283
284 least_load = load;
285 least_objsize = SD->max_objsize;
286 most_free = cur_free;
287 dirn = i;
a2899918 288 }
62e76326 289
ade906c8 290 if (dirn >= 0)
c8f4eac4 291 dynamic_cast<SwapDir *>(INDEXSD(dirn))->flags.selected = 1;
62e76326 292
cd748f27 293 return dirn;
596dddc1 294}
295
b109de6b 296/*
297 * An entry written to the swap log MUST have the following
298 * properties.
299 * 1. It MUST be a public key. It does no good to log
300 * a public ADD, change the key, then log a private
301 * DEL. So we need to log a DEL before we change a
302 * key from public to private.
cd748f27 303 * 2. It MUST have a valid (> -1) swap_filen.
b109de6b 304 */
4683e377 305void
5830cdb3 306storeDirSwapLog(const StoreEntry * e, int op)
4683e377 307{
d3b3ab85 308 assert (e);
d46a87a8 309 assert(!EBIT_TEST(e->flags, KEY_PRIVATE));
cd748f27 310 assert(e->swap_filen >= 0);
6c57e268 311 /*
312 * icons and such; don't write them to the swap log
313 */
62e76326 314
d46a87a8 315 if (EBIT_TEST(e->flags, ENTRY_SPECIAL))
62e76326 316 return;
317
b109de6b 318 assert(op > SWAP_LOG_NOP && op < SWAP_LOG_MAX);
62e76326 319
26ac0430
AJ
320 debugs(20, 3, "storeDirSwapLog: " <<
321 swap_log_op_str[op] << " " <<
322 e->getMD5Text() << " " <<
323 e->swap_dirn << " " <<
bf8fe701 324 std::hex << std::uppercase << std::setfill('0') << std::setw(8) << e->swap_filen);
62e76326 325
c8f4eac4 326 dynamic_cast<SwapDir *>(INDEXSD(e->swap_dirn))->logEntry(*e, op);
327}
328
93bc1434
AR
329void
330StoreController::getStats(StoreInfoStats &stats) const
331{
332 if (memStore)
333 memStore->getStats(stats);
334 else {
335 // move this code to a non-shared memory cache class when we have it
336 stats.mem.shared = false;
337 stats.mem.capacity = Config.memMaxSize;
338 stats.mem.size = mem_node::StoreMemSize();
339 stats.mem.count = hot_obj_count;
340 }
341
342 swapDir->getStats(stats);
343
344 // low-level info not specific to memory or disk cache
345 stats.store_entry_count = StoreEntry::inUseCount();
346 stats.mem_object_count = MemObject::inUseCount();
347}
348
c932b107 349void
c8f4eac4 350StoreController::stat(StoreEntry &output) const
c932b107 351{
c8f4eac4 352 storeAppendPrintf(&output, "Store Directory Statistics:\n");
353 storeAppendPrintf(&output, "Store Entries : %lu\n",
62e76326 354 (unsigned long int)StoreEntry::inUseCount());
12e11a5c 355 storeAppendPrintf(&output, "Maximum Swap Size : %"PRIu64" KB\n",
cc34568d 356 maxSize() >> 10);
57f583f1 357 storeAppendPrintf(&output, "Current Store Swap Size: %.2f KB\n",
cc34568d 358 currentSize() / 1024.0);
57f583f1 359 storeAppendPrintf(&output, "Current Capacity : %.2f%% used, %.2f%% free\n",
cc34568d
DK
360 Math::doublePercent(currentSize(), maxSize()),
361 Math::doublePercent((maxSize() - currentSize()), maxSize()));
ada9124c
AR
362
363 if (memStore)
364 memStore->stat(output);
e3ef2b09 365
c8f4eac4 366 /* now the swapDir */
367 swapDir->stat(output);
5d406e78 368}
369
c8f4eac4 370/* if needed, this could be taught to cache the result */
12e11a5c 371uint64_t
c8f4eac4 372StoreController::maxSize() const
f4e3fa54 373{
c8f4eac4 374 /* TODO: include memory cache ? */
375 return swapDir->maxSize();
376}
62e76326 377
12e11a5c 378uint64_t
c8f4eac4 379StoreController::minSize() const
380{
381 /* TODO: include memory cache ? */
382 return swapDir->minSize();
f4e3fa54 383}
384
39c1e1d9
DK
385uint64_t
386StoreController::currentSize() const
387{
388 return swapDir->currentSize();
389}
390
391uint64_t
392StoreController::currentCount() const
393{
394 return swapDir->currentCount();
395}
396
af2fda07
DK
397int64_t
398StoreController::maxObjectSize() const
399{
400 return swapDir->maxObjectSize();
401}
402
f4e3fa54 403void
c8f4eac4 404SwapDir::diskFull()
f4e3fa54 405{
cc34568d 406 if (currentSize() >= maxSize())
62e76326 407 return;
408
cc34568d 409 max_size = currentSize();
62e76326 410
8765bc8c 411 debugs(20, 1, "WARNING: Shrinking cache_dir #" << index << " to " << currentSize() / 1024.0 << " KB");
f4e3fa54 412}
95dcd2b8 413
414void
415storeDirOpenSwapLogs(void)
416{
d3b3ab85 417 for (int dirn = 0; dirn < Config.cacheSwap.n_configured; ++dirn)
c8f4eac4 418 dynamic_cast<SwapDir *>(INDEXSD(dirn))->openLog();
95dcd2b8 419}
420
421void
422storeDirCloseSwapLogs(void)
423{
d3b3ab85 424 for (int dirn = 0; dirn < Config.cacheSwap.n_configured; ++dirn)
c8f4eac4 425 dynamic_cast<SwapDir *>(INDEXSD(dirn))->closeLog();
95dcd2b8 426}
427
b2c141d4 428/*
429 * storeDirWriteCleanLogs
26ac0430 430 *
b2c141d4 431 * Writes a "clean" swap log file from in-memory metadata.
cd748f27 432 * This is a rewrite of the original function to troll each
433 * StoreDir and write the logs, and flush at the end of
434 * the run. Thanks goes to Eric Stern, since this solution
435 * came out of his COSS code.
b2c141d4 436 */
b2c141d4 437int
438storeDirWriteCleanLogs(int reopen)
95dcd2b8 439{
6a566b9c 440 const StoreEntry *e = NULL;
b2c141d4 441 int n = 0;
62e76326 442
e812ecfc 443 struct timeval start;
444 double dt;
c8f4eac4 445 RefCount<SwapDir> sd;
b2c141d4 446 int dirn;
6a566b9c 447 int notdone = 1;
62e76326 448
bef81ea5 449 if (StoreController::store_dirs_rebuilding) {
bf8fe701 450 debugs(20, 1, "Not currently OK to rewrite swap log.");
451 debugs(20, 1, "storeDirWriteCleanLogs: Operation aborted.");
62e76326 452 return 0;
b2c141d4 453 }
62e76326 454
bf8fe701 455 debugs(20, 1, "storeDirWriteCleanLogs: Starting...");
e812ecfc 456 getCurrentTime();
457 start = current_time;
62e76326 458
b2c141d4 459 for (dirn = 0; dirn < Config.cacheSwap.n_configured; dirn++) {
c8f4eac4 460 sd = dynamic_cast<SwapDir *>(INDEXSD(dirn));
62e76326 461
462 if (sd->writeCleanStart() < 0) {
bf8fe701 463 debugs(20, 1, "log.clean.start() failed for dir #" << sd->index);
62e76326 464 continue;
465 }
6a566b9c 466 }
62e76326 467
e78ef51b 468 /*
469 * This may look inefficient as CPU wise it is more efficient to do this
470 * sequentially, but I/O wise the parallellism helps as it allows more
471 * hdd spindles to be active.
d3b3ab85 472 */
c1dd71ae 473 while (notdone) {
62e76326 474 notdone = 0;
475
476 for (dirn = 0; dirn < Config.cacheSwap.n_configured; dirn++) {
c8f4eac4 477 sd = dynamic_cast<SwapDir *>(INDEXSD(dirn));
62e76326 478
479 if (NULL == sd->cleanLog)
480 continue;
481
482 e = sd->cleanLog->nextEntry();
483
484 if (!e)
485 continue;
486
487 notdone = 1;
488
489 if (!sd->canLog(*e))
490 continue;
491
492 sd->cleanLog->write(*e);
493
494 if ((++n & 0xFFFF) == 0) {
495 getCurrentTime();
bf8fe701 496 debugs(20, 1, " " << std::setw(7) << n <<
497 " entries written so far.");
62e76326 498 }
499 }
6a566b9c 500 }
62e76326 501
6a566b9c 502 /* Flush */
d3b3ab85 503 for (dirn = 0; dirn < Config.cacheSwap.n_configured; dirn++)
c8f4eac4 504 dynamic_cast<SwapDir *>(INDEXSD(dirn))->writeCleanDone();
62e76326 505
b2c141d4 506 if (reopen)
62e76326 507 storeDirOpenSwapLogs();
508
e812ecfc 509 getCurrentTime();
62e76326 510
e812ecfc 511 dt = tvSubDsec(start, current_time);
62e76326 512
bf8fe701 513 debugs(20, 1, " Finished. Wrote " << n << " entries.");
514 debugs(20, 1, " Took "<< std::setw(3)<< std::setprecision(2) << dt <<
515 " seconds ("<< std::setw(6) << ((double) n / (dt > 0.0 ? dt : 1.0)) << " entries/sec).");
62e76326 516
62e76326 517
b2c141d4 518 return n;
95dcd2b8 519}
d141c677 520
c8f4eac4 521StoreSearch *
30abd221 522StoreController::search(String const url, HttpRequest *request)
c8f4eac4 523{
524 /* cheat, for now you can't search the memory hot cache */
525 return swapDir->search(url, request);
526}
527
528StorePointer
529StoreHashIndex::store(int const x) const
530{
531 return INDEXSD(x);
532}
533
14911a4e
AR
534SwapDir &
535StoreHashIndex::dir(const int i) const
536{
537 SwapDir *sd = dynamic_cast<SwapDir*>(INDEXSD(i));
538 assert(sd);
539 return *sd;
540}
541
cd748f27 542void
c8f4eac4 543StoreController::sync(void)
cd748f27 544{
9487bae9
AR
545 if (memStore)
546 memStore->sync();
c8f4eac4 547 swapDir->sync();
cd748f27 548}
549
550/*
26ac0430 551 * handle callbacks all avaliable fs'es
cd748f27 552 */
c8f4eac4 553int
554StoreController::callback()
cd748f27 555{
1d5161bd 556 /* This will likely double count. Thats ok. */
557 PROF_start(storeDirCallback);
558
c8f4eac4 559 /* mem cache callbacks ? */
560 int result = swapDir->callback();
1d5161bd 561
562 PROF_stop(storeDirCallback);
c8f4eac4 563
564 return result;
d141c677 565}
90d42c28 566
567int
568storeDirGetBlkSize(const char *path, int *blksize)
569{
570#if HAVE_STATVFS
62e76326 571
90d42c28 572 struct statvfs sfs;
62e76326 573
90d42c28 574 if (statvfs(path, &sfs)) {
bf8fe701 575 debugs(50, 1, "" << path << ": " << xstrerror());
62e76326 576 *blksize = 2048;
577 return 1;
90d42c28 578 }
62e76326 579
6759a7aa 580 *blksize = (int) sfs.f_frsize;
90d42c28 581#else
62e76326 582
90d42c28 583 struct statfs sfs;
62e76326 584
90d42c28 585 if (statfs(path, &sfs)) {
bf8fe701 586 debugs(50, 1, "" << path << ": " << xstrerror());
62e76326 587 *blksize = 2048;
588 return 1;
90d42c28 589 }
62e76326 590
90d42c28 591 *blksize = (int) sfs.f_bsize;
6759a7aa 592#endif
4b3af09f 593 /*
594 * Sanity check; make sure we have a meaningful value.
595 */
62e76326 596
d5b72fe7 597 if (*blksize < 512)
62e76326 598 *blksize = 2048;
599
90d42c28 600 return 0;
601}
781d6656 602
603#define fsbtoblk(num, fsbs, bs) \
604 (((fsbs) != 0 && (fsbs) < (bs)) ? \
605 (num) / ((bs) / (fsbs)) : (num) * ((fsbs) / (bs)))
606int
607storeDirGetUFSStats(const char *path, int *totl_kb, int *free_kb, int *totl_in, int *free_in)
608{
609#if HAVE_STATVFS
62e76326 610
781d6656 611 struct statvfs sfs;
62e76326 612
781d6656 613 if (statvfs(path, &sfs)) {
bf8fe701 614 debugs(50, 1, "" << path << ": " << xstrerror());
62e76326 615 return 1;
781d6656 616 }
62e76326 617
781d6656 618 *totl_kb = (int) fsbtoblk(sfs.f_blocks, sfs.f_frsize, 1024);
619 *free_kb = (int) fsbtoblk(sfs.f_bfree, sfs.f_frsize, 1024);
620 *totl_in = (int) sfs.f_files;
621 *free_in = (int) sfs.f_ffree;
622#else
62e76326 623
781d6656 624 struct statfs sfs;
62e76326 625
781d6656 626 if (statfs(path, &sfs)) {
bf8fe701 627 debugs(50, 1, "" << path << ": " << xstrerror());
62e76326 628 return 1;
781d6656 629 }
62e76326 630
781d6656 631 *totl_kb = (int) fsbtoblk(sfs.f_blocks, sfs.f_bsize, 1024);
632 *free_kb = (int) fsbtoblk(sfs.f_bfree, sfs.f_bsize, 1024);
633 *totl_in = (int) sfs.f_files;
634 *free_in = (int) sfs.f_ffree;
635#endif
62e76326 636
781d6656 637 return 0;
638}
c8f4eac4 639
640void
e1f7507e 641allocate_new_swapdir(SquidConfig::_cacheSwap * swap)
c8f4eac4 642{
643 if (swap->swapDirs == NULL) {
644 swap->n_allocated = 4;
7d3c4ca1 645 swap->swapDirs = static_cast<SwapDir::Pointer *>(xcalloc(swap->n_allocated, sizeof(SwapDir::Pointer)));
c8f4eac4 646 }
647
648 if (swap->n_allocated == swap->n_configured) {
c8f4eac4 649 swap->n_allocated <<= 1;
7d3c4ca1 650 SwapDir::Pointer *const tmp = static_cast<SwapDir::Pointer *>(xcalloc(swap->n_allocated, sizeof(SwapDir::Pointer)));
41d00cd3 651 memcpy(tmp, swap->swapDirs, swap->n_configured * sizeof(SwapDir *));
c8f4eac4 652 xfree(swap->swapDirs);
653 swap->swapDirs = tmp;
654 }
655}
656
657void
e1f7507e 658free_cachedir(SquidConfig::_cacheSwap * swap)
c8f4eac4 659{
660 int i;
661 /* DON'T FREE THESE FOR RECONFIGURE */
662
663 if (reconfiguring)
664 return;
665
666 for (i = 0; i < swap->n_configured; i++) {
667 /* TODO XXX this lets the swapdir free resources asynchronously
668 * swap->swapDirs[i]->deactivate();
26ac0430 669 * but there may be such a means already.
c8f4eac4 670 * RBC 20041225
671 */
672 swap->swapDirs[i] = NULL;
673 }
674
675 safe_free(swap->swapDirs);
676 swap->swapDirs = NULL;
677 swap->n_allocated = 0;
678 swap->n_configured = 0;
679}
680
681/* this should be a virtual method on StoreEntry,
682 * i.e. e->referenced()
683 * so that the entry can notify the creating Store
684 */
685void
686StoreController::reference(StoreEntry &e)
687{
688 /* Notify the fs that we're referencing this object again */
689
690 if (e.swap_dirn > -1)
4c973beb 691 swapDir->reference(e);
c8f4eac4 692
9487bae9
AR
693 // Notify the memory cache that we're referencing this object again
694 if (memStore && e.mem_status == IN_MEMORY)
695 memStore->reference(e);
696
697 // TODO: move this code to a non-shared memory cache class when we have it
c8f4eac4 698 if (e.mem_obj) {
699 if (mem_policy->Referenced)
700 mem_policy->Referenced(mem_policy, &e, &e.mem_obj->repl);
701 }
702}
703
4c973beb 704bool
c8f4eac4 705StoreController::dereference(StoreEntry & e)
706{
897d5656 707 bool keepInStoreTable = true; // keep if there are no objections
4c973beb 708
c8f4eac4 709 /* Notify the fs that we're not referencing this object any more */
710
711 if (e.swap_filen > -1)
897d5656 712 keepInStoreTable = swapDir->dereference(e) && keepInStoreTable;
c8f4eac4 713
9487bae9
AR
714 // Notify the memory cache that we're not referencing this object any more
715 if (memStore && e.mem_status == IN_MEMORY)
897d5656 716 keepInStoreTable = memStore->dereference(e) && keepInStoreTable;
9487bae9
AR
717
718 // TODO: move this code to a non-shared memory cache class when we have it
c8f4eac4 719 if (e.mem_obj) {
720 if (mem_policy->Dereferenced)
721 mem_policy->Dereferenced(mem_policy, &e, &e.mem_obj->repl);
722 }
4c973beb
AR
723
724 return keepInStoreTable;
c8f4eac4 725}
726
727StoreEntry *
6ca34f6f 728StoreController::get(const cache_key *key)
c8f4eac4 729{
44def0f9 730 if (StoreEntry *e = swapDir->get(key)) {
9487bae9
AR
731 // TODO: ignore and maybe handleIdleEntry() unlocked intransit entries
732 // because their backing store slot may be gone already.
171d5429 733 debugs(20, 3, HERE << "got in-transit entry: " << *e);
44def0f9
AR
734 return e;
735 }
736
9487bae9
AR
737 if (memStore) {
738 if (StoreEntry *e = memStore->get(key)) {
739 debugs(20, 3, HERE << "got mem-cached entry: " << *e);
740 return e;
741 }
742 }
743
022f96ad
AR
744 // TODO: this disk iteration is misplaced; move to StoreHashIndex when
745 // the global store_table is no longer used for in-transit objects.
44def0f9
AR
746 if (const int cacheDirs = Config.cacheSwap.n_configured) {
747 // ask each cache_dir until the entry is found; use static starting
748 // point to avoid asking the same subset of disks more often
749 // TODO: coordinate with put() to be able to guess the right disk often
14911a4e 750 static int idx = 0;
44def0f9 751 for (int n = 0; n < cacheDirs; ++n) {
14911a4e 752 idx = (idx + 1) % cacheDirs;
44def0f9 753 SwapDir *sd = dynamic_cast<SwapDir*>(INDEXSD(idx));
14911a4e
AR
754 if (!sd->active())
755 continue;
756
44def0f9 757 if (StoreEntry *e = sd->get(key)) {
eccba1d9 758 debugs(20, 3, HERE << "cache_dir " << idx <<
9199139f 759 " got cached entry: " << *e);
44def0f9
AR
760 return e;
761 }
762 }
763 }
c8f4eac4 764
eccba1d9 765 debugs(20, 4, HERE << "none of " << Config.cacheSwap.n_configured <<
9199139f 766 " cache_dirs have " << storeKeyText(key));
44def0f9 767 return NULL;
c8f4eac4 768}
769
770void
6ca34f6f 771StoreController::get(String const key, STOREGETCLIENT aCallback, void *aCallbackData)
c8f4eac4 772{
773 fatal("not implemented");
774}
775
9487bae9
AR
776void
777StoreController::handleIdleEntry(StoreEntry &e)
778{
779 bool keepInLocalMemory = false;
780 if (memStore) {
781 memStore->considerKeeping(e);
782 // leave keepInLocalMemory false; memStore maintains its own cache
783 } else {
784 keepInLocalMemory = e.memoryCachable() && // entry is in good shape and
9199139f
AR
785 // the local memory cache is not overflowing
786 (mem_node::InUseCount() <= store_pages_max);
9487bae9
AR
787 }
788
4c973beb
AR
789 // An idle, unlocked entry that belongs to a SwapDir which controls
790 // its own index, should not stay in the global store_table.
791 if (!dereference(e)) {
9487bae9
AR
792 debugs(20, 5, HERE << "destroying unlocked entry: " << &e << ' ' << e);
793 destroyStoreEntry(static_cast<hash_link*>(&e));
794 return;
795 }
796
797 // TODO: move this into [non-shared] memory cache class when we have one
798 if (keepInLocalMemory) {
799 e.setMemStatus(IN_MEMORY);
800 e.mem_obj->unlinkRequest();
801 } else {
802 e.purgeMem(); // may free e
803 }
804}
805
c8f4eac4 806StoreHashIndex::StoreHashIndex()
807{
47f6e231 808 if (store_table)
26ac0430 809 abort();
c8f4eac4 810 assert (store_table == NULL);
811}
812
813StoreHashIndex::~StoreHashIndex()
814{
815 if (store_table) {
816 hashFreeItems(store_table, destroyStoreEntry);
817 hashFreeMemory(store_table);
818 store_table = NULL;
819 }
820}
821
822int
823StoreHashIndex::callback()
824{
825 int result = 0;
826 int j;
827 static int ndir = 0;
828
829 do {
830 j = 0;
831
832 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
833 if (ndir >= Config.cacheSwap.n_configured)
834 ndir = ndir % Config.cacheSwap.n_configured;
835
836 int temp_result = store(ndir)->callback();
837
838 ++ndir;
839
840 j += temp_result;
841
842 result += temp_result;
843
844 if (j > 100)
845 fatal ("too much io\n");
846 }
847 } while (j > 0);
848
849 ndir++;
850
851 return result;
852}
853
854void
855StoreHashIndex::create()
856{
14911a4e
AR
857 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
858 if (dir(i).active())
859 store(i)->create();
860 }
c8f4eac4 861}
862
863/* Lookup an object in the cache.
864 * return just a reference to object, don't start swapping in yet. */
865StoreEntry *
6ca34f6f 866StoreHashIndex::get(const cache_key *key)
c8f4eac4 867{
868 PROF_start(storeGet);
bf8fe701 869 debugs(20, 3, "storeGet: looking up " << storeKeyText(key));
c8f4eac4 870 StoreEntry *p = static_cast<StoreEntry *>(hash_lookup(store_table, key));
871 PROF_stop(storeGet);
872 return p;
873}
874
875void
6ca34f6f 876StoreHashIndex::get(String const key, STOREGETCLIENT aCallback, void *aCallbackData)
c8f4eac4 877{
878 fatal("not implemented");
879}
880
881void
882StoreHashIndex::init()
883{
884 /* Calculate size of hash table (maximum currently 64k buckets). */
885 /* this is very bogus, its specific to the any Store maintaining an
886 * in-core index, not global */
58d5c5dd 887 size_t buckets = (Store::Root().maxSize() + Config.memMaxSize) / Config.Store.avgObjectSize;
cc34568d 888 debugs(20, 1, "Swap maxSize " << (Store::Root().maxSize() >> 10) <<
13a07022 889 " + " << ( Config.memMaxSize >> 10) << " KB, estimated " << buckets << " objects");
c8f4eac4 890 buckets /= Config.Store.objectsPerBucket;
891 debugs(20, 1, "Target number of buckets: " << buckets);
892 /* ideally the full scan period should be configurable, for the
893 * moment it remains at approximately 24 hours. */
894 store_hash_buckets = storeKeyHashBuckets(buckets);
895 debugs(20, 1, "Using " << store_hash_buckets << " Store buckets");
57af1e3f 896 debugs(20, 1, "Max Mem size: " << ( Config.memMaxSize >> 10) << " KB" <<
9199139f 897 (Config.memShared ? " [shared]" : ""));
cc34568d 898 debugs(20, 1, "Max Swap size: " << (Store::Root().maxSize() >> 10) << " KB");
c8f4eac4 899
900 store_table = hash_create(storeKeyHashCmp,
901 store_hash_buckets, storeKeyHashHash);
902
13a07022 903 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
c8f4eac4 904 /* this starts a search of the store dirs, loading their
905 * index. under the new Store api this should be
906 * driven by the StoreHashIndex, not by each store.
bef81ea5 907 *
908 * That is, the HashIndex should perform a search of each dir it is
26ac0430 909 * indexing to do the hash insertions. The search is then able to
bef81ea5 910 * decide 'from-memory', or 'from-clean-log' or 'from-dirty-log' or
911 * 'from-no-log'.
912 *
c8f4eac4 913 * Step 1: make the store rebuilds use a search internally
bef81ea5 914 * Step 2: change the search logic to use the four modes described
915 * above
916 * Step 3: have the hash index walk the searches itself.
c8f4eac4 917 */
14911a4e
AR
918 if (dir(i).active())
919 store(i)->init();
13a07022 920 }
c8f4eac4 921}
922
12e11a5c 923uint64_t
c8f4eac4 924StoreHashIndex::maxSize() const
925{
12e11a5c 926 uint64_t result = 0;
c8f4eac4 927
39c1e1d9
DK
928 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
929 if (dir(i).doReportStat())
930 result += store(i)->maxSize();
931 }
c8f4eac4 932
933 return result;
934}
935
12e11a5c 936uint64_t
c8f4eac4 937StoreHashIndex::minSize() const
938{
12e11a5c 939 uint64_t result = 0;
c8f4eac4 940
39c1e1d9
DK
941 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
942 if (dir(i).doReportStat())
943 result += store(i)->minSize();
944 }
945
946 return result;
947}
948
949uint64_t
950StoreHashIndex::currentSize() const
951{
952 uint64_t result = 0;
953
954 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
955 if (dir(i).doReportStat())
956 result += store(i)->currentSize();
957 }
958
959 return result;
960}
961
962uint64_t
963StoreHashIndex::currentCount() const
964{
965 uint64_t result = 0;
966
967 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
968 if (dir(i).doReportStat())
969 result += store(i)->currentCount();
970 }
c8f4eac4 971
972 return result;
973}
974
af2fda07
DK
975int64_t
976StoreHashIndex::maxObjectSize() const
977{
978 int64_t result = -1;
979
980 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
981 if (dir(i).active() && store(i)->maxObjectSize() > result)
982 result = store(i)->maxObjectSize();
983 }
984
985 return result;
986}
987
93bc1434
AR
988void
989StoreHashIndex::getStats(StoreInfoStats &stats) const
990{
991 // accumulate per-disk cache stats
992 for (int i = 0; i < Config.cacheSwap.n_configured; i++) {
993 StoreInfoStats dirStats;
994 store(i)->getStats(dirStats);
995 stats += dirStats;
996 }
997
998 // common to all disks
999 stats.swap.open_disk_fd = store_open_disk_fd;
1000
1001 // memory cache stats are collected in StoreController::getStats(), for now
1002}
1003
c8f4eac4 1004void
1005StoreHashIndex::stat(StoreEntry & output) const
1006{
1007 int i;
1008
1009 /* Now go through each store, calling its stat routine */
1010
1011 for (i = 0; i < Config.cacheSwap.n_configured; i++) {
1012 storeAppendPrintf(&output, "\n");
1013 store(i)->stat(output);
1014 }
1015}
1016
1017void
4c973beb
AR
1018StoreHashIndex::reference(StoreEntry &e)
1019{
1020 e.store()->reference(e);
1021}
c8f4eac4 1022
4c973beb
AR
1023bool
1024StoreHashIndex::dereference(StoreEntry &e)
1025{
1026 return e.store()->dereference(e);
1027}
c8f4eac4 1028
1029void
1030StoreHashIndex::maintain()
1031{
1032 int i;
1033 /* walk each fs */
1034
1035 for (i = 0; i < Config.cacheSwap.n_configured; i++) {
1036 /* XXX FixMe: This should be done "in parallell" on the different
1037 * cache_dirs, not one at a time.
1038 */
1039 /* call the maintain function .. */
1040 store(i)->maintain();
1041 }
1042}
1043
c8f4eac4 1044void
1045StoreHashIndex::sync()
1046{
1047 for (int i = 0; i < Config.cacheSwap.n_configured; ++i)
1048 store(i)->sync();
1049}
1050
1051StoreSearch *
30abd221 1052StoreHashIndex::search(String const url, HttpRequest *)
c8f4eac4 1053{
1054 if (url.size())
1055 fatal ("Cannot search by url yet\n");
1056
1057 return new StoreSearchHashIndex (this);
1058}
1059
1060CBDATA_CLASS_INIT(StoreSearchHashIndex);
aa839030 1061
c8f4eac4 1062StoreSearchHashIndex::StoreSearchHashIndex(RefCount<StoreHashIndex> aSwapDir) : sd(aSwapDir), _done (false), bucket (0)
1063{}
1064
1065/* do not link
1066StoreSearchHashIndex::StoreSearchHashIndex(StoreSearchHashIndex const &);
1067*/
1068
1069StoreSearchHashIndex::~StoreSearchHashIndex()
1070{}
1071
1072void
70efcae0 1073StoreSearchHashIndex::next(void (aCallback)(void *), void *aCallbackData)
c8f4eac4 1074{
1075 next();
70efcae0 1076 aCallback (aCallbackData);
c8f4eac4 1077}
1078
1079bool
1080StoreSearchHashIndex::next()
1081{
1082 if (entries.size())
1083 entries.pop_back();
1084
1085 while (!isDone() && !entries.size())
1086 copyBucket();
1087
1088 return currentItem() != NULL;
1089}
1090
1091bool
1092StoreSearchHashIndex::error() const
1093{
1094 return false;
1095}
1096
1097bool
1098StoreSearchHashIndex::isDone() const
1099{
1100 return bucket >= store_hash_buckets || _done;
1101}
1102
1103StoreEntry *
1104StoreSearchHashIndex::currentItem()
1105{
1106 if (!entries.size())
1107 return NULL;
1108
1109 return entries.back();
1110}
1111
1112void
1113StoreSearchHashIndex::copyBucket()
1114{
1115 /* probably need to lock the store entries...
1116 * we copy them all to prevent races on the links. */
1117 debugs(47, 3, "StoreSearchHashIndex::copyBucket #" << bucket);
1118 assert (!entries.size());
1119 hash_link *link_ptr = NULL;
1120 hash_link *link_next = NULL;
1121 link_next = hash_get_bucket(store_table, bucket);
1122
1123 while (NULL != (link_ptr = link_next)) {
1124 link_next = link_ptr->next;
1125 StoreEntry *e = (StoreEntry *) link_ptr;
1126
1127 entries.push_back(e);
1128 }
1129
1130 bucket++;
1131 debugs(47,3, "got entries: " << entries.size());
1132}