Commit 58e85a8b authored by duanjinfei's avatar duanjinfei

change taskInfo struct

parent d8de0242
package controllers package controllers
import ( import (
"bytes"
"crypto/ecdsa" "crypto/ecdsa"
"encoding/json" "encoding/json"
"fmt" "fmt"
...@@ -31,14 +32,14 @@ func (c *LLMController) GetContainerSign() { ...@@ -31,14 +32,14 @@ func (c *LLMController) GetContainerSign() {
return return
} }
signPrv := beego.AppConfig.String("sign_prv") signPrv := beego.AppConfig.String("sign_prv")
if signData.TaskResult == "" || signData.TaskParam == "" || signData.TaskId == "" { if signData.TaskResult == nil || signData.TaskParam == nil || signData.TaskId == "" {
c.ResponseInfo(500, "sign container is failed", nil) c.ResponseInfo(500, "sign container is failed", nil)
return return
} }
bridgePrv := stringToPrivateKey(signPrv) bridgePrv := stringToPrivateKey(signPrv)
reqHash := crypto.Keccak256Hash([]byte(signData.TaskParam)) reqHash := crypto.Keccak256Hash(signData.TaskParam)
respHash := crypto.Keccak256Hash([]byte(signData.TaskResult)) respHash := crypto.Keccak256Hash(signData.TaskResult)
signHash := crypto.Keccak256Hash([]byte(signData.TaskId), reqHash.Bytes(), respHash.Bytes()) signHash := crypto.Keccak256Hash(bytes.NewBufferString(signData.TaskId).Bytes(), reqHash.Bytes(), respHash.Bytes())
fmt.Println("signHash :", signHash.String()) fmt.Println("signHash :", signHash.String())
signature, err := crypto.Sign(signHash.Bytes(), bridgePrv) signature, err := crypto.Sign(signHash.Bytes(), bridgePrv)
if err != nil { if err != nil {
......
...@@ -8,8 +8,8 @@ type SignInfoRep struct { ...@@ -8,8 +8,8 @@ type SignInfoRep struct {
type TaskInfo struct { type TaskInfo struct {
TaskId string `json:"task_id"` TaskId string `json:"task_id"`
TaskParam string `json:"task_param"` TaskParam []byte `json:"task_param"`
TaskResult string `json:"task_result"` TaskResult []byte `json:"task_result"`
} }
type ModelInfo struct { type ModelInfo struct {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment