file writing

Author: whizsid

src/main.rs

@@ -21,13 +21,21 @@ use quick_xml::Reader;
 use std::env;
 use std::path::Path;
 
+pub mod sql_file;
+use sql_file::NodeFile;
+
 fn main() {
+
  let args: Vec<_> = env::args().collect();
 
  let formated_arg = Arguments::parse_args(args);
 
  let result = Reader::from_file(&Path::new(&formated_arg.input_file));
 
+ let nodes_table = NodeFile {..NodeFile::new(formated_arg.output_dir,formated_arg.maximum_rows_per_query)};
+
+ println!("One file is created:- {:?}",nodes_table.sql.rows);
+
  // let mut count = 0;
  let mut buf = Vec::new();
  match result {

src/sql_file.rs

@@ -1,5 +1,53 @@
+use std::fs::File;
+use std::fs::OpenOptions;
+use std::io::prelude::*;
+
+use std::path::PathBuf;
+
 pub struct SqlFile {
  pub table_name: String,
  pub rows: i32,
- pub sets: i32
-}
+ pub sets: i32,
+ pub maximum_rows_per_query: i32
+}
+
+pub struct NodeFile {
+ pub sql:SqlFile,
+ pub file: File
+}
+
+impl NodeFile {
+ pub fn new(output_dir:String,maximum_rows_per_query:i32)->NodeFile{
+ let file_name = String::from("nodes.sql");
+
+ let file_path = PathBuf::from(output_dir).join(file_name);
+
+ let mut file = OpenOptions::new()
+ .write(true)
+ .append(true)
+ .create_new(true)
+ .open(file_path)
+ .unwrap();
+
+ if let Err(e) = writeln!(file, "DROP TABLE IF EXISTS nodes;") {
+ drop(file);
+ panic!("Could not to write to the file: {:?}",e);
+ }
+
+ if let Err(e) = writeln!(file, "CREATE TABLE nodes ();") {
+ drop(file);
+ panic!("Could not to write to the file: {:?}",e);
+ }
+
+ return NodeFile {
+ sql: SqlFile {
+ table_name: String::from("nodes"),
+ rows: 0,
+ sets: 0,
+ maximum_rows_per_query:maximum_rows_per_query
+ },
+ file: file
+ }
+ }
+}
+