bim

app/process-connection/process-connection.go → lab1/app/process-connection/process-connection.go

@@ -96,7 +96,6 @@ func processRate(conn net.Conn, serverListener net.Listener, serverConfig Config
 	trans := utilities.TranslateMessageToTransaction(mess)
 	utilities.PrintTransaction(trans)
 
-	// todo change this for cloud
 	address := strings.Split(conn.RemoteAddr().String(), ":")[0]
 	vote(serverListener, serverConfig, trans, address, objectStorage, amIRoot)
 }
@@ -106,10 +105,11 @@ func processRate(conn net.Conn, serverListener net.Listener, serverConfig Config
 // transactions which are the same as the one stored locally. This function relies on the broadcast by wave
 // with ACK distributed algorithm. However, some adjustments are needed to implement this function.
 func vote(server net.Listener, serverConfig Config, trans Transaction, parentAddress string, objectStorage Blob, amIRoot bool) AckTransaction {
-	var ack AckTransaction
-	ack.Id = ""
-	ack.TotalNodes = 0
-	ack.AmountOfCorrectNode = 0
+	ack := AckTransaction{
+		Id:                  "",
+		TotalNodes:          0,
+		AmountOfCorrectNode: 0,
+	}
 
 	var transactionToRate Transaction
 
@@ -225,7 +225,8 @@ func processVoteRequest(conn net.Conn, serverListener net.Listener, serverConfig
 		}
 	}
 
-	vote(serverListener, serverConfig, transToRate, "", objectStorage, amIRoot)
+	ack := vote(serverListener, serverConfig, transToRate, "", objectStorage, amIRoot)
+	utilities.PrintAckTransaction(ack)
 }
 
 func ProcessClient(conn net.Conn, server net.Listener, objectStorage Blob, serverConfig Config, amIRoot bool, mutex *sync.Mutex) {

app/utilities/utilities.go → lab1/app/utilities/utilities.go

@@ -98,14 +98,29 @@ func PrintConnection(connection net.Conn, connectionNumber int) {
 }
 
 func TranslateMessageToAckTransaction(mess Message) AckTransaction {
-	var newAck AckTransaction
+	var transactionId string
+	var amountOfCorrectNode int
+	var totalNodes int
+	var err error
 
 	var body map[string]interface{} = mess.MessageBody.(map[string]interface{})
-	newAck.Id = body["id"].(string)
-	newAck.AmountOfCorrectNode = body["amountOfCorrectNode"].(int)
-	newAck.TotalNodes = body["totalNodes"].(int)
+	transactionId = body["id"].(string)
+	amountOfCorrectNode, err = strconv.Atoi(body["amountOfCorrectNode"].(string))
+	if err != nil {
+		fmt.Println("Error while converting amountOfCorrectNode to int", err)
+		return AckTransaction{}
+	}
+	totalNodes, err = strconv.Atoi(body["totalNodes"].(string))
+	if err != nil {
+		fmt.Println("Error while converting totalNodes to int", err)
+		return AckTransaction{}
+	}
 
-	return newAck
+	return AckTransaction{
+		Id:                  transactionId,
+		AmountOfCorrectNode: amountOfCorrectNode,
+		TotalNodes:          totalNodes,
+	}
 }
 
 func TranslateMessageToTransaction(mess Message) Transaction {
@@ -131,3 +146,12 @@ func PrintTransactionDoneMessage() {
 	fmt.Println("All transactions have been received")
 	fmt.Println("***********************************")
 }
+
+func PrintAckTransaction(ack AckTransaction) {
+	fmt.Println("***********************************")
+	fmt.Println("Printing ack transaction")
+	fmt.Println("The id is ", ack.Id)
+	fmt.Println("The amount of correct node is ", ack.AmountOfCorrectNode)
+	fmt.Println("The total nodes is ", ack.TotalNodes)
+	fmt.Println("***********************************")
+}