129 lines
3.2 KiB
Go
129 lines
3.2 KiB
Go
// Transaction makes payment of X units from A to B
|
|
func (t *SimpleChaincode) invoke(stub shim.ChaincodeStubInterface, args []string) pb.Response {
|
|
var A, B string // Entities
|
|
var Aval, Bval int // Asset holdings
|
|
var X int // Transaction value
|
|
var err error
|
|
|
|
if len(args) != 3 {
|
|
return shim.Error("Incorrect number of arguments. Expecting 3")
|
|
}
|
|
|
|
A = args[0]
|
|
B = args[1]
|
|
|
|
// Get the state from the ledger
|
|
// TODO: will be nice to have a GetAllState call to ledger
|
|
Avalbytes, err := stub.GetState(A)
|
|
if err != nil {
|
|
return shim.Error("Failed to get state")
|
|
}
|
|
if Avalbytes == nil {
|
|
return shim.Error("Entity not found")
|
|
}
|
|
Aval, _ = strconv.Atoi(string(Avalbytes))
|
|
|
|
Bvalbytes, err := stub.GetState(B)
|
|
if err != nil {
|
|
return shim.Error("Failed to get state")
|
|
}
|
|
if Bvalbytes == nil {
|
|
return shim.Error("Entity not found")
|
|
}
|
|
Bval, _ = strconv.Atoi(string(Bvalbytes))
|
|
|
|
// Perform the execution
|
|
X, err = strconv.Atoi(args[2])
|
|
if err != nil {
|
|
return shim.Error("Invalid transaction amount, expecting a integer value")
|
|
}
|
|
Aval = Aval - X
|
|
Bval = Bval + X
|
|
fmt.Printf("Aval = %d, Bval = %d\n", Aval, Bval)
|
|
|
|
// Write the state back to the ledger
|
|
err = stub.PutState(A, []byte(strconv.Itoa(Aval)))
|
|
if err != nil {
|
|
return shim.Error(err.Error())
|
|
}
|
|
|
|
err = stub.PutState(B, []byte(strconv.Itoa(Bval)))
|
|
if err != nil {
|
|
return shim.Error(err.Error())
|
|
}
|
|
|
|
return shim.Success(nil)
|
|
}
|
|
|
|
|
|
// Deletes an entity from state
|
|
func (t *SimpleChaincode) delete(stub shim.ChaincodeStubInterface, args []string) pb.Response {
|
|
if len(args) != 1 {
|
|
return shim.Error("Incorrect number of arguments. Expecting 1")
|
|
}
|
|
|
|
A := args[0]
|
|
|
|
// Delete the key from the state in ledger
|
|
err := stub.DelState(A)
|
|
if err != nil {
|
|
return shim.Error("Failed to delete state")
|
|
}
|
|
|
|
return shim.Success(nil)
|
|
}
|
|
|
|
|
|
// query callback representing the query of a chaincode
|
|
func (t *SimpleChaincode) query(stub shim.ChaincodeStubInterface, args []string) pb.Response {
|
|
var A string // Entities
|
|
var err error
|
|
|
|
if len(args) != 1 {
|
|
return shim.Error("Incorrect number of arguments. Expecting name of the person to query")
|
|
}
|
|
|
|
A = args[0]
|
|
|
|
// Get the state from the ledger
|
|
Avalbytes, err := stub.GetState(A)
|
|
if err != nil {
|
|
jsonResp := "{\"Error\":\"Failed to get state for " + A + "\"}"
|
|
return shim.Error(jsonResp)
|
|
}
|
|
|
|
if Avalbytes == nil {
|
|
jsonResp := "{\"Error\":\"Nil amount for " + A + "\"}"
|
|
return shim.Error(jsonResp)
|
|
}
|
|
|
|
jsonResp := "{\"Name\":\"" + A + "\",\"Amount\":\"" + string(Avalbytes) + "\"}"
|
|
fmt.Printf("Query Response:%s\n", jsonResp)
|
|
return shim.Success(Avalbytes)
|
|
}
|
|
|
|
|
|
func (t *SimpleChaincode) register(stub shim.ChaincodeStubInterface, args []string) pb.Response {
|
|
var A string // Entities
|
|
var Aval int // Asset holdings
|
|
var err error
|
|
|
|
if len(args) != 2 {
|
|
return shim.Error("Incorrect number of arguments. Expecting 2")
|
|
}
|
|
|
|
// Initialize the chaincode
|
|
A = args[0]
|
|
Aval, err = strconv.Atoi(args[1])
|
|
if err != nil {
|
|
return shim.Error("Expecting integer value for asset holding")
|
|
}
|
|
fmt.Printf("Aval = %d\n", Aval)
|
|
|
|
// Write the state to the ledger
|
|
err = stub.PutState(A, []byte(strconv.Itoa(Aval)))
|
|
if err != nil {
|
|
return shim.Error(err.Error())
|
|
}
|
|
return shim.Success(nil)
|
|
} |