summaryrefslogtreecommitdiff
path: root/util/file_allocator.cpp
diff options
context:
space:
mode:
authorAntonin Kral <a.kral@bobek.cz>2012-08-29 20:54:51 +0200
committerAntonin Kral <a.kral@bobek.cz>2012-08-29 20:54:51 +0200
commit83957b73f9177f6e38bd5375bd93ca1f6a47188c (patch)
treef20b7d6ac9a9c64ff5bb6b5910a24abbb356b1d5 /util/file_allocator.cpp
parent5071d203970edd4c995493d810abe20987e76fe9 (diff)
Imported Upstream version 2.2.0upstream/2.2.0
Diffstat (limited to 'util/file_allocator.cpp')
-rw-r--r--util/file_allocator.cpp329
1 files changed, 0 insertions, 329 deletions
diff --git a/util/file_allocator.cpp b/util/file_allocator.cpp
deleted file mode 100644
index b0572f971bd..00000000000
--- a/util/file_allocator.cpp
+++ /dev/null
@@ -1,329 +0,0 @@
-// @file file_allocator.cpp
-
-/* Copyright 2009 10gen Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-#include "pch.h"
-#include <fcntl.h>
-#include <errno.h>
-
-#if defined(__freebsd__) || defined(__openbsd__)
-#include <sys/stat.h>
-#endif
-
-#include "timer.h"
-#include "mongoutils/str.h"
-using namespace mongoutils;
-
-#ifndef O_NOATIME
-#define O_NOATIME (0)
-#endif
-
-#include "file_allocator.h"
-#include "paths.h"
-
-namespace mongo {
-
- boost::filesystem::path ensureParentDirCreated(const boost::filesystem::path& p){
- const boost::filesystem::path parent = p.branch_path();
-
- if (! boost::filesystem::exists(parent)){
- ensureParentDirCreated(parent);
- log() << "creating directory " << parent.string() << endl;
- boost::filesystem::create_directory(parent);
- flushMyDirectory(parent); // flushes grandparent to ensure parent exists after crash
- }
-
- assert(boost::filesystem::is_directory(parent));
- return parent;
- }
-
-#if defined(_WIN32)
- FileAllocator::FileAllocator() {
- }
-
- void FileAllocator::start() {
- }
-
- void FileAllocator::requestAllocation( const string &name, long &size ) {
- /* Some of the system calls in the file allocator don't work in win,
- so no win support - 32 or 64 bit. Plus we don't seem to need preallocation
- on windows anyway as we don't have to pre-zero the file there.
- */
- }
-
- void FileAllocator::allocateAsap( const string &name, unsigned long long &size ) {
- // no-op
- }
-
- void FileAllocator::waitUntilFinished() const {
- // no-op
- }
-
- void FileAllocator::ensureLength(int fd , long size) {
- // we don't zero on windows
- // TODO : we should to avoid fragmentation
- }
-
- bool FileAllocator::hasFailed() const {
- return false;
- }
-
-#else
-
- FileAllocator::FileAllocator()
- : _pendingMutex("FileAllocator"), _failed() {
- }
-
-
- void FileAllocator::start() {
- boost::thread t( boost::bind( &FileAllocator::run , this ) );
- }
-
- void FileAllocator::requestAllocation( const string &name, long &size ) {
- scoped_lock lk( _pendingMutex );
- if ( _failed )
- return;
- long oldSize = prevSize( name );
- if ( oldSize != -1 ) {
- size = oldSize;
- return;
- }
- _pending.push_back( name );
- _pendingSize[ name ] = size;
- _pendingUpdated.notify_all();
- }
-
- void FileAllocator::allocateAsap( const string &name, unsigned long long &size ) {
- scoped_lock lk( _pendingMutex );
- long oldSize = prevSize( name );
- if ( oldSize != -1 ) {
- size = oldSize;
- if ( !inProgress( name ) )
- return;
- }
- checkFailure();
- _pendingSize[ name ] = size;
- if ( _pending.size() == 0 )
- _pending.push_back( name );
- else if ( _pending.front() != name ) {
- _pending.remove( name );
- list< string >::iterator i = _pending.begin();
- ++i;
- _pending.insert( i, name );
- }
- _pendingUpdated.notify_all();
- while( inProgress( name ) ) {
- checkFailure();
- _pendingUpdated.wait( lk.boost() );
- }
-
- }
-
- void FileAllocator::waitUntilFinished() const {
- if ( _failed )
- return;
- scoped_lock lk( _pendingMutex );
- while( _pending.size() != 0 )
- _pendingUpdated.wait( lk.boost() );
- }
-
- void FileAllocator::ensureLength(int fd , long size) {
-#if defined(__linux__)
- int ret = posix_fallocate(fd,0,size);
- if ( ret == 0 )
- return;
-
- log() << "FileAllocator: posix_fallocate failed: " << errnoWithDescription( ret ) << " falling back" << endl;
-#endif
-
- off_t filelen = lseek(fd, 0, SEEK_END);
- if ( filelen < size ) {
- if (filelen != 0) {
- stringstream ss;
- ss << "failure creating new datafile; lseek failed for fd " << fd << " with errno: " << errnoWithDescription();
- uassert( 10440 , ss.str(), filelen == 0 );
- }
- // Check for end of disk.
-
- uassert( 10441 , str::stream() << "Unable to allocate new file of size " << size << ' ' << errnoWithDescription(),
- size - 1 == lseek(fd, size - 1, SEEK_SET) );
- uassert( 10442 , str::stream() << "Unable to allocate new file of size " << size << ' ' << errnoWithDescription(),
- 1 == write(fd, "", 1) );
- lseek(fd, 0, SEEK_SET);
-
- const long z = 256 * 1024;
- const boost::scoped_array<char> buf_holder (new char[z]);
- char* buf = buf_holder.get();
- memset(buf, 0, z);
- long left = size;
- while ( left > 0 ) {
- long towrite = left;
- if ( towrite > z )
- towrite = z;
-
- int written = write( fd , buf , towrite );
- uassert( 10443 , errnoWithPrefix("FileAllocator: file write failed" ), written > 0 );
- left -= written;
- }
- }
- }
-
- bool FileAllocator::hasFailed() const {
- return _failed;
- }
-
- void FileAllocator::checkFailure() {
- if (_failed) {
- // we want to log the problem (diskfull.js expects it) but we do not want to dump a stack tracke
- msgassertedNoTrace( 12520, "new file allocation failure" );
- }
- }
-
- long FileAllocator::prevSize( const string &name ) const {
- if ( _pendingSize.count( name ) > 0 )
- return _pendingSize[ name ];
- if ( boost::filesystem::exists( name ) )
- return boost::filesystem::file_size( name );
- return -1;
- }
-
- // caller must hold _pendingMutex lock.
- bool FileAllocator::inProgress( const string &name ) const {
- for( list< string >::const_iterator i = _pending.begin(); i != _pending.end(); ++i )
- if ( *i == name )
- return true;
- return false;
- }
-
- string makeTempFileName( path root ) {
- while( 1 ) {
- path p = root / "_tmp";
- stringstream ss;
- ss << (unsigned) rand();
- p /= ss.str();
- string fn = p.string();
- if( !boost::filesystem::exists(p) )
- return fn;
- }
- return "";
- }
-
- void FileAllocator::run( FileAllocator * fa ) {
- setThreadName( "FileAllocator" );
- while( 1 ) {
- {
- scoped_lock lk( fa->_pendingMutex );
- if ( fa->_pending.size() == 0 )
- fa->_pendingUpdated.wait( lk.boost() );
- }
- while( 1 ) {
- string name;
- long size;
- {
- scoped_lock lk( fa->_pendingMutex );
- if ( fa->_pending.size() == 0 )
- break;
- name = fa->_pending.front();
- size = fa->_pendingSize[ name ];
- }
-
- string tmp;
- long fd = 0;
- try {
- log() << "allocating new datafile " << name << ", filling with zeroes..." << endl;
-
- boost::filesystem::path parent = ensureParentDirCreated(name);
- tmp = makeTempFileName( parent );
- ensureParentDirCreated(tmp);
-
- fd = open(tmp.c_str(), O_CREAT | O_RDWR | O_NOATIME, S_IRUSR | S_IWUSR);
- if ( fd <= 0 ) {
- log() << "FileAllocator: couldn't create " << name << " (" << tmp << ") " << errnoWithDescription() << endl;
- uasserted(10439, "");
- }
-
-#if defined(POSIX_FADV_DONTNEED)
- if( posix_fadvise(fd, 0, size, POSIX_FADV_DONTNEED) ) {
- log() << "warning: posix_fadvise fails " << name << " (" << tmp << ") " << errnoWithDescription() << endl;
- }
-#endif
-
- Timer t;
-
- /* make sure the file is the full desired length */
- ensureLength( fd , size );
-
- close( fd );
- fd = 0;
-
- if( rename(tmp.c_str(), name.c_str()) ) {
- log() << "error: couldn't rename " << tmp << " to " << name << ' ' << errnoWithDescription() << endl;
- uasserted(13653, "");
- }
- flushMyDirectory(name);
-
- log() << "done allocating datafile " << name << ", "
- << "size: " << size/1024/1024 << "MB, "
- << " took " << ((double)t.millis())/1000.0 << " secs"
- << endl;
-
- // no longer in a failed state. allow new writers.
- fa->_failed = false;
- }
- catch ( ... ) {
- if ( fd > 0 )
- close( fd );
- log() << "error failed to allocate new file: " << name
- << " size: " << size << ' ' << errnoWithDescription() << warnings;
- log() << " will try again in 10 seconds" << endl; // not going to warning logs
- try {
- if ( tmp.size() )
- BOOST_CHECK_EXCEPTION( boost::filesystem::remove( tmp ) );
- BOOST_CHECK_EXCEPTION( boost::filesystem::remove( name ) );
- }
- catch ( ... ) {
- }
- scoped_lock lk( fa->_pendingMutex );
- fa->_failed = true;
- // not erasing from pending
- fa->_pendingUpdated.notify_all();
-
-
- sleepsecs(10);
- continue;
- }
-
- {
- scoped_lock lk( fa->_pendingMutex );
- fa->_pendingSize.erase( name );
- fa->_pending.pop_front();
- fa->_pendingUpdated.notify_all();
- }
- }
- }
- }
-
-#endif
-
- FileAllocator* FileAllocator::_instance = 0;
-
- FileAllocator* FileAllocator::get(){
- if ( ! _instance )
- _instance = new FileAllocator();
- return _instance;
- }
-
-} // namespace mongo