added azure storage handler

local/client/client.go

@@ -3,7 +3,7 @@ package main
 import (
 	"encoding/json"
 	"fmt"
-	types "local/types"
+	types "local/helper"
 	"net"
 	"os"
 	"strconv"

local/helper/azure_handler.go

+package helper
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"github.com/Azure/azure-sdk-for-go/sdk/storage/azblob"
+	dotenv "github.com/joho/godotenv"
+	"io"
+	"os"
+)
+
+var client *azblob.Client
+var err error
+var ctx context.Context
+
+// test program for the object storage module
+func main() {
+	err := dotenv.Load() //load .env file
+	if err != nil {
+		panic(err)
+	}
+	var connectionString = os.Getenv("AZ_STORAGE_CONNECTION_STRING")
+	var containerName = ""
+
+	ConnectToStorage(connectionString)
+	ListBlobsOnContainer(containerName, nil)
+
+	data := ReadFromStorage(containerName, "test.json")
+	var transactions []Transaction
+	err = json.Unmarshal(data, &transactions)
+	if err != nil {
+		panic(err)
+	}
+	fmt.Printf("%s\n", transactions)
+}
+
+func ConnectToStorage(connectionString string) {
+	client, err = azblob.NewClientFromConnectionString(connectionString, nil)
+	if err != nil {
+		panic(err)
+	}
+	ctx = context.Background()
+}
+
+func ListBlobsOnContainer(containerName string, options *azblob.ListBlobsFlatOptions) {
+	fmt.Printf("Listing blobs on container [%s] ...\n", containerName)
+	pager := client.NewListBlobsFlatPager(containerName, options)
+	for pager.More() {
+		resp, err := pager.NextPage(ctx)
+		if err != nil {
+			panic(err)
+		}
+		for _, _blob := range resp.Segment.BlobItems {
+			fmt.Printf("%s\n", *_blob.Name)
+		}
+	}
+}
+
+func ReadFromStorage(containerName string, blobName string) []byte {
+	resp, err := client.DownloadStream(ctx, containerName, blobName, nil)
+	if err != nil {
+		panic(err)
+	}
+	blobData, err := io.ReadAll(resp.Body)
+	if err != nil {
+		panic(err)
+	}
+	return blobData
+}
+
+func WriteToStorage() {
+	//TODO: Implements
+
+}
+
+/*
+jsonString must be formatted like that :
+[{},{},{}] with each object inside the array having the properties as named on the Transaction type of the types.go file
+*/
+func JsonDataToTransactions(jsonString []byte) []Transaction {
+	var transactions []Transaction
+	err = json.Unmarshal(jsonString, &transactions)
+	if err != nil {
+		panic(err)
+	}
+	return transactions
+}

local/types/types.go → local/helper/types.go

-package types
+package helper
 
 import "fmt"
 

local/server/azure_handler.go

-package main
-
-import (
-	azblob "github.com/Azure/azure-sdk-for-go/sdk/storage/azblob"
-)
-
-var client *azblob.Client
-var err error
-
-func connectToStorage(connectionString string) {
-	client, err = azblob.NewClientFromConnectionString(connectionString, nil)
-	if err != nil {
-		panic(err)
-	}
-}
-
-func readFromStorage() {
-	//TODO: Implements
-}
-
-func writeToStorage() {
-	//TODO: Implements
-}

local/server/server.go

@@ -3,8 +3,9 @@ package main
 import (
 	"encoding/json"
 	"fmt"
+	dotenv "github.com/joho/godotenv"
 	"gopkg.in/yaml.v2"
-	types "local/types"
+	helper "local/helper"
 	"net"
 	"os"
 )
