下载视频

yzg-util-file/src/main/java/com/yanzuoguang/cloud/file/dao/impl/YzgFileDaoImpl.java

@@ -5,6 +5,7 @@ import com.yanzuoguang.cloud.vo.YzgFileVo;
 import com.yanzuoguang.cloud.vo.req.YzgFileStatusReqVo;
 import com.yanzuoguang.dao.DaoConst;
 import com.yanzuoguang.dao.impl.BaseDaoImpl;
+import com.yanzuoguang.util.helper.StringHelper;
 import com.yanzuoguang.util.vo.MapRow;
 import org.springframework.beans.factory.InitializingBean;
 import org.springframework.stereotype.Component;
@@ -18,6 +19,8 @@ import java.util.List;
  */
 @Component
 public class YzgFileDaoImpl extends BaseDaoImpl implements YzgFileDao, InitializingBean {
+    private static final String GET_COUNT_SQL = "GET_COUNT_SQL";
+    private static final String UPDATE_STATUS_SQL = "UPDATE_STATUS_SQL";
     private static final String QUERY_TABLE_SQL = "SHOW TABLES LIKE 'Yzg_File'";
     private static final String CREATE_TABLE_SQL = "CREATE TABLE Yzg_File ( " +
             "   fileId               VARCHAR(32) NOT NULL COMMENT '编号', " +
@@ -47,6 +50,15 @@ public class YzgFileDaoImpl extends BaseDaoImpl implements YzgFileDao, Initializ
         getSql(DaoConst.LOAD)
                 .addCode("{ORDER}", "ORDER BY createDate DESC")
         ;
+
+        table.add(GET_COUNT_SQL, "SELECT COUNT(1) FROM Yzg_File WHERE 1=1")
+                .add(YzgFileVo.class, "a.");
+
+        table.add(UPDATE_STATUS_SQL, "UPDATE Yzg_File AS a SET a.fileId=a.fileId {FIELD} WHERE a.fileId=@fileId")
+                .addPara("fileStatus", ",a.fileStatus=CASE WHEN @fileStatus>a.fileStatus THEN @fileStatus ELSE a.fileStatus END")
+                .addPara("fileRow", ",a.fileRow=CASE WHEN @fileRow>a.fileRow THEN @fileRow ELSE a.fileRow END")
+                .addPara("errorMessage", ",a.errorMessage=@errorMessage")
+        ;
     }
 
     /**
@@ -77,8 +89,14 @@ public class YzgFileDaoImpl extends BaseDaoImpl implements YzgFileDao, Initializ
      */
     @Override
     public int getUserCompanyCount(YzgFileVo req) {
+        if (StringHelper.isEmpty(req.getCompanyId(), req.getCreateUserId())) {
             return 0;
         }
+        YzgFileVo countReq = new YzgFileVo();
+        countReq.setCompanyId(req.getCompanyId());
+        countReq.setCreateUserId(req.getCreateUserId());
+        return StringHelper.toInt(queryCell(GET_COUNT_SQL, countReq));
+    }
 
     /**
      * 获取导出文件数量
@@ -88,8 +106,13 @@ public class YzgFileDaoImpl extends BaseDaoImpl implements YzgFileDao, Initializ
      */
     @Override
     public int getCallbackCount(YzgFileVo req) {
+        if (StringHelper.isEmpty(req.getCompanyId(), req.getCreateUserId())) {
             return 0;
         }
+        YzgFileVo countReq = new YzgFileVo();
+        countReq.setCallbackMQ(req.getCallbackMQ());
+        return StringHelper.toInt(queryCell(GET_COUNT_SQL, countReq));
+    }
 
     /**
      * 修改文件状态
@@ -99,6 +122,6 @@ public class YzgFileDaoImpl extends BaseDaoImpl implements YzgFileDao, Initializ
      */
     @Override
     public int updateStatus(YzgFileStatusReqVo req) {
-        return 0;
+        return this.updateSql(UPDATE_STATUS_SQL, req);
     }
 }