mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-07-14 23:45:44 +00:00
Shawn Tabrizi
ff5d56fb76
* cargo +nightly fmt * add cargo-fmt check to ci * update ci * fmt * fmt * skip macro * ignore bridges
105 lines
3.1 KiB
Rust
105 lines
3.1 KiB
Rust
// Copyright 2021 Parity Technologies (UK) Ltd.
|
|
// This file is part of Polkadot.
|
|
|
|
// Polkadot 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.
|
|
|
|
// Polkadot 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.
|
|
|
|
//! Migration code for the parachain's DB.
|
|
|
|
#![cfg(feature = "full-node")]
|
|
|
|
use std::{
|
|
fs, io,
|
|
path::{Path, PathBuf},
|
|
str::FromStr,
|
|
};
|
|
|
|
type Version = u32;
|
|
|
|
/// Version file name.
|
|
const VERSION_FILE_NAME: &'static str = "parachain_db_version";
|
|
|
|
/// Current db version.
|
|
const CURRENT_VERSION: Version = 1;
|
|
|
|
#[derive(thiserror::Error, Debug)]
|
|
pub enum Error {
|
|
#[error("I/O error when reading/writing the version")]
|
|
Io(#[from] io::Error),
|
|
#[error("The version file format is incorrect")]
|
|
CorruptedVersionFile,
|
|
#[error("Future version (expected {current:?}, found {got:?})")]
|
|
FutureVersion { current: Version, got: Version },
|
|
}
|
|
|
|
impl From<Error> for io::Error {
|
|
fn from(me: Error) -> io::Error {
|
|
match me {
|
|
Error::Io(e) => e,
|
|
_ => super::other_io_error(me.to_string()),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Try upgrading parachain's database to the current version.
|
|
pub fn try_upgrade_db(db_path: &Path) -> Result<(), Error> {
|
|
let is_empty = db_path.read_dir().map_or(true, |mut d| d.next().is_none());
|
|
if !is_empty {
|
|
match current_version(db_path)? {
|
|
0 => migrate_from_version_0_to_1(db_path)?,
|
|
CURRENT_VERSION => (),
|
|
v => return Err(Error::FutureVersion { current: CURRENT_VERSION, got: v }),
|
|
}
|
|
}
|
|
|
|
update_version(db_path)
|
|
}
|
|
|
|
/// Reads current database version from the file at given path.
|
|
/// If the file does not exist, assumes the current version.
|
|
fn current_version(path: &Path) -> Result<Version, Error> {
|
|
match fs::read_to_string(version_file_path(path)) {
|
|
Err(ref err) if err.kind() == io::ErrorKind::NotFound => Ok(CURRENT_VERSION),
|
|
Err(err) => Err(err.into()),
|
|
Ok(content) => u32::from_str(&content).map_err(|_| Error::CorruptedVersionFile),
|
|
}
|
|
}
|
|
|
|
/// Writes current database version to the file.
|
|
/// Creates a new file if the version file does not exist yet.
|
|
fn update_version(path: &Path) -> Result<(), Error> {
|
|
fs::create_dir_all(path)?;
|
|
fs::write(version_file_path(path), CURRENT_VERSION.to_string()).map_err(Into::into)
|
|
}
|
|
|
|
/// Returns the version file path.
|
|
fn version_file_path(path: &Path) -> PathBuf {
|
|
let mut file_path = path.to_owned();
|
|
file_path.push(VERSION_FILE_NAME);
|
|
file_path
|
|
}
|
|
|
|
/// Migration from version 0 to version 1:
|
|
/// * the number of columns has changed from 3 to 5;
|
|
fn migrate_from_version_0_to_1(path: &Path) -> Result<(), Error> {
|
|
use kvdb_rocksdb::{Database, DatabaseConfig};
|
|
|
|
let db_path = path
|
|
.to_str()
|
|
.ok_or_else(|| super::other_io_error("Invalid database path".into()))?;
|
|
let db_cfg = DatabaseConfig::with_columns(super::columns::v0::NUM_COLUMNS);
|
|
let db = Database::open(&db_cfg, db_path)?;
|
|
|
|
db.add_column()?;
|
|
db.add_column()?;
|
|
|
|
Ok(())
|
|
}
|