com.tencentcloudapi.common.provider.CvmRoleCredential Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of tencentcloud-sdk-java-common Show documentation
Tencent Cloud Open API SDK for Java
There is a newer version: 3.1.1108
Show newest version
package com.tencentcloudapi.common.provider;

import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

public class CvmRoleCredential extends Credential {
    private static final String ENDPOINT = "http://metadata.tencentyun.com/latest/meta-data/cam/security-credentials/";
    private static final int EXPIRED_TIME = 300;
    private String roleName;
    private String secretId;
    private String secretKey;
    private String token;
    private int expiredTime;

    public CvmRoleCredential() {
        super();
    }

    public CvmRoleCredential(String roleName) {
        super();
        this.roleName = roleName;
    }

    private void updateCredential() throws TencentCloudSDKException {
        if (roleName == null) {
            roleName = loadJson(ENDPOINT);
        }
        String resp = loadJson(ENDPOINT + roleName);
        Map maps = new Gson().fromJson(resp, new TypeToken>() {
        }.getType());
        if (!maps.get("Code").equals("Success")) {
            throw new TencentCloudSDKException("CVM role token data failed");
        }
        secretId = (String) maps.get("TmpSecretId");
        secretKey = (String) maps.get("TmpSecretKey");
        token = (String) maps.get("Token");
        expiredTime = ((Double) maps.get("ExpiredTime")).intValue();
    }

    public String getSecretId() {
        if (secretId == null || needRefresh()) {
            try {
                updateCredential();
            } catch (TencentCloudSDKException e) {
                return null;
            }
        }
        return secretId;
    }

    public String getSecretKey() {
        if (secretKey == null || needRefresh()) {
            try {
                updateCredential();
            } catch (TencentCloudSDKException e) {
                return null;
            }
        }
        return secretKey;
    }

    public String getToken() {
        if (token == null || needRefresh()) {
            try {
                updateCredential();
            } catch (TencentCloudSDKException e) {
                return null;
            }
        }
        return token;
    }

    private boolean needRefresh() {
        if (expiredTime - new Date().getTime() / 1000 <= EXPIRED_TIME) {
            return true;
        } else {
            return false;
        }
    }

    private String loadJson(String url) throws TencentCloudSDKException {
        StringBuilder json = new StringBuilder();
        try {
            URL urlObject = new URL(url);
            HttpURLConnection uc = (HttpURLConnection) urlObject.openConnection();
            BufferedReader in = null;
            in = new BufferedReader(new InputStreamReader(uc.getInputStream(), "utf-8"));
            String inputLine = null;
            while ((inputLine = in.readLine()) != null) {
                json.append(inputLine);
            }
            in.close();
            uc.disconnect();
        } catch (Exception e) {
            throw new TencentCloudSDKException(e.getMessage());
        }
        return json.toString();
    }
}