com.qcloud.vod.VodUpload Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vod_api Show documentation
vod sdk for tencentyun
There is a newer version: 2.1.5
package com.qcloud.vod;

import com.qcloud.QcloudApiModuleCenter;
import com.qcloud.cos.COSClient;
import com.qcloud.cos.ClientConfig;
import com.qcloud.cos.auth.BasicCOSCredentials;
import com.qcloud.cos.auth.COSCredentials;
import com.qcloud.cos.model.PutObjectRequest;
import com.qcloud.cos.model.PutObjectResult;
import com.qcloud.cos.region.Region;
import com.qcloud.cos.transfer.TransferManager;
import com.qcloud.cos.transfer.Upload;
import com.qcloud.vod.common.VodConst;
import com.qcloud.vod.common.VodCosConf;
import com.qcloud.vod.common.VodParam;
import com.qcloud.vod.exception.VodParamException;
import com.qcloud.vod.response.VodUploadApplyResponse;
import com.qcloud.vod.response.VodUploadCommitResponse;
import com.qcloud.vod.util.FileUtil;
import com.qcloud.vod.util.JacksonUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.util.Map;
import java.util.TreeMap;

/**
 * 点播上传
 * @author jianguoxu
 * @time 2017/9/4 17:36
 */
public class VodUpload {

    private static final Logger logger = LoggerFactory.getLogger(VodUpload.class);

    /**
     * 提交上传
     * @param moduleCenter moduleCenter
     * @param vodParam param
     * @return
     * @throws Exception
     */
    public static VodUploadApplyResponse applyUpload(QcloudApiModuleCenter moduleCenter, VodParam vodParam, Map extraParams) throws Exception {
        TreeMap params = new TreeMap(extraParams);

        if (vodParam.getVideoPath() == null) {
            throw new VodParamException("videoPath is null");
        }

        params.put(VodConst.KEY_VIDEO_NAME, FileUtil.getFileName(vodParam.getVideoPath()));
        params.put(VodConst.KEY_VIDEO_SIZE, FileUtil.getFileSize(vodParam.getVideoPath()));
        params.put(VodConst.KEY_VIDEO_TYPE, FileUtil.getFileType(vodParam.getVideoPath()));

        if (vodParam.getCoverPath() != null) {
            params.put(VodConst.KEY_COVER_NAME, FileUtil.getFileName(vodParam.getCoverPath()));
            params.put(VodConst.KEY_COVER_SIZE, FileUtil.getFileSize(vodParam.getCoverPath()));
            params.put(VodConst.KEY_COVER_TYPE, FileUtil.getFileType(vodParam.getCoverPath()));
        }

        VodUploadApplyResponse response = null;
        try {
            String result = moduleCenter.call(VodConst.MODULE_APPLY_UPLOAD, params);
            response = JacksonUtil.readValue(result, VodUploadApplyResponse.class);
            return response;
        } catch (Exception e) {
            logger.error("apply upload error, param=" + vodParam, e);
            throw e;
        }
    }

    /**
     * 获取TransferManager
     * @param param
     * @param uploadApplyResponse
     * @return
     */
    public static TransferManager getTransferManager(VodParam param, VodUploadApplyResponse uploadApplyResponse, int signExpired) {
        String region = "cos." + uploadApplyResponse.getStorageRegionV5();
        ClientConfig clientConfig = new ClientConfig(new Region(region));
        clientConfig.setSignExpired(signExpired);

        COSCredentials credentials = new BasicCOSCredentials(String.valueOf(uploadApplyResponse.getStorageAppId()), param.getSecretId(), param.getSecretKey());
        COSClient cosClient = new COSClient(credentials, clientConfig);

        return new TransferManager(cosClient);
    }

    /**
     * 上传文件到Cos
     * @param transferManager
     * @param conf
     * @return
     */
    public static void uploadCos(TransferManager transferManager, VodCosConf conf) throws Exception {
        File file = new File(conf.getFilePath());

        try {
            Upload upload = transferManager.upload(conf.getBucketName(), conf.getStoragePath(), file);
            upload.waitForCompletion();
        } catch (Exception e) {
            logger.error("upload cos error, conf=" + conf, e);
            throw e;
        }
    }

    /**
     * 确认上传
     * @param moduleCenter
     * @param uploadApplyResponse
     * @return
     * @throws Exception
     */
    public static VodUploadCommitResponse commitUpload(QcloudApiModuleCenter moduleCenter, VodUploadApplyResponse uploadApplyResponse) throws Exception {
        TreeMap params = new TreeMap();

        params.put(VodConst.KEY_SESSION_KEY, uploadApplyResponse.getVodSessionKey());

        VodUploadCommitResponse response = null;
        try {
            String result = moduleCenter.call(VodConst.MODULE_COMMIT_UPLOAD, params);
            response = JacksonUtil.readValue(result, VodUploadCommitResponse.class);
            return response;
        } catch (Exception e) {
            logger.error("commit upload error, vodSessionKey=" + uploadApplyResponse.getVodSessionKey(), e);
            throw e;
        }
    }
}