관리-도구

편집 파일: swoole_file.h

/*
  +----------------------------------------------------------------------+
  | Swoole                                                               |
  +----------------------------------------------------------------------+
  | This source file is subject to version 2.0 of the Apache license,    |
  | that is bundled with this package in the file LICENSE, and is        |
  | available through the world-wide-web at the following url:           |
  | http://www.apache.org/licenses/LICENSE-2.0.html                      |
  | If you did not receive a copy of the Apache2.0 license and are unable|
  | to obtain it through the world-wide-web, please send a note to       |
  | license@swoole.com so we can mail you a copy immediately.            |
  +----------------------------------------------------------------------+
  | Author: Tianfeng Han  <rango@swoole.com>                             |
  +----------------------------------------------------------------------+
*/

#pragma once

#include "swoole.h"
#include "swoole_string.h"

#include <sys/file.h>
#include <sys/stat.h>
#include <fcntl.h>

namespace swoole {

ssize_t file_get_size(FILE *fp);
ssize_t file_get_size(int fd);
ssize_t file_get_size(const std::string &filename);
std::shared_ptr<String> file_get_contents(const std::string &filename);
bool file_put_contents(const std::string &filename, const char *content, size_t length);
bool file_exists(const std::string &filename);

typedef struct stat FileStatus;

class File {
  private:
    int fd_;
    int flags_;
    std::string path_;

public:
    enum Flag {
        READ = O_RDONLY,
        WRITE = O_WRONLY,
        RW = O_RDWR,
        CREATE = O_CREAT,
        EXCL = O_EXCL,
        APPEND = O_APPEND,
    };

explicit File(int fd) {
        fd_ = fd;
        flags_ = 0;
    }

File(int fd, const std::string &path) {
        fd_ = fd;
        path_ = path;
        flags_ = 0;
    }

File(const std::string &path, int oflags) {
        fd_ = ::open(path.c_str(), oflags);
        path_ = path;
        flags_ = oflags;
    }

File(const std::string &path, int oflags, int mode) {
        fd_ = ::open(path.c_str(), oflags, mode);
        path_ = path;
        flags_ = oflags;
    }

~File() {
        if (fd_ >= 0) {
            ::close(fd_);
        }
    }

bool ready() {
        return fd_ != -1;
    }

ssize_t write(const void *__buf, size_t __n) const {
        return ::write(fd_, __buf, __n);
    }

ssize_t read(void *__buf, size_t __n) const {
        return ::read(fd_, __buf, __n);
    }

ssize_t pwrite(const void *__buf, size_t __n, off_t __offset) const {
        return ::pwrite(fd_, __buf, __n, __offset);
    }

ssize_t pread(void *__buf, size_t __n, off_t __offset) const {
        return ::pread(fd_, __buf, __n, __offset);
    }

size_t write_all(const void *__buf, size_t __n);
    size_t read_all(void *__buf, size_t __n);

std::shared_ptr<String> read_content();

bool stat(FileStatus *_stat) const {
        if (::fstat(fd_, _stat) < 0) {
            swoole_sys_warning("fstat() failed");
            return false;
        } else {
            return true;
        }
    }

bool sync() {
        return ::fsync(fd_) == 0;
    }

bool truncate(size_t size) {
        return ::ftruncate(fd_, size) == 0;
    }

off_t set_offest(off_t offset) {
        return lseek(fd_, offset, SEEK_SET);
    }

off_t get_offset() {
        return lseek(fd_, 0, SEEK_CUR);
    }

bool lock(int operation) {
        return ::flock(fd_, operation) == 0;
    }

bool unlock() {
        return ::flock(fd_, LOCK_UN) == 0;
    }

ssize_t get_size() {
        return file_get_size(fd_);
    }

bool close() {
        if (fd_ == -1) {
            return false;
        }
        int tmp_fd = fd_;
        fd_ = -1;
        return ::close(tmp_fd) == 0;
    }

void release() {
        fd_ = -1;
    }

int get_fd() {
        return fd_;
    }

const std::string &get_path() {
        return path_;
    }

static bool exists(const std::string &file) {
        return access(file.c_str(), R_OK) == 0;
    }
};

File make_tmpfile();

}  // namespace swoole