mirror of
https://github.com/scottlamb/moonfire-nvr.git
synced 2025-01-07 13:03:22 -05:00
253f3de399
The filenames now represent composite ids (stream id + recording id) rather
than a separate uuid system with its own reservation for a few benefits:
* This provides more information when there are inconsistencies.
* This avoids the need for managing the reservations during recording. I
expect this to simplify delaying flushing of newly written sample files.
Now the directory has to be scanned at startup for files that never got
written to the database, but that's acceptably fast even with millions of
files.
* Less information to keep in memory and in the recording_playback table.
I'd considered using one directory per stream, which might help if the
filesystem has trouble coping with huge directories. But that would mean each
dir has to be fsync()ed separately (more latency and/or more multithreading).
So I'll stick with this until I see concrete evidence of a problem that would
solve.
Test coverage of the error conditions is poor. I plan to do some restructuring
of the db/dir code, hopefully making steps toward testability along the way.
105 lines
3.7 KiB
Rust
105 lines
3.7 KiB
Rust
// This file is part of Moonfire NVR, a security camera digital video recorder.
|
|
// Copyright (C) 2016 Scott Lamb <slamb@slamb.org>
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// In addition, as a special exception, the copyright holders give
|
|
// permission to link the code of portions of this program with the
|
|
// OpenSSL library under certain conditions as described in each
|
|
// individual source file, and distribute linked combinations including
|
|
// the two.
|
|
//
|
|
// You must obey the GNU General Public License in all respects for all
|
|
// of the code used other than OpenSSL. If you modify file(s) with this
|
|
// exception, you may extend this exception to your version of the
|
|
// file(s), but you are not obligated to do so. If you do not wish to do
|
|
// so, delete this exception statement from your version. If you delete
|
|
// this exception statement from all source files in the program, then
|
|
// also delete it here.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
use dir;
|
|
use docopt;
|
|
use error::Error;
|
|
use libc;
|
|
use rusqlite;
|
|
use std::path::Path;
|
|
|
|
mod check;
|
|
mod config;
|
|
mod init;
|
|
mod run;
|
|
mod ts;
|
|
mod upgrade;
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
pub enum Command {
|
|
Check,
|
|
Config,
|
|
Init,
|
|
Run,
|
|
Ts,
|
|
Upgrade,
|
|
}
|
|
|
|
impl Command {
|
|
pub fn run(&self) -> Result<(), Error> {
|
|
match *self {
|
|
Command::Check => check::run(),
|
|
Command::Config => config::run(),
|
|
Command::Init => init::run(),
|
|
Command::Run => run::run(),
|
|
Command::Ts => ts::run(),
|
|
Command::Upgrade => upgrade::run(),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(PartialEq, Eq)]
|
|
enum OpenMode {
|
|
ReadOnly,
|
|
ReadWrite,
|
|
Create
|
|
}
|
|
|
|
/// Locks and opens the database.
|
|
/// The returned `dir::Fd` holds the lock and should be kept open as long as the `Connection` is.
|
|
fn open_conn(db_dir: &str, mode: OpenMode) -> Result<(dir::Fd, rusqlite::Connection), Error> {
|
|
let dir = dir::Fd::open(None, db_dir, mode == OpenMode::Create)?;
|
|
let ro = mode == OpenMode::ReadOnly;
|
|
dir.lock(if ro { libc::LOCK_SH } else { libc::LOCK_EX } | libc::LOCK_NB)
|
|
.map_err(|e| Error{description: format!("db dir {:?} already in use; can't get {} lock",
|
|
db_dir,
|
|
if ro { "shared" } else { "exclusive" }),
|
|
cause: Some(Box::new(e))})?;
|
|
let conn = rusqlite::Connection::open_with_flags(
|
|
Path::new(&db_dir).join("db"),
|
|
match mode {
|
|
OpenMode::ReadOnly => rusqlite::OpenFlags::SQLITE_OPEN_READ_ONLY,
|
|
OpenMode::ReadWrite => rusqlite::OpenFlags::SQLITE_OPEN_READ_WRITE,
|
|
OpenMode::Create => {
|
|
rusqlite::OpenFlags::SQLITE_OPEN_READ_WRITE | rusqlite::OpenFlags::SQLITE_OPEN_CREATE
|
|
},
|
|
} |
|
|
// rusqlite::Connection is not Sync, so there's no reason to tell SQLite3 to use the
|
|
// serialized threading mode.
|
|
rusqlite::OpenFlags::SQLITE_OPEN_NO_MUTEX)?;
|
|
Ok((dir, conn))
|
|
}
|
|
|
|
fn parse_args<'a, T>(usage: &str) -> Result<T, Error> where T: ::serde::Deserialize<'a> {
|
|
Ok(docopt::Docopt::new(usage)
|
|
.and_then(|d| d.deserialize())
|
|
.unwrap_or_else(|e| e.exit()))
|
|
}
|