From 98dad22233297dd1328247cb774c3cffbf24fdbc Mon Sep 17 00:00:00 2001
From: Lz <imlangzi@qq.com>
Date: Mon, 25 Apr 2022 18:02:32 +0800
Subject: [PATCH] feat(alloc): added name field (#221)

---
 cmd/getallocation.go    |  1 +
 cmd/listallocations.go  |  4 ++--
 cmd/newallocation.go    | 15 +++++++++++++--
 cmd/updateallocation.go | 11 +++++++++++
 go.mod                  |  2 +-
 go.sum                  |  4 ++--
 6 files changed, 30 insertions(+), 7 deletions(-)

diff --git a/cmd/getallocation.go b/cmd/getallocation.go
index faafef42..027f00cc 100644
--- a/cmd/getallocation.go
+++ b/cmd/getallocation.go
@@ -82,6 +82,7 @@ var getallocationCmd = &cobra.Command{
 		fmt.Println("allocation:")
 		fmt.Println("  id:             ", alloc.ID)
 		fmt.Println("  tx:             ", alloc.Tx, "(latest create/update allocation transaction hash)")
+		fmt.Println("  name:           ", alloc.Name)
 		fmt.Println("  data_shards:    ", alloc.DataShards)
 		fmt.Println("  parity_shards:  ", alloc.ParityShards)
 		fmt.Println("  size:           ", common.Size(alloc.Size))
diff --git a/cmd/listallocations.go b/cmd/listallocations.go
index 7fd3c26f..8f2bfbd2 100644
--- a/cmd/listallocations.go
+++ b/cmd/listallocations.go
@@ -28,7 +28,7 @@ var listallocationsCmd = &cobra.Command{
 			util.PrintJSON(allocations)
 			return
 		}
-		header := []string{"ID", "Size", "Expiration", "Datashards",
+		header := []string{"ID", "Name", "Size", "Expiration", "Datashards",
 			"Parityshards", "Finalized", "Canceled", "R. Price", "W. Price"}
 		data := make([][]string, len(allocations))
 		for idx, allocation := range allocations {
@@ -51,7 +51,7 @@ var listallocationsCmd = &cobra.Command{
 			}
 
 			data[idx] = []string{
-				allocation.ID, size, expStr, d, p,
+				allocation.ID, allocation.Name, size, expStr, d, p,
 				strconv.FormatBool(allocation.Finalized),
 				strconv.FormatBool(allocation.Canceled),
 				rp.String(), wp.String(),
diff --git a/cmd/newallocation.go b/cmd/newallocation.go
index 91db5fb5..671a8beb 100644
--- a/cmd/newallocation.go
+++ b/cmd/newallocation.go
@@ -170,9 +170,18 @@ var newallocationCmd = &cobra.Command{
 				log.Fatal("invalid owner value: ", err)
 			}
 		}
+
+		var allocationName string
+		if flags.Changed("name") {
+			allocationName, err = flags.GetString("name")
+			if err != nil {
+				log.Fatal("invalid allocation name: ", err)
+			}
+		}
+
 		var allocationID string
 		if len(owner) == 0 {
-			allocationID, err = sdk.CreateAllocation(*datashards, *parityshards,
+			allocationID, err = sdk.CreateAllocation(allocationName, *datashards, *parityshards,
 				*size, expireAt, readPrice, writePrice, mcct, lock)
 			if err != nil {
 				log.Fatal("Error creating allocation: ", err)
@@ -188,7 +197,7 @@ var newallocationCmd = &cobra.Command{
 				}
 			}
 
-			allocationID, err = sdk.CreateAllocationForOwner(owner, ownerPublicKey, *datashards, *parityshards,
+			allocationID, err = sdk.CreateAllocationForOwner(allocationName, owner, ownerPublicKey, *datashards, *parityshards,
 				*size, expireAt, readPrice, writePrice, mcct, lock, blockchain.GetPreferredBlobbers())
 			if err != nil {
 				log.Fatal("Error creating allocation: ", err)
@@ -263,6 +272,8 @@ func init() {
 	newallocationCmd.Flags().String("owner_public_key", "",
 		"public key of owner, user when creating an allocation for somone else")
 
+	newallocationCmd.Flags().String("name", "", "allocation name")
+
 }
 
 func storeAllocation(allocationID string) {
diff --git a/cmd/updateallocation.go b/cmd/updateallocation.go
index d3ecdebf..73f3cb68 100644
--- a/cmd/updateallocation.go
+++ b/cmd/updateallocation.go
@@ -89,7 +89,16 @@ var updateAllocationCmd = &cobra.Command{
 
 		setImmutable, _ := cmd.Flags().GetBool("set_immutable")
 
+		var allocationName string
+		if flags.Changed("name") {
+			allocationName, err = flags.GetString("name")
+			if err != nil {
+				log.Fatal("invalid allocation name: ", err)
+			}
+		}
+
 		txnHash, err := sdk.UpdateAllocation(
+			allocationName,
 			size,
 			int64(expiry/time.Second),
 			allocID,
@@ -128,4 +137,6 @@ func init() {
 
 	updateAllocationCmd.MarkFlagRequired("allocation")
 
+	updateAllocationCmd.Flags().String("name", "", "allocation name")
+
 }
diff --git a/go.mod b/go.mod
index fb25b084..9f8b5ee8 100644
--- a/go.mod
+++ b/go.mod
@@ -4,7 +4,7 @@ go 1.18
 
 require (
 	github.com/0chain/errors v1.0.3
-	github.com/0chain/gosdk v1.8.2-0.20220423151921-75736d82dc53
+	github.com/0chain/gosdk v1.8.2-0.20220425031623-55bd6fee2f46
 	github.com/icza/bitio v1.1.0
 	github.com/olekukonko/tablewriter v0.0.5
 	github.com/spf13/cobra v1.4.0
diff --git a/go.sum b/go.sum
index df7340ee..a4621054 100644
--- a/go.sum
+++ b/go.sum
@@ -42,8 +42,8 @@ collectd.org v0.3.0/go.mod h1:A/8DzQBkF6abtvrT2j/AU/4tiBgJWYyh0y/oB/4MlWE=
 dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU=
 github.com/0chain/errors v1.0.3 h1:QQZPFxTfnMcRdt32DXbzRQIfGWmBsKoEdszKQDb0rRM=
 github.com/0chain/errors v1.0.3/go.mod h1:xymD6nVgrbgttWwkpSCfLLEJbFO6iHGQwk/yeSuYkIc=
-github.com/0chain/gosdk v1.8.2-0.20220423151921-75736d82dc53 h1:xhldNdThQEU1+JPOg65avldOvPEQYeGByNeTvGMFM6I=
-github.com/0chain/gosdk v1.8.2-0.20220423151921-75736d82dc53/go.mod h1:Ug0PnuaP1E7FmNrRrZ4/UaC6AveGqU/fDleVFRyFGYg=
+github.com/0chain/gosdk v1.8.2-0.20220425031623-55bd6fee2f46 h1:sG8clBv7xgDyLFpzUPy4sjWvRqthxqp/ghNYp801NiA=
+github.com/0chain/gosdk v1.8.2-0.20220425031623-55bd6fee2f46/go.mod h1:Ug0PnuaP1E7FmNrRrZ4/UaC6AveGqU/fDleVFRyFGYg=
 github.com/Azure/azure-sdk-for-go/sdk/azcore v0.21.1/go.mod h1:fBF9PQNqB8scdgpZ3ufzaLntG0AG7C1WjPMsiFOmfHM=
 github.com/Azure/azure-sdk-for-go/sdk/internal v0.8.3/go.mod h1:KLF4gFr6DcKFZwSuH8w8yEK6DpFl3LP5rhdvAb7Yz5I=
 github.com/Azure/azure-sdk-for-go/sdk/storage/azblob v0.3.0/go.mod h1:tPaiy8S5bQ+S5sOiDlINkp7+Ef339+Nz5L5XO+cnOHo=