@@ -12,14 +13,14 @@ import (
 type Config struct {
 	Id         int           `yaml:"id"`
 	Address    string        `yaml:"address"`
-	Neighbours []types.Node `yaml:"neighbours"`
+	Neighbours []helper.Node `yaml:"neighbours"`
 }
 
-var transactions []types.Transaction
+var transactions []helper.Transaction
 
 var config Config
 
-var node types.CurrentNode
+var node helper.CurrentNode
 
 const (
 	ServerPort = "8017"
@@ -30,10 +31,14 @@ var server net.Listener
 var proactive string
 
 func main() {
+	dotenv.Load()
+
 	args := os.Args
 	yaml := fmt.Sprintf("../neighbour-%s.yaml", args[1])
 	config := parseYAML(yaml)
 	initNodeFromConfig(config)
+	connectionString := os.Getenv("AZ_STORAGE_CONNECTION_STRING")
+	helper.ConnectToStorage(connectionString)
 
 	fmt.Printf("Loaded yaml file: %s\n", yaml)
 
@@ -59,7 +64,7 @@ func main() {
 }
 
 func initNodeFromConfig(config Config) {
-	node = types.CurrentNode{
+	node = helper.CurrentNode{
 		Address:      config.Address,
 		Neighbours:   config.Neighbours,
 		Reached:      false,
@@ -77,7 +82,7 @@ func processClient(connection net.Conn) {
 		fmt.Println("Error reading:", err.Error())
 	}
 
-	var m types.Message
+	var m helper.Message
 	err = json.Unmarshal(buffer[:mlen], &m)
 	if err != nil {
 		panic(err)
@@ -86,10 +91,10 @@ func processClient(connection net.Conn) {
 	fmt.Printf("%s\n", m.String())
 
 	switch m.Action {
-	case types.MCreateTransaction:
+	case helper.MCreateTransaction:
 		createTransaction(connection, m.Transaction)
 		break
-	case types.MListTransactions:
+	case helper.MListTransactions:
 		listTransactions(connection)
 		break
 	default:
@@ -113,7 +118,7 @@ func parseYAML(file string) Config {
 	return config
 }
 
-func createTransaction(connection net.Conn, t types.Transaction) {
+func createTransaction(connection net.Conn, t helper.Transaction) {
 	if !isTransactionKnown(t) {
 		transactions = append(transactions, t)
 		broadcastTransaction(connection, t)
@@ -121,8 +126,8 @@ func createTransaction(connection net.Conn, t types.Transaction) {
 }
 
 func listTransactions(connection net.Conn) {
-	m := types.Message{
-		Action:       types.MListTransactions,
+	m := helper.Message{
+		Action:       helper.MListTransactions,
 		Proactive:    proactive,
 		Transactions: transactions,
 	}
@@ -135,7 +140,7 @@ func listTransactions(connection net.Conn) {
 	defer connection.Close()
 }
 
-func broadcastTransaction(connection net.Conn, t types.Transaction) {
+func broadcastTransaction(connection net.Conn, t helper.Transaction) {
 	fmt.Printf("Starting transaction broadcast\n")
 
 	sendTransactionToNeighbours(t)
@@ -145,21 +150,21 @@ func broadcastTransaction(connection net.Conn, t types.Transaction) {
 
 }
 
-func sendTransactionToNeighbours(t types.Transaction) {
+func sendTransactionToNeighbours(t helper.Transaction) {
 	for _, n := range node.Neighbours {
 		sendTransactionToServer(t, n.Address)
 	}
 }
 
-func sendTransactionToServer(t types.Transaction, address string) {
+func sendTransactionToServer(t helper.Transaction, address string) {
 	fmt.Printf("Sending transaction %s to server [%s]\n", t.String(), address)
 	connection, err := net.Dial(ServerType, fmt.Sprintf("%s:%s", address, ServerPort))
 	if err != nil {
 		panic(err)
 	}
 
-	m := types.Message{
-		Action:      types.MCreateTransaction,
+	m := helper.Message{
+		Action:      helper.MCreateTransaction,
 		Transaction: t,
 		Proactive:   proactive,
 	}
@@ -172,7 +177,7 @@ func sendTransactionToServer(t types.Transaction, address string) {
 
 }
 
-func isTransactionKnown(t types.Transaction) bool {
+func isTransactionKnown(t helper.Transaction) bool {
 	for _, transaction := range transactions {
 		if transaction.Id == t.Id {
 			fmt.Printf("Transaction already exists\n")

local/server/test.json

+{
+    "transactions": [
+        {
+            "id": 1,
+            "from": "Peiry",
+            "to": "Troller",
+            "amount": 6000
+        },
+        {
+            "id": 2,
+            "from": "Birner",
+            "to": "Peiry",
+            "amount": 2000
+        }
+    ]
+}
\ No newline at end of file