edit api for get-work and submit work (#507)

[bytom/bytom.git] / api / miner.go

diff --git a/api/miner.go b/api/miner.go
index 9bafba4..30220b3 100644 (file)
--- a/api/miner.go
+++ b/api/miner.go
@@ -14,8 +14,15 @@ func (a *API) getWork() Response {
        return NewSuccessResponse(work)
 }
 
-func (a *API) submitWork(bh *types.BlockHeader) Response {
-       return NewSuccessResponse(a.bcr.SubmitWork(bh))
+type SubmitWorkReq struct {
+       BlockHeader *types.BlockHeader `json:"block_header"`
+}
+
+func (a *API) submitWork(ctx context.Context, req *SubmitWorkReq) Response {
+       if err := a.bcr.SubmitWork(req.BlockHeader); err != nil {
+               return NewErrorResponse(err)
+       }
+       return NewSuccessResponse(true)
 }
 
 func (a *API) getBlockHeaderByHeight(ctx context.Context, req struct {