From 1e1d9dc179242f9e57b886303d9adee6bc7f94f0 Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Mon, 24 Apr 2017 15:17:38 +0200
Subject: [PATCH] Changed duration type to Duration

Signed-off-by: Jim Martens <github@2martens.de>
---
 02/src/de/dis2017/data/TenancyContract.java | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/02/src/de/dis2017/data/TenancyContract.java b/02/src/de/dis2017/data/TenancyContract.java
index 65a7968..d00d2a1 100644
--- a/02/src/de/dis2017/data/TenancyContract.java
+++ b/02/src/de/dis2017/data/TenancyContract.java
@@ -1,10 +1,11 @@
 package de.dis2017.data;
 
 import java.sql.Timestamp;
+import java.time.Duration;
 
 public class TenancyContract extends Contract {
 	private Timestamp startDate;
-	private Timestamp duration;
+	private Duration  duration;
 	private int       additionalCost;
 	
 	private int apartment;
@@ -17,13 +18,17 @@ public class TenancyContract extends Contract {
 		this.startDate = startDate;
 	}
 
-	public Timestamp getDuration() {
+	public Duration getDuration() {
 		return duration;
 	}
 
-	public void setDuration(Timestamp duration) {
+	public void setDuration(Duration duration) {
 		this.duration = duration;
 	}
+    
+    public void setDuration(Timestamp duration) {
+        this.duration = Duration.ofMillis(duration.getTime());
+    }
 
 	public int getAdditionalCost() {
 		return additionalCost;