From efc17189c0ea3ac46b929c67704111351bbb8b5e Mon Sep 17 00:00:00 2001
From: Matthew R <matthew@staff.libraryofcode.org>
Date: Fri, 20 Dec 2019 12:25:35 -0500
Subject: [PATCH] move go files to another directory

---
 src/{intervals => go}/dirsize.go |  5 +++--
 src/{intervals => go}/storage.go | 23 ++++++++++++++---------
 2 files changed, 17 insertions(+), 11 deletions(-)
 rename src/{intervals => go}/dirsize.go (77%)
 rename src/{intervals => go}/storage.go (77%)

diff --git a/src/intervals/dirsize.go b/src/go/dirsize.go
similarity index 77%
rename from src/intervals/dirsize.go
rename to src/go/dirsize.go
index f54802c..4f2e8b7 100644
--- a/src/intervals/dirsize.go
+++ b/src/go/dirsize.go
@@ -6,7 +6,7 @@ import (
 )
 
 // DirSize This function returns the total size of a directory in bytes.
-func DirSize(path string) float64 {
+func DirSize(path* string) float64 {
 	var dirSize int64 = 0
 
 	readSize := func(path string, file os.FileInfo, err error) error {
@@ -17,7 +17,8 @@ func DirSize(path string) float64 {
 		return nil
 	}
 
-	filepath.Walk(path, readSize)
+	err := filepath.Walk(*path, readSize)
+	HandleError(err, 0)
 
 	size := float64(dirSize)
 
diff --git a/src/intervals/storage.go b/src/go/storage.go
similarity index 77%
rename from src/intervals/storage.go
rename to src/go/storage.go
index a6dad39..954e97d 100644
--- a/src/intervals/storage.go
+++ b/src/go/storage.go
@@ -9,7 +9,8 @@ import (
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
 	"io/ioutil"
-	"time"
+  "time"
+  "strings"
 )
 
 // Collection the MongoDB Account collection
@@ -55,6 +56,8 @@ func main() {
   fmt.Printf("Connected to MongoDB [GO]\n")
 	HandleError(err, 1)
 
+  Collection = client.Database("cloudservices").Collection("accounts")
+
 	RedisClient = redis.NewClient(&redis.Options{
 		Addr:     "localhost:6379",
 		Password: "",
@@ -73,7 +76,7 @@ func main() {
 }
 
 func handler() {
-	cur, err := Collection.Find(context.TODO(), bson.M{})
+	cur, err := Collection.Find(context.TODO(), bson.D{})
 	HandleError(err, 0)
 
 	for cur.Next(context.TODO()) {
@@ -84,11 +87,13 @@ func handler() {
 }
 
 func checkAccountSizeAndUpdate(username string, id string) {
-  var bytes float64 = 0
-  sizeHome := DirSize("/home/" + username)
-  bytes += sizeHome
-  sizeMail := DirSize("/var/mail/" + username)
-  bytes += sizeMail
-  RedisClient.Set("storage"+"-"+id, bytes, 0)
-  fmt.Printf("Set Call | Username: %v, ID: %v | Bytes: %f\n", username, id, bytes)
+  var size float64 = 0
+  var userHomeDirectory string = strings.Replace(strings.Join([]string{"/home/", string(username)}, ""), "\"", "", -1)
+  fmt.Println(userHomeDirectory)
+  sizeHome := DirSize(&userHomeDirectory)
+  size += sizeHome
+  sizeMail := DirSize(&userHomeDirectory)
+  size += sizeMail
+  RedisClient.Set("storage"+"-"+string(id), size, 0)
+  fmt.Printf("Set Call | Username: %v, ID: %v | Bytes: %f\n", string(username), string(id), size)
 }