From 0278ec1b03310cd46313db088a133aa0dad99151 Mon Sep 17 00:00:00 2001
From: simlecode <69969590+simlecode@users.noreply.github.com>
Date: Thu, 11 Jul 2024 15:04:24 +0800
Subject: [PATCH] feat: market api: update deal payload size

---
 venus-shared/api/market/v1/api.go              |  2 ++
 venus-shared/api/market/v1/method.md           | 18 ++++++++++++++++++
 .../api/market/v1/mock/mock_imarket.go         | 14 ++++++++++++++
 venus-shared/api/market/v1/proxy_gen.go        |  4 ++++
 4 files changed, 38 insertions(+)

diff --git a/venus-shared/api/market/v1/api.go b/venus-shared/api/market/v1/api.go
index 9033f2041d..c80aa8b6af 100644
--- a/venus-shared/api/market/v1/api.go
+++ b/venus-shared/api/market/v1/api.go
@@ -198,5 +198,7 @@ type IMarket interface {
 	ListDirectDeals(ctx context.Context, queryParams market.DirectDealQueryParams) ([]*market.DirectDeal, error) //perm:read
 	UpdateDirectDealState(ctx context.Context, id uuid.UUID, state market.DirectDealState) error                 //perm:write
 
+	UpdateStorageDealPayloadSize(ctx context.Context, dealProposal cid.Cid, payloadSize uint64) error //perm:write
+
 	api.Version
 }
diff --git a/venus-shared/api/market/v1/method.md b/venus-shared/api/market/v1/method.md
index e33af96d03..3740b26863 100644
--- a/venus-shared/api/market/v1/method.md
+++ b/venus-shared/api/market/v1/method.md
@@ -109,6 +109,7 @@ curl http://<ip>:<port>/rpc/v1 -X POST -H "Content-Type: application/json"  -H "
   * [UpdateDealOnPacking](#updatedealonpacking)
   * [UpdateDealStatus](#updatedealstatus)
   * [UpdateDirectDealState](#updatedirectdealstate)
+  * [UpdateStorageDealPayloadSize](#updatestoragedealpayloadsize)
   * [UpdateStorageDealStatus](#updatestoragedealstatus)
   * [Version](#version)
 
@@ -2772,6 +2773,23 @@ Inputs:
 
 Response: `{}`
 
+### UpdateStorageDealPayloadSize
+
+
+Perms: write
+
+Inputs:
+```json
+[
+  {
+    "/": "bafy2bzacea3wsdh6y3a36tb3skempjoxqpuyompjbmfeyf34fi3uy6uue42v4"
+  },
+  42
+]
+```
+
+Response: `{}`
+
 ### UpdateStorageDealStatus
 
 
diff --git a/venus-shared/api/market/v1/mock/mock_imarket.go b/venus-shared/api/market/v1/mock/mock_imarket.go
index 32339ca1d0..f3645ade1d 100644
--- a/venus-shared/api/market/v1/mock/mock_imarket.go
+++ b/venus-shared/api/market/v1/mock/mock_imarket.go
@@ -1539,6 +1539,20 @@ func (mr *MockIMarketMockRecorder) UpdateDirectDealState(arg0, arg1, arg2 interf
 	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDirectDealState", reflect.TypeOf((*MockIMarket)(nil).UpdateDirectDealState), arg0, arg1, arg2)
 }
 
+// UpdateStorageDealPayloadSize mocks base method.
+func (m *MockIMarket) UpdateStorageDealPayloadSize(arg0 context.Context, arg1 cid.Cid, arg2 uint64) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "UpdateStorageDealPayloadSize", arg0, arg1, arg2)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// UpdateStorageDealPayloadSize indicates an expected call of UpdateStorageDealPayloadSize.
+func (mr *MockIMarketMockRecorder) UpdateStorageDealPayloadSize(arg0, arg1, arg2 interface{}) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateStorageDealPayloadSize", reflect.TypeOf((*MockIMarket)(nil).UpdateStorageDealPayloadSize), arg0, arg1, arg2)
+}
+
 // UpdateStorageDealStatus mocks base method.
 func (m *MockIMarket) UpdateStorageDealStatus(arg0 context.Context, arg1 cid.Cid, arg2 uint64, arg3 market.PieceStatus) error {
 	m.ctrl.T.Helper()
diff --git a/venus-shared/api/market/v1/proxy_gen.go b/venus-shared/api/market/v1/proxy_gen.go
index ed6e264ac9..5b9d76b733 100644
--- a/venus-shared/api/market/v1/proxy_gen.go
+++ b/venus-shared/api/market/v1/proxy_gen.go
@@ -125,6 +125,7 @@ type IMarketStruct struct {
 		UpdateDealOnPacking                     func(ctx context.Context, miner address.Address, dealID abi.DealID, sectorid abi.SectorNumber, offset abi.PaddedPieceSize) error                                                                    `perm:"write"`
 		UpdateDealStatus                        func(ctx context.Context, miner address.Address, dealID abi.DealID, pieceStatus market.PieceStatus, dealStatus storagemarket.StorageDealStatus) error                                               `perm:"write"`
 		UpdateDirectDealState                   func(ctx context.Context, id uuid.UUID, state market.DirectDealState) error                                                                                                                         `perm:"write"`
+		UpdateStorageDealPayloadSize            func(ctx context.Context, dealProposal cid.Cid, payloadSize uint64) error                                                                                                                           `perm:"write"`
 		UpdateStorageDealStatus                 func(ctx context.Context, dealProposalCid cid.Cid, state storagemarket.StorageDealStatus, pieceState market.PieceStatus) error                                                                      `perm:"write"`
 		Version                                 func(ctx context.Context) (types.Version, error)                                                                                                                                                    `perm:"read"`
 	}
@@ -434,6 +435,9 @@ func (s *IMarketStruct) UpdateDealStatus(p0 context.Context, p1 address.Address,
 func (s *IMarketStruct) UpdateDirectDealState(p0 context.Context, p1 uuid.UUID, p2 market.DirectDealState) error {
 	return s.Internal.UpdateDirectDealState(p0, p1, p2)
 }
+func (s *IMarketStruct) UpdateStorageDealPayloadSize(p0 context.Context, p1 cid.Cid, p2 uint64) error {
+	return s.Internal.UpdateStorageDealPayloadSize(p0, p1, p2)
+}
 func (s *IMarketStruct) UpdateStorageDealStatus(p0 context.Context, p1 cid.Cid, p2 storagemarket.StorageDealStatus, p3 market.PieceStatus) error {
 	return s.Internal.UpdateStorageDealStatus(p0, p1, p2, p3)
 }