func (t *basicInfo) Init(stub shim.ChaincodeStubInterface) pb.Response { return shim.Success(nil) } func (t *basicInfo) Invoke(stub shim.ChaincodeStubInterface) pb.Response { funcName, args := stub.GetFunctionAndParameters() if funcName == "save" { return t.saveBasic(stub, args) } else if funcName == "query" { return t.queryBasic(stub, args) } else if funcName == "delete" { return t.deleteBasic(stub, args) } else if funcName == "queryAll" { return t.queryAll(stub, args) } else { return shim.Error("no such function") } } func (t *basicInfo) saveBasic(stub shim.ChaincodeStubInterface, args []string) pb.Response { if len(args) != 2 { return shim.Error("except two args") } else { err := stub.PutState(args[0], []byte(args[1])) if err != nil { return shim.Error(err.Error()) } return shim.Success(nil) } }
时间: 2024-03-12 10:43:44 浏览: 92
这段代码已经比较简洁,但可以进一步优化:
1. 在 `Init` 函数中,可以使用 `shim.Success([]byte{})` 替代 `shim.Success(nil)`,以避免在调用端出现错误。
2. 在 `Invoke` 函数中,可以使用 `switch` 语句替代 `if else` 语句,使代码更加简洁易读。
3. 在 `saveBasic` 函数中,可以使用 `shim.Success([]byte{})` 替代 `shim.Success(nil)`,以避免在调用端出现错误。
下面是优化后的代码:
```
func (t *basicInfo) Init(stub shim.ChaincodeStubInterface) pb.Response {
return shim.Success([]byte{})
}
func (t *basicInfo) Invoke(stub shim.ChaincodeStubInterface) pb.Response {
funcName, args := stub.GetFunctionAndParameters()
switch funcName {
case "save":
return t.saveBasic(stub, args)
case "query":
return t.queryBasic(stub, args)
case "delete":
return t.deleteBasic(stub, args)
case "queryAll":
return t.queryAll(stub, args)
default:
return shim.Error("no such function")
}
}
func (t *basicInfo) saveBasic(stub shim.ChaincodeStubInterface, args []string) pb.Response {
if len(args) != 2 {
return shim.Error("except two args")
}
err := stub.PutState(args[0], []byte(args[1]))
if err != nil {
return shim.Error(err.Error())
}
return shim.Success([]byte{})
}
```
阅读全文