From 94827e20377f95e781cb88eb805f3c2f49f25d5e Mon Sep 17 00:00:00 2001
From: AstaFrode <frode@cess.one>
Date: Fri, 6 Sep 2024 14:33:05 +0800
Subject: [PATCH] Trade limit (#186)

* update sdk

* update sdk to v0.6.2
---
 go.mod            |  2 +-
 go.sum            |  4 ++--
 node/node.go      |  2 +-
 node/trackerv2.go | 22 ++++++++++------------
 4 files changed, 14 insertions(+), 16 deletions(-)

diff --git a/go.mod b/go.mod
index 454ad31..416ff00 100644
--- a/go.mod
+++ b/go.mod
@@ -5,7 +5,7 @@ go 1.21
 toolchain go1.22.4
 
 require (
-	github.com/CESSProject/cess-go-sdk v0.6.2-0.20240822075251-93843c2a4017
+	github.com/CESSProject/cess-go-sdk v0.6.2
 	github.com/CESSProject/cess-go-tools v0.2.12
 	github.com/CESSProject/go-keyring v0.0.0-20220614131247-ee3a8da30fde
 	github.com/CESSProject/p2p-go v0.4.1-0.20240809013702-e989d204843e
diff --git a/go.sum b/go.sum
index 9a641f1..855abc2 100644
--- a/go.sum
+++ b/go.sum
@@ -9,8 +9,8 @@ dmitri.shuralyov.com/state v0.0.0-20180228185332-28bcc343414c/go.mod h1:0PRwlb0D
 git.apache.org/thrift.git v0.0.0-20180902110319-2566ecd5d999/go.mod h1:fPE2ZNJGynbRyZ4dJvy6G277gSllfV2HJqblrnkyeyg=
 github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ=
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
-github.com/CESSProject/cess-go-sdk v0.6.2-0.20240822075251-93843c2a4017 h1:q488VDsCSykpJ1v+QmOxPVolkCBHvSxEP8dW3/+/MH0=
-github.com/CESSProject/cess-go-sdk v0.6.2-0.20240822075251-93843c2a4017/go.mod h1:L5IDeS2ydsdgtdybzhWk9fIdBsDkU2XrQyo5mrWHkdQ=
+github.com/CESSProject/cess-go-sdk v0.6.2 h1:B1MHE4xWErXVFMIwy9JGJKJxtjmtQhLm9Dnk/pYAh9k=
+github.com/CESSProject/cess-go-sdk v0.6.2/go.mod h1:Ue6EJicYGCW3CtEYCni/XT9mJFFc0YAvaBuQuciUdYk=
 github.com/CESSProject/cess-go-tools v0.2.12 h1:VqghaGaWgL+JelKXCXaABj4rCDFFRdbhiXifyMpkjqo=
 github.com/CESSProject/cess-go-tools v0.2.12/go.mod h1:ov1vSPbTlBSRWl3XqsibrKrK9smIq2hWeGs0TEhactc=
 github.com/CESSProject/go-keyring v0.0.0-20220614131247-ee3a8da30fde h1:5MDRjjtg6PEhqyVjupwaapN96cOZiddOGAYwKQeaTu0=
diff --git a/node/node.go b/node/node.go
index 798a890..184dda7 100644
--- a/node/node.go
+++ b/node/node.go
@@ -265,8 +265,8 @@ func (n *Node) ListTrackFiles() ([]string, error) {
 
 func (n *Node) DeleteTrackFile(filehash string) {
 	n.trackLock.Lock()
-	defer n.trackLock.Unlock()
 	os.Remove(filepath.Join(n.trackDir, filehash))
+	n.trackLock.Unlock()
 }
 
 func (n *Node) RebuildDirs() {
diff --git a/node/trackerv2.go b/node/trackerv2.go
index f177cd9..28d7856 100644
--- a/node/trackerv2.go
+++ b/node/trackerv2.go
@@ -150,6 +150,14 @@ func (n *Node) checkFileState(fid string) (StorageDataType, bool, error) {
 			return StorageDataType{}, false, fmt.Errorf("The fid after reprocessing is inconsistent [%s != %s] %v", recordFile.Fid, hash, err)
 		}
 		recordFile.Segment = segment
+		b, err := json.Marshal(&recordFile)
+		if err != nil {
+			return StorageDataType{}, false, errors.Wrapf(err, "[%s] [json.Marshal]", fid)
+		}
+		err = n.WriteTrackFile(fid, b)
+		if err != nil {
+			return StorageDataType{}, false, errors.Wrapf(err, "[%s] [WriteTrackFile]", fid)
+		}
 	}
 
 	var storageDataType = StorageDataType{
@@ -179,16 +187,6 @@ func (n *Node) checkFileState(fid string) (StorageDataType, bool, error) {
 		return storageDataType, false, nil
 	}
 
-	recordFile.PutFlag = false
-	b, err := json.Marshal(&recordFile)
-	if err != nil {
-		return StorageDataType{}, false, errors.Wrapf(err, "[%s] [json.Marshal]", fid)
-	}
-	err = n.WriteTrackFile(fid, b)
-	if err != nil {
-		return StorageDataType{}, false, errors.Wrapf(err, "[%s] [WriteTrackFile]", fid)
-	}
-
 	// verify the space is authorized
 	authAccs, err := n.QueryAuthorityList(recordFile.Owner, -1)
 	if err != nil {
@@ -222,9 +220,9 @@ func (n *Node) checkFileState(fid string) (StorageDataType, bool, error) {
 		recordFile.FileSize,
 	)
 	if err != nil {
-		return StorageDataType{}, false, err
+		return StorageDataType{}, false, fmt.Errorf(" %s [UploadDeclaration] %v", fid, err)
 	}
-	n.Logtrack("info", fmt.Sprintf("[%s] PlaceStorageOrder suc: %s", fid, txhash))
+	n.Logtrack("info", fmt.Sprintf(" %s [UploadDeclaration] suc: %s", fid, txhash))
 
 	for index := 0; index < (sconfig.DataShards + sconfig.ParShards); index++ {
 		var value = make([]string, 0)