added function and refactored the code

1 files changed, 29 insertions, 0 deletions

diff --git a/datastructs.go b/datastructs.go
new file mode 100644
index 0000000..c885dea
--- /dev/null
+++ b/datastructs.go
@@ -0,0 +1,29 @@
+package main
+
+type RawData struct {
+	ipDst      string
+	time       time.Time
+	port       int
+	packetSize int
+}
+
+func (rd *RawData) getVolSize() string {
+	return "medium"
+}
+
+type CleanData struct {
+	ipbSrc     string
+	ipbDst     string
+	time       time.Time
+	port       int
+	volume     string
+	occurances int
+}
+
+func (cd *CleanData) equals(other *CleanData) bool {
+	return cd.ipdbSrc == other.ipbSrc &&
+		cd.ipbDst == other.ipbDst &&
+		cd.time == other.time &&
+		cd.port == other.port &&
+		cd.volume == other.volume
+}