JiuShengServer/src/dsb/com/cnd3b/restcontroller/enterprise/datacenter/datacenter.java

package com.cnd3b.restcontroller.enterprise.datacenter;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.aliyun.oss.model.*;
import com.cnd3b.common.Controller;
import com.cnd3b.common.data.Rows;
import com.cnd3b.common.data.SQLFactory;
import com.cnd3b.utility.aliyun.oss.AliyunOSSConfigConstant;
import com.cnd3b.utility.aliyun.oss.AliyunOSSUtil;
import p2.common.parse.A;
import p2.common.parse.S;
import p2.pao.PaoRemote;
import p2.pao.PaoSetRemote;
import p2.util.P2Exception;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.*;

import static com.cnd3b.utility.aliyun.oss.AliyunOSSUtil.*;


public class datacenter extends Controller {


    //    String url = BUCKE_NAME_1 + "." + AliyunOSSConfigConstant.END_POINT;
    //根目录
    String rootpath = "资料中心/";

    /**
     * 构造函数
     *
     * @param content
     */
    public datacenter(JSONObject content) {
        super(content);

    }


    /**
     * 创建文件夹(OSS)
     *
     * @return
     */
    public String createFolder() {
        String folderName = content.getString("folderName");
        if (folderName.equals("") || folderName == null) {
            return getErrReturnObject().setErrMsg("文件夹名称不能为空").toString();
        }
        ossClient.putObject(new PutObjectRequest(BUCKE_NAME_1, rootpath + folderName, new ByteArrayInputStream("".getBytes())));
        ossClient.shutdown();
        return getSucReturnObject().toString();
    }

    /**
     * 删除文件夹(OSS)
     *
     * @return
     */
    public String deleteFolder() {
        JSONArray jsonArray = content.getJSONArray("folderName");
        if (jsonArray == null || jsonArray.size() == 0) {
            return getErrReturnObject().setErrMsg("文件夹名称不能为空").toString();
        }
        ArrayList<String> keysList = new ArrayList<>();
        for (Object obj : jsonArray) {
            keysList.add(obj.toString());
            ListObjectsRequest listObjectsRequest = new ListObjectsRequest(BUCKE_NAME_1);
            // 设置prefix参数来获取fun目录下的所有文件。
            listObjectsRequest.setPrefix(obj.toString());
            // 递归列举目录下的所有文件。
            ObjectListing listing = ossClient.listObjects(listObjectsRequest);
            // 遍历所有文件。
            for (OSSObjectSummary objectSummary : listing.getObjectSummaries()) {
                String key = objectSummary.getKey();
                keysList.add(key);
            }
        }


        DeleteObjectsRequest request = new DeleteObjectsRequest(BUCKE_NAME_1);
        request.setKeys(keysList);
        // 发起deleteObjects请求。
        ossClient.deleteObjects(request);

//        // 查看删除结果。
//        for (String o : delObjResult.getDeletedObjects()) {
//            String keyName = null;
//            try {
//                keyName = URLDecoder.decode(o, "UTF-8");
//            } catch (UnsupportedEncodingException e) {
//                e.printStackTrace();
//            }
//            System.out.println("delete key name: " + keyName);
//        }

        ossClient.shutdown();

        SQLFactory sqlFactory = new SQLFactory(this, "删除文件夹");
        sqlFactory.addParameter_in("fpath", keysList);
        String sql = sqlFactory.getSQL();
        dbConnect.runSqlUpdate(sql);

        return getSucReturnObject().toString();
    }

    /**
     * 上传文件(OSS)
     *
     * @return
     */
    public String uploadFile() {
//        String filePath = content.getString("filePath");
        String folderPath = content.getString("folderPath");
        String name = content.getString("filename");
        String base64Str = content.getString("byteStr");
        byte[] byteStr = Base64.getDecoder().decode(base64Str);
//        if (filePath.equals("") || filePath == null) {
//            return getErrReturnObject().setErrMsg("文件路径不能为空").toString();
//        }
//        if (folderPath.equals("") || folderPath == null) {
//            folderPath = "";
//        }
//        File file = new File(filePath);
//        if (!file.exists()) {
//            return getErrReturnObject().setErrMsg("文件不存在").toString();
//        }

        String key = folderPath + name;
        System.err.println(key);

        ossClient.putObject(new PutObjectRequest(BUCKE_NAME_1, rootpath + key, new ByteArrayInputStream(byteStr)));
        ossClient.shutdown();
        return getSucReturnObject().toString();
    }

