данных в формате .sql и экспортирует в csv файл
use std::fs::File;
use std::io::{BufRead, BufReader};
fn main() -> std::io::Result<()> {
// Открытие SQL-дампа и CSV-файла
let sql_file = File::open("db.sql")?;
let sql_reader = BufReader::new(sql_file);
let mut csv_file = File::create("output.csv")?;
let mut csv_writer = csv::Writer::from_writer(&mut csv_file);
// Чтение SQL-дампа и запись данных в CSV-файл
for line in sql_reader.lines() {
let line = line.unwrap();
// Пропускаем строки комментариев и пустые строки
if line.starts_with("--") || line.starts_with("/*") || line.trim().is_empty() {
continue;
}
// Определяем тип SQL-команды
if line.starts_with("CREATE TABLE") {
// Игнорируем команду CREATE TABLE
continue;
} else if line.starts_with("INSERT INTO") {
// Извлекаем данные из команды INSERT INTO
let values_start = line.find("VALUES (").unwrap() + "VALUES (".len();
let values_end = line.rfind(")").unwrap();
let values = &line[values_start..values_end];
// Запись данных в CSV-файл
let values = values.replace("),(", "\n").replace("(", "").replace(")", "");
for row in values.split('\n') {
let fields: Vec<&str> = row.split(',').collect();
csv_writer.write_record(fields)?;
}
}
}
csv_writer.flush()?;
println!("Данные успешно экспортированы в CSV-файл.");
Ok(())
}
Но проблема - читает файл размером 6 гигов, а csv файл получается размером 5 мбайт. Т.е. явно не все строки. Подскажите, что не так?
1) Подсунь ей короткий SQL-файл и посмотри в отладке, 2) напиши тесты
Всё, проблема уже решена. Вот правильный код use std::fs::File; use std::io::{self, BufRead, BufReader, Write}; fn main() -> io::Result<()> { // Открытие файла db.sql let file = File::open("db.sql")?; let reader = BufReader::new(file); // Создание CSV-файла для экспорта let mut csv_file = File::create("export.csv")?; // Флаг для определения начала данных таблицы let mut found_table = false; // Чтение файла построчно for line in reader.lines() { let line = line?; // Поиск начала данных таблицы if found_table && line.starts_with("INSERT INTO") { let values = line.trim_start_matches("INSERT INTO") .trim_end_matches(";"); // Разделение значений по запятой let values: Vec<&str> = values.split(",").collect(); // Запись значений в CSV-файл writeln!(csv_file, "{}", values.join(","))?; } // Поиск начала данных таблицы if !found_table && line.starts_with("INSERT INTO") { found_table = true; } } Ok(()) }
Обсуждают сегодня