openchain-cc/src/main/java/com/alipay/openchain/flow/OperateChainAccount.java

package com.alipay.openchain.flow;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.alipay.mychain.sdk.domain.account.Account;
import com.alipay.openchain.entity.User;
import com.antfinancial.mychain.baas.tool.restclient.RestClient;
import com.antfinancial.mychain.baas.tool.restclient.RestClientProperties;
import com.antfinancial.mychain.baas.tool.restclient.model.AccountRequest;
import com.antfinancial.mychain.baas.tool.restclient.model.CallRestBizParam;
import com.antfinancial.mychain.baas.tool.restclient.model.Method;
import com.antfinancial.mychain.baas.tool.restclient.response.BaseResp;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.UUID;

/**
 * block chain account operations
 */
@Service
public class OperateChainAccount {
    @Autowired
    private RestClient restClient;

    @Autowired
    private RestClientProperties restClientProperties;

//    public void runFlow () throws Exception {
//        createChainAccount();
//        queryChainAccount();
//    }
    /**
     * 创建KMS托管密钥的区块链账户
     */
    public User createChainAccount(String accountName, String newKmsId) throws Exception {
        User user = new User();
        CallRestBizParam param = CallRestBizParam.builder().
                // 创建账户方法名称，固定值
                method(Method.TENANTCREATEACCUNT).
                // 本次交易请求ID
                orderId(UUID.randomUUID().toString()).
                // 执行创建账户交易的已有区块链账户
                account(restClientProperties.getAccount()).
                // 执行创建账户交易的已有区块链账户KMS密钥ID
                mykmsKeyId(restClientProperties.getKmsId()).
                // 新建区块链账户   不可重复  包括失败的账户名也不可使用
                newAccountId(accountName).
                // 新建区块链账户KMS密钥ID 可重复
                newAccountKmsId(newKmsId).
                // 确保设置的Gas参数足够大，且执行创建的账户中有足够Gas，并且账户燃料要大于参数数值
                gas(100000L).
                build();
        BaseResp resp = restClient.chainCallForBiz(param);
        if(resp.isSuccess()) {
            System.out.println("创建账户成功，密钥:" + resp.toString());
            user.setCode( "200" );
            user.setUserMYStr( resp.getData() );
            user.setMsg( "创建账户成功，密钥:" + resp.getData() );
            //查询 账户信息id
            AccountRequest request = restClient.createQueryAccountForBiz( accountName );

            BaseResp resps = restClient.chainCall(
                    null,
                    JSON.toJSONString(request),
                    Method.QUERYACCOUNT);
            if(resps.isSuccess()) {
                JSONObject jsonObject = JSONObject.parseObject(resps.getData());
                Account account = new Account();
                account.fromJson(jsonObject);
                System.out.println("查询账户成功， 账户信息: ID " + account.getIdentity() + ", 余额 " + account.getBalance());
                user.setId(account.getIdentity().hexStrValue()  );
                user.setBalance( account.getBalance() );
            } else {
                System.err.println("查询账户失败: " + resps.getCode() + ", " + resps.getData());
                user.setId("查询账户失败: " + resps.getCode() + ", " + resps.getData());
            }
            return user;
            //-----end--
        } else {
            System.err.println("创建账户失败: " + resp.getCode() + ", " + resp.getData());
            user.setCode( resp.getCode() );
            user.setUserMYStr( resp.getData() );
            user.setMsg( "创建账户失败: " + resp.getCode() + ", " + resp.getData() );
            return user;
        }
    }

    /**
     * 查询指定名称的区块链账户
     */
    public User queryChainAccount(String accountName) throws Exception {
        AccountRequest request = restClient.createQueryAccountForBiz( accountName );
        User user = new User();
        BaseResp resp = restClient.chainCall(
                null,
                JSON.toJSONString(request),
                Method.QUERYACCOUNT);
        System.out.println("response: " + JSON.toJSONString(resp));
        if(resp.isSuccess()) {
            JSONObject jsonObject = JSONObject.parseObject(resp.getData());
            Account account = new Account();
            account.fromJson(jsonObject);
            user.setId( account.getIdentity().hexStrValue() );
            user.setBalance( account.getBalance() );
            user.setMsg(JSON.toJSONString(resp)  );
            user.setCode( "200" );
            System.out.println("查询账户成功， 账户信息: ID " + account.getIdentity() + ", 余额 " + account.getBalance());
        } else {
            user.setCode( resp.getCode() );
            user.setMsg( "查询账户失败: " + resp.getCode() + ", " + resp.getData() );
            System.err.println("查询账户失败: " + resp.getCode() + ", " + resp.getData());
        }
        return user;
    }
}