    /**
     * 删除文件(OSS)
     *
     * @return
     */
    public String deleteFile() {
        JSONArray jsonArray = content.getJSONArray("folderName");
        if (jsonArray == null || jsonArray.size() == 0) {
            return getErrReturnObject().setErrMsg("文件夹名称不能为空").toString();
        }
        List<String> keysList = new ArrayList<>();
        for (Object obj : jsonArray) {
            keysList.add(obj.toString());
        }
        DeleteObjectsRequest request = new DeleteObjectsRequest(BUCKE_NAME_1);
        request.setKeys(keysList);
        // 发起deleteObjects请求。
        DeleteObjectsResult delObjResult = ossClient.deleteObjects(request);

        // 查看删除结果。
        for (String o : delObjResult.getDeletedObjects()) {
            String keyName = null;
            try {
                keyName = URLDecoder.decode(o, "UTF-8");
            } catch (UnsupportedEncodingException e) {
                e.printStackTrace();
            }
            System.out.println("delete key name: " + keyName);
        }

        ossClient.shutdown();
        return getSucReturnObject().toString();
    }

    /**
     * 获取根目录
     *
     * @return
     */
    public String getFolders() {

        // 构造ListObjectsRequest请求。
        ListObjectsRequest listObjectsRequest = new ListObjectsRequest(BUCKE_NAME_1);
        // 设置正斜线（/）为文件夹的分隔符。
        listObjectsRequest.setDelimiter("/");
        // 列出fun目录下的所有文件和文件夹。
        listObjectsRequest.setPrefix("资料中心/");
        ObjectListing listing = ossClient.listObjects(listObjectsRequest);

        // 遍历所有commonPrefix。
        ArrayList<String> commonPrefixList = new ArrayList<>();
        for (String commonPrefix : listing.getCommonPrefixes()) {


            String selectSql = "SELECT*FROM tfolder WHERE fpath='" + commonPrefix + "'";
            Rows rows = dbConnect.runSqlQuery(selectSql);
            if (rows.isEmpty()) {
                inseretTfolder(commonPrefix);
            }
            commonPrefixList.add(commonPrefix);
        }
        // 关闭OSSClient。
        ossClient.shutdown();

        JSONObject object = new JSONObject();
        object.put("folder", commonPrefixList);

        SQLFactory sqlFactory = new SQLFactory(this, "查询根目录");
        sqlFactory.addParameter_in("fpath", commonPrefixList);
        String sql = sqlFactory.getSQL();

        Rows rows = dbConnect.runSqlQuery(sql);

        return getSucReturnObject().setData(rows).toString();
    }

    /**
     * 获取文件夹
     *
     * @return
     */
    public String getAllFolders() {
        // 构造ListObjectsRequest请求。
        ListObjectsRequest listObjectsRequest = new ListObjectsRequest(BUCKE_NAME_1);

        // 列举文件。
        ObjectListing listing = ossClient.listObjects(listObjectsRequest);

        //遍历所有文件。
        //System.out.println("Objects:");
        //List<String> stringList = new ArrayList<>();
        for (OSSObjectSummary objectSummary : listing.getObjectSummaries()) {
            System.out.println(objectSummary.getKey());
            if (objectSummary.getKey().endsWith("/")) {
                String key = objectSummary.getKey();
                //stringList.add(key);
                String selectSql = "SELECT*FROM tfolder WHERE fpath='" + key + "'";
                Rows rows = dbConnect.runSqlQuery(selectSql);
                if (rows.isEmpty()) {

                    inseretTfolder(objectSummary.getKey());
                }

            }

        }

        // 关闭OSSClient。
        ossClient.shutdown();

        String sql = "SELECT tfolderid,fpath ,tfilename FROM tfolder ORDER BY tfolderid ";
        Rows rows = dbConnect.runSqlQuery(sql);

        return getSucReturnObject().setData(rows).toString();
    }


