Доброе утро. При помощи нейросети писал код, который читает базу

данных в формате .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 мбайт. Т.е. явно не все строки. Подскажите, что не так?

2 ответов

16 просмотров

1) Подсунь ей короткий SQL-файл и посмотри в отладке, 2) напиши тесты

krakotay- Автор вопроса
Денис Артемов
1) Подсунь ей короткий SQL-файл и посмотри в отлад...

Всё, проблема уже решена. Вот правильный код 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(()) }

Похожие вопросы

Обсуждают сегодня

Всем привет. Ребята, подскажите, пожалуйста. у ботов есть ограничение на отправку сообщений - 30 сообщений в секунду, эти ограничения накладываются на все сообщения? или на со...
Artem Stormageddon
4
Ребята, всем привет. Подскажите, пожалуйста, можно ли как-то через бота понять, что этого бота добавили в группу\канал и выдали ему права администратора?
Artem Stormageddon
9
Привет всем! Почему этот код не срабатывает при добавлении или удалении пользователя из чата? bot.on('chat_member', async (ctx) => { console.log(ctx); }) bot.launch({allo...
Alexander
7
Всем привет. Не понимаю, в чём тут шутка юмора. Убирается только разрешение на send_messages. А send_media_messages остаётся. Как сделать, чтобы оба убирались? await b...
Alexander
2
Есть тут кто занимается разработкой серваков майна? Или знакомые
meow *
3
'frakturBold' => ['𝖆', '𝖇', '𝖈', '𝖉', '𝖊', '𝖋', '𝖌', '𝖍', '𝖎', '𝖏', '𝖐', '𝖑', '𝖒', '𝖓', '𝖔', '𝖕', '𝖖', '𝖗', '𝖘', '𝖙', '𝖚', '𝖛', '𝖜', '𝖝', '𝖞', '𝖟', '𝕬', '𝕭', '𝕮', '𝕯'...
Roma
4
Есть ли лимиты на кол-во вебхук по домену? Стоит в данный момент 900+ ботов и бывает бот перестает отвечать (не приходят вебхуки) 🐒 Помогает только перезапуск
ᅠ [ Кому не ответил, дублируйте ]
11
Товарищи, здравствуйте Подскажите, пожалуйста, может кто-нибудь сталкивался с такой задачей Через вебапку можно сканировать qr-код, а есть ли возможность считывать nfc?
Artem Stormageddon
8
а что делать если тебя убивают на картах?
Yarik yarik kyda ti lezesh
43
Визуальное отображение моделей таблиц sql какое посоветуете?
Shen
7
Карта сайта