From 02eedd6246e82ba0e708c72e0b79de93b0004d56 Mon Sep 17 00:00:00 2001
From: Mikhail Svetov <svmk-work@yandex.ru>
Date: Wed, 13 Jan 2016 21:18:07 +0700
Subject: [PATCH] Improvements.

---
 src/lib.rs   | 17 ++++++++++-------
 tests/lib.rs |  4 ++--
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 526e4b5..d669754 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -13,6 +13,7 @@ use std::cell::{Cell, RefCell};
 use url::Url;
 use hyper::{Client};
 use hyper::status::StatusCode;
+use std::time::Duration;
 
 /// A rule line is a single "Allow:" (allowance==True) or "Disallow:"
 /// (allowance==False) followed by a path."""
@@ -27,7 +28,7 @@ struct RuleLine {
 struct Entry {
     useragents: RefCell<Vec<String>>,
     rulelines: RefCell<Vec<RuleLine>>,
-    crawl_delay: Option<usize>,
+    crawl_delay: Option<Duration>,
 }
 
 #[derive(Debug, Eq, PartialEq, Clone)]
@@ -121,11 +122,11 @@ impl Entry {
         useragents.is_empty() && rulelines.is_empty()
     }
 
-    fn set_crawl_delay(&mut self,delay: usize) {
+    fn set_crawl_delay(&mut self,delay: Duration) {
         self.crawl_delay = Some(delay);
     }
 
-    fn get_crawl_delay(&self) -> Option<usize> {
+    fn get_crawl_delay(&self) -> Option<Duration> {
         return self.crawl_delay.clone();
     }
 }
@@ -285,8 +286,10 @@ impl RobotFileParser {
                             let delay = part1.parse::<f64>();
                             match delay {
                                 Ok(delay) => {
-                                    let delay = delay * 1000.0;
-                                    entry.set_crawl_delay(delay.round() as usize);
+                                    let delay_seconds = delay.trunc();
+                                    let delay_nanoseconds = delay.fract()* 10f64.powi(9);
+                                    let delay = Duration::new(delay_seconds as u64,delay_nanoseconds as u32);
+                                    entry.set_crawl_delay(delay);
                                 },
                                 Err(_) => {}
                             }
@@ -344,8 +347,8 @@ impl RobotFileParser {
         true
     }
 
-    /// Returns the crawl delay for this user agent as a `usize` in milliseconds, or None if no crawl delay is defined.
-    pub fn get_crawl_delay<T: AsRef<str>>(&self,useragent: T) -> Option<usize> {
+    /// Returns the crawl delay for this user agent as a `Duration`, or None if no crawl delay is defined.
+    pub fn get_crawl_delay<T: AsRef<str>>(&self,useragent: T) -> Option<Duration> {
         let useragent = useragent.as_ref();        
         if self.last_checked.get() == 0 {
             return None;
diff --git a/tests/lib.rs b/tests/lib.rs
index 6b4434b..ada37cb 100644
--- a/tests/lib.rs
+++ b/tests/lib.rs
@@ -1,10 +1,10 @@
 extern crate robotparser;
 
 use robotparser::RobotFileParser;
+use std::time::Duration;
 
 const AGENT: &'static str = "test_robotparser";
 
-
 fn robot_test(doc: &str, good_urls: Vec<&str>, bad_urls: Vec<&str>, agent: &str) {
     let parser = RobotFileParser::new("http://www.baidu.com/robots.txt");
     let lines: Vec<&str> = doc.split("\n").collect();
@@ -226,5 +226,5 @@ fn test_robots_text_crawl_delay() {
     Disallow: /search/\n";
     let lines: Vec<&str> = doc.split("\n").collect();
     parser.parse(&lines);
-    assert_eq!(2350, parser.get_crawl_delay("Yandex").unwrap());
+    assert_eq!(Duration::new(2,350 * 1000 * 1000), parser.get_crawl_delay("Yandex").unwrap());
 }