    /**
     * 获取指定目录下的文件和文件夹
     *
     * @return
     */
    public String getFilesOfPath() {
        String folderPath = content.getString("folderPath");
        // 构造ListObjectsRequest请求。
        ListObjectsRequest listObjectsRequest = new ListObjectsRequest(BUCKE_NAME_1);
        // 设置正斜线（/）为文件夹的分隔符。
        listObjectsRequest.setDelimiter("/");
        // 列出fun目录下的所有文件和文件夹。
        listObjectsRequest.setPrefix(folderPath);
        ObjectListing listing = ossClient.listObjects(listObjectsRequest);
        // 遍历所有文件。
        System.out.println("Objects:");
        List<JSONObject> objectSummaryList = new ArrayList<>();
        List<OSSObjectSummary> sums = listing.getObjectSummaries();
        //按照时间排序
        Collections.sort(sums, (o1, o2) -> {
            // TODO Auto-generated method stub
            long dateTime1 = o1.getLastModified().getTime();
            long dateTime2 = o2.getLastModified().getTime();
            if (dateTime1 == dateTime2) {
                return 0;
            } else {
                //时间正序
//					return dateTime1 > dateTime2 ? 1 : -1;
                //时间倒序
                return dateTime1 > dateTime2 ? -1 : 1;
            }

        });
        for (OSSObjectSummary objectSummary : sums) {
            if (!objectSummary.getKey().equals(folderPath)) {
                JSONObject object = new JSONObject();
                object.put("key", objectSummary.getKey());
                object.put("name", objectSummary.getKey().replace(folderPath, ""));
                object.put("size", objectSummary.getSize() / 1024);
                object.put("time", getDateTime_Str(objectSummary.getLastModified()));
                object.put("url", file_url + "/" + objectSummary.getKey());
                objectSummaryList.add(object);
            }
        }

        // 遍历所有commonPrefix。
        System.out.println("CommonPrefixes:");
        ArrayList<String> commonPrefixList = new ArrayList<>();
        for (String commonPrefix : listing.getCommonPrefixes()) {
//            JSONObject object = new JSONObject();
//            object.put("folderpath", commonPrefix);
            String selectSql = "SELECT*FROM tfolder WHERE fpath='" + commonPrefix + "'";
            Rows rows = dbConnect.runSqlQuery(selectSql);
            if (rows.isEmpty()) {
                inseretTfolder(commonPrefix);

            }
            commonPrefixList.add(commonPrefix);
        }
        // 关闭OSSClient。
        ossClient.shutdown();

        SQLFactory sqlFactory = new SQLFactory(this, "查询根目录");
        sqlFactory.addParameter_in("fpath", commonPrefixList);
        String sql = sqlFactory.getSQL();

        Rows rows = dbConnect.runSqlQuery(sql);

        JSONObject object = new JSONObject();
        object.put("files", objectSummaryList);
        object.put("folder", rows);
        return getSucReturnObject().setData(object).toString();
    }

    /**
     * 更新文件夹名称
     *
     * @return
     */
    public String updateFolderName() {
        String oldName = content.get("oldName").toString();
        String newName = content.get("newName").toString();


        String sqlSelect = "SELECT*FROM tfolder WHERE tfilename='" + oldName + "'";
        Rows rowsSelect = dbConnect.runSqlQuery(sqlSelect);
        if (rowsSelect.isEmpty()) {
            return getErrReturnObject().setErrMsg("未找到修改项").toString();
        }

        String sqlCheck = "SELECT*FROM tfolder WHERE tfilename='" + newName + "'";
        Rows rowsCheck = dbConnect.runSqlQuery(sqlCheck);
        if (!rowsCheck.isEmpty()) {
            return getErrReturnObject().setErrMsg("文件名称不能重名").toString();
        }

        String sql = "update  tfolder  set  tfilename=replace(tfilename,'" + oldName + "','" + newName + "') where tfilename like '" + oldName + "%'";

        String res = dbConnect.runSqlUpdate(sql);

        return getSucReturnObject().setData(res).toString();

    }

    public void inseretTfolder(String name) {
        Long id = createTableID("tfolder", "tfolderid");
        String sql = "INSERT INTO tfolder(tfolderid,rmkenable,fpath,createdate,tfilename) VALUES(" + id + ", 0,'" + name + "','" + getDateTime_Str() + "','" + name + "')";
        dbConnect.runSqlUpdate(sql);
    }

    /**
     * 申请视频转码
     *
     * @return
     */
    public String applyTranscoding() throws P2Exception {
        //阿里云视频转码模板
        String templateId = "S00000001-200030";
        //大于这个码率进行视频转码，mov格式的直接转码
        int tempBitrate = 1800;
        //获取支持的视频格式列表
        ArrayList<String> listVideo = getSupportVieoList();

        String key = content.getString("key");
        if (key.toLowerCase().contains("mov")) {
            AliyunOSSUtil.transcoding(templateId, key, key);
            addTranscodingLog(key, key);
            return getSucReturnObject().setData("MOV转码成功").toString();
        }
        //查询是否是可支持转码的视频
        boolean isVideo = false;
        for (String str : listVideo) {
            if (key.toLowerCase().contains(str.toLowerCase())) {
                isVideo = true;
                break;
            }
        }
        if (isVideo) {
            String res = AliyunOSSUtil.submitMediaInfoJob(key);
            JSONObject jsonObject = JSONObject.parseObject(res);
            //码率
            String bitrate = "0";
            if (jsonObject.containsKey("mediaInfoJob")) {
                JSONObject mediaInfoJob_jsonObject = jsonObject.getJSONObject("mediaInfoJob");
                if (mediaInfoJob_jsonObject.containsKey("properties")) {
                    JSONObject properties_jsonObject = mediaInfoJob_jsonObject.getJSONObject("properties");
                    bitrate = properties_jsonObject.getString("bitrate");
                }
            }
            System.err.println(bitrate);
            if (Double.parseDouble(bitrate) > tempBitrate) {
                AliyunOSSUtil.transcoding(templateId, key, key);
                addTranscodingLog(key, key);
                return getSucReturnObject().setData("转码成功").toString();
            } else {
                return getSucReturnObject().setData("码率低于" + tempBitrate + "，无须转码").toString();
            }
        } else {
            return getSucReturnObject().setData("非视频格式，无法转码").toString();
        }

    }

    public String applyTranscodingOfPath() throws P2Exception {
        String folderPath = content.getString("folderPath");
        List<String> list = getOssFileList(folderPath);
        //阿里云视频转码模板
        String templateId = "S00000001-200030";
        //大于这个码率进行视频转码，mov格式的直接转码
        int tempBitrate = 1800;

        for (String keyPath : list) {

            String key = keyPath;
            if (key.toLowerCase().contains("mov")) {
                AliyunOSSUtil.transcoding(templateId, key, key);
                addTranscodingLog(key, key);
                System.err.println("MOV转码成功:" + keyPath);
            } else {
                ArrayList<String> listVideo = getSupportVieoList();
                //查询是否是可支持转码的视频
                boolean isVideo = false;
                for (String str : listVideo) {
                    if (key.toLowerCase().contains(str.toLowerCase())) {
                        isVideo = true;
                        break;
                    }
                }
                if (isVideo) {
                    String res = AliyunOSSUtil.submitMediaInfoJob(key);
                    JSONObject jsonObject = JSONObject.parseObject(res);
                    //码率
                    String bitrate = "0";
                    if (jsonObject.containsKey("mediaInfoJob")) {
                        JSONObject mediaInfoJob_jsonObject = jsonObject.getJSONObject("mediaInfoJob");
                        if (mediaInfoJob_jsonObject.containsKey("properties")) {
                            JSONObject properties_jsonObject = mediaInfoJob_jsonObject.getJSONObject("properties");
                            bitrate = properties_jsonObject.getString("bitrate");
                        }
                    }
                    System.err.println(bitrate);
                    if (Double.parseDouble(bitrate) > tempBitrate) {
                        AliyunOSSUtil.transcoding(templateId, key, key);
                        addTranscodingLog(key, key);
                        System.err.println("转码成功:" + keyPath);

                    } else {
                        System.err.println("码率低于" + tempBitrate + "，无须转码");

                    }
                } else {
                    System.err.println("非视频格式，无法转码");

                }
            }

        }
        return getSucReturnObject().toString();
    }

    //支持转转码的格式
    public ArrayList<String> getSupportVieoList() {
        //3GP、AVI、FLV、MP4、M3U8、MPG、ASF、WMV、MKV、MOV、TS、WebM、MXF
        ArrayList<String> list = new ArrayList<>();
        list.add("3GP");
        list.add("AVI");
        list.add("FLV");
        list.add("MP4");
        list.add("M3U8");
        list.add("MPG");
        list.add("ASF");
        list.add("WMV");
        list.add("MKV");
        list.add("MOV");
        list.add("TS");
        list.add("WebM");
        list.add("MXF");
        return list;
    }

    /**
     * 添加转码记录
     *
     * @return
     */
    public void addTranscodingLog(String tkey, String tkey_new) throws P2Exception {

        PaoSetRemote paoSetRemote = getP2ServerSystemPaoSet("TTRANSCODINGLOG", "tkey = '" + tkey + "'");
        if (paoSetRemote.isEmpty()) {
            PaoRemote paoRemote = paoSetRemote.addAtEnd();
            paoRemote.setValue("tkey", tkey, 11L);
            paoRemote.setValue("tkey_new", "code/" + tkey_new, 11L);

        } else {
            PaoRemote paoRemote = paoSetRemote.getPao(0);
            paoRemote.setValue("tkey_new", "转码视频/" + tkey_new, 11L);

        }

        paoSetRemote.save();

    }


}