AdminVideoApiController.java 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354
  1. package com.goafanti.admin.controller;
  2. import java.io.File;
  3. import java.io.FileInputStream;
  4. import java.io.FileNotFoundException;
  5. import java.io.IOException;
  6. import java.math.BigInteger;
  7. import java.security.MessageDigest;
  8. import java.security.NoSuchAlgorithmException;
  9. import java.util.Base64;
  10. import java.util.Calendar;
  11. import java.util.concurrent.Callable;
  12. import java.util.concurrent.ExecutionException;
  13. import java.util.concurrent.ExecutorService;
  14. import java.util.concurrent.Executors;
  15. import java.util.concurrent.FutureTask;
  16. import java.util.concurrent.TimeUnit;
  17. import javax.annotation.Resource;
  18. import javax.servlet.http.HttpServletRequest;
  19. import javax.validation.Valid;
  20. import net.bramp.ffmpeg.FFmpeg;
  21. import net.bramp.ffmpeg.FFmpegExecutor;
  22. import net.bramp.ffmpeg.FFmpegUtils;
  23. import net.bramp.ffmpeg.FFprobe;
  24. import net.bramp.ffmpeg.builder.FFmpegBuilder;
  25. import net.bramp.ffmpeg.job.FFmpegJob;
  26. import net.bramp.ffmpeg.probe.FFmpegProbeResult;
  27. import net.bramp.ffmpeg.progress.Progress;
  28. import net.bramp.ffmpeg.progress.ProgressListener;
  29. import org.springframework.beans.factory.annotation.Value;
  30. import org.springframework.stereotype.Controller;
  31. import org.springframework.validation.BindingResult;
  32. import org.springframework.web.bind.annotation.RequestMapping;
  33. import org.springframework.web.bind.annotation.RequestMethod;
  34. import org.springframework.web.bind.annotation.ResponseBody;
  35. import com.goafanti.admin.service.AdminVideoService;
  36. import com.goafanti.common.bo.Result;
  37. import com.goafanti.common.constant.ErrorConstants;
  38. import com.goafanti.common.controller.CertifyApiController;
  39. import com.goafanti.common.enums.AttachmentType;
  40. import com.goafanti.common.enums.VideoFields;
  41. import com.goafanti.common.model.JtVideo;
  42. import com.goafanti.common.utils.FtpUtils;
  43. import com.goafanti.common.utils.StringUtils;
  44. import com.goafanti.core.shiro.token.TokenManager;
  45. @Controller
  46. @RequestMapping(value = "/api/admin/video")
  47. public class AdminVideoApiController extends CertifyApiController {
  48. @Resource
  49. private AdminVideoService adminVideoService;
  50. @Value(value = "${ffmpeg.path}")
  51. public String ffmpegPath = null;
  52. @Value(value = "${ffprobe.path}")
  53. public String ffprobePath = null;
  54. // 上传视频只做MP4格式的
  55. @RequestMapping(value = "/upload", method = RequestMethod.POST)
  56. @ResponseBody
  57. public Result uploadPicture(HttpServletRequest req, String sign) {
  58. Result res = new Result();
  59. AttachmentType attachmentType = AttachmentType.getField(sign);
  60. if (attachmentType == AttachmentType.VIDEO) {
  61. // 先将视频存在本地文件中
  62. String filename = handleVideoFiles(res, req);
  63. res.setData(filename);
  64. } else if (attachmentType == AttachmentType.VIDEO_COVER) {
  65. String picturebase = req.getParameter("picturebase");
  66. String filename = req.getParameter("filename");
  67. byte[] bs = Base64.getDecoder().decode(picturebase);
  68. filename = System.nanoTime()
  69. + filename.substring(filename.indexOf("."));
  70. res.setData(handleBaseFiles(res, "/video_cover/", false, bs, sign,
  71. TokenManager.getUserId(), filename));
  72. } else {
  73. res.getError().add(
  74. buildError(ErrorConstants.PARAM_ERROR, "", "文件标示"));
  75. }
  76. return res;
  77. }
  78. private String createFileName() {
  79. // 年月日
  80. Calendar now = Calendar.getInstance();
  81. int year = now.get(Calendar.YEAR);
  82. int month = now.get(Calendar.MONTH) + 1;
  83. int day = now.get(Calendar.DAY_OF_MONTH);
  84. String dir = "/" + year + "/" + month + "/" + day;
  85. return dir;
  86. }
  87. // 查看所有视频信息
  88. @RequestMapping(value = "/getVideoList", method = RequestMethod.GET)
  89. @ResponseBody
  90. public Result getVideoList(JtVideo video, Integer pageNo, Integer pageSize) {
  91. Result result = new Result();
  92. result.setData(adminVideoService.getVideoList(video, pageNo, pageSize,1));
  93. return result;
  94. }
  95. // 查看单个信息
  96. @RequestMapping(value = "/getVideoById", method = RequestMethod.GET)
  97. @ResponseBody
  98. public Result getVideoById(String id) {
  99. Result result = new Result();
  100. result.setData(adminVideoService.getVideoById(id));
  101. return result;
  102. }
  103. // 新增视频信息
  104. @RequestMapping(value = "/insertVideo", method = RequestMethod.POST)
  105. @ResponseBody
  106. public Result insertVideo(@Valid JtVideo video , BindingResult bindingResult) throws IOException,
  107. InterruptedException, ExecutionException {
  108. // 数据限制
  109. Result result = new Result();
  110. if (bindingResult.hasErrors()) {
  111. result.getError().add(buildErrorByMsg(bindingResult.getFieldError().getDefaultMessage(),
  112. VideoFields.getFieldDesc(bindingResult.getFieldError().getField())));
  113. return result;
  114. }
  115. String filename = video.getUrl();
  116. // 生成保存到数据库的url
  117. String output = System.nanoTime() + ".mp4";
  118. video.setUrl(createFileName() + "/" + output);
  119. video.setTranscoding(1);
  120. // 保存到数据库
  121. result.setData(adminVideoService.insertVideo(video));
  122. // 压缩视频并将视频
  123. videoChange(filename, output, video.getUrl());
  124. return result;
  125. }
  126. // 修改视频
  127. @RequestMapping(value = "/updateVideo", method = RequestMethod.POST)
  128. @ResponseBody
  129. public Result updateVideo(@Valid JtVideo video , BindingResult bindingResult) throws IOException,
  130. InterruptedException, ExecutionException {
  131. Result result = new Result();
  132. if (bindingResult.hasErrors()) {
  133. result.getError().add(buildErrorByMsg(bindingResult.getFieldError().getDefaultMessage(),
  134. VideoFields.getFieldDesc(bindingResult.getFieldError().getField())));
  135. return result;
  136. }
  137. JtVideo oldVideo = adminVideoService.getVideoById(video.getId());
  138. if (StringUtils.isBlank(video.getId())) {
  139. result.getError().add(
  140. buildError(ErrorConstants.PARAM_ERROR, "", "视频id不能为空"));
  141. return result;
  142. } else if (null == oldVideo) {
  143. result.getError().add(
  144. buildError(ErrorConstants.PARAM_ERROR, "", "视频id不存在"));
  145. return result;
  146. }
  147. if (StringUtils.isNotBlank(video.getUrl()) && !video.getUrl().equals(oldVideo.getUrl())) { // 如果修改了视频
  148. video.setTranscoding(1);
  149. // 压缩视频并将视频
  150. String filename = video.getUrl();
  151. String output = System.nanoTime() + ".mp4";
  152. video.setUrl(createFileName() + "/" + output);
  153. videoChange(filename, output, video.getUrl());
  154. }else{
  155. video.setUrl(null);
  156. }
  157. result.setData(adminVideoService.updateVideo(video));
  158. return result;
  159. }
  160. // 将文件转码
  161. private String videoChange(String input, String output, String url) throws IOException,
  162. InterruptedException, ExecutionException {
  163. FFmpeg ffmpeg = new FFmpeg(ffmpegPath);
  164. FFprobe ffprobe = new FFprobe(ffprobePath);
  165. FFmpegProbeResult in = ffprobe.probe(input);
  166. FFmpegBuilder builder = new FFmpegBuilder();
  167. builder.setInput(in) // 输入文件
  168. .overrideOutputFiles(true) // 覆盖重复文件
  169. .addOutput(output) // 输出文件
  170. .setFormat("mp4") // 设置格式
  171. .setTargetSize(250_000) // 目标大小
  172. .disableSubtitle() // 没有子标题
  173. .setAudioChannels(1) // 声道
  174. .setAudioSampleRate(48_000) // 音频采样率
  175. .setAudioBitRate(32768) // 音频传输率
  176. .setVideoCodec("libx264") // 视频编解码器
  177. .setVideoFrameRate(24, 1) // 视频帧速率
  178. .setVideoResolution(640, 480) // 视频分辨率
  179. .setStrict(FFmpegBuilder.Strict.EXPERIMENTAL) // 严格形式
  180. .done();
  181. FFmpegExecutor executor = new FFmpegExecutor(ffmpeg, ffprobe);
  182. ExecutorService service = Executors.newCachedThreadPool();
  183. // 3.直接new一个FutureTask
  184. SubTask subTask = new SubTask(executor, builder, in, url, input, output);
  185. FutureTask<Boolean> result = new FutureTask<Boolean>(subTask);
  186. // 4.提交任务
  187. service.submit(result);
  188. // 5.关闭线程池
  189. service.shutdown();
  190. // LOGGER.info("=============返回给前端=============");
  191. return "SUCESS";
  192. }
  193. private void updateSql(String url, String filename) {
  194. // 修改数据
  195. File file = new File(filename);
  196. JtVideo video = new JtVideo();
  197. video.setMd5(FileMD5(file));
  198. video.setTranscoding(2);
  199. video.setUrl(url);
  200. int c = adminVideoService.updateByUrl(video);
  201. if (c > 0){
  202. url = url.substring(0,url.lastIndexOf("/"));
  203. videoChange(url, filename);// 上传到远程
  204. }
  205. }
  206. private void videoChange(String url, String filename) {
  207. File f2 = new File(filename);// 临时压缩后的文件
  208. if (f2.exists()) {
  209. // 传到远程服务器
  210. FtpUtils ftp = new FtpUtils();
  211. ftp.uploadFile(url, filename, filename);
  212. // 删除视频
  213. f2.delete();
  214. }
  215. }
  216. // 获得文件md5
  217. private String FileMD5(File file) {
  218. try {
  219. FileInputStream fis = new FileInputStream(file);
  220. MessageDigest md = MessageDigest.getInstance("MD5");
  221. byte[] buffer = new byte[1024];
  222. int length = -1;
  223. while ((length = fis.read(buffer, 0, 1024)) != -1) {
  224. md.update(buffer, 0, length);
  225. }
  226. BigInteger bigInt = new BigInteger(1, md.digest());
  227. return bigInt.toString(16);
  228. } catch (FileNotFoundException e) {
  229. e.printStackTrace();
  230. } catch (NoSuchAlgorithmException e) {
  231. e.printStackTrace();
  232. } catch (IOException e) {
  233. e.printStackTrace();
  234. }
  235. return null;
  236. }
  237. class SubTask implements Callable<Boolean> {
  238. private Boolean state = false;
  239. private FFmpegExecutor executor;
  240. private FFmpegBuilder builder;
  241. private FFmpegProbeResult probeResult;
  242. private String url;
  243. private String input;
  244. private String output;
  245. public SubTask(FFmpegExecutor executor, FFmpegBuilder builder,
  246. FFmpegProbeResult probeResult, String url, String input,
  247. String output) {
  248. this.executor = executor;
  249. this.builder = builder;
  250. this.probeResult = probeResult;
  251. this.url = url;
  252. this.input = input;
  253. this.output = output;
  254. }
  255. public FFmpegExecutor getExecutor() {
  256. return executor;
  257. }
  258. public FFmpegBuilder getBuilder() {
  259. return builder;
  260. }
  261. public String getUrl() {
  262. return url;
  263. }
  264. public String getInput() {
  265. return input;
  266. }
  267. public String getOutput() {
  268. return output;
  269. }
  270. public Boolean getState() {
  271. return state;
  272. }
  273. @Override
  274. public Boolean call() throws Exception {
  275. final double duration_ns = probeResult.getFormat().duration
  276. * TimeUnit.SECONDS.toNanos(1);
  277. try {
  278. FFmpegJob job = executor.createJob(builder,
  279. new ProgressListener() {
  280. @Override
  281. public void progress(Progress progress) {
  282. double percentage = progress.out_time_ns
  283. / duration_ns;
  284. System.out.println(String
  285. .format("[%.0f%%] status:%s frame:%d time:%s ms fps:%.0f speed:%.2fx",
  286. percentage * 100,
  287. progress.status,
  288. progress.frame,
  289. FFmpegUtils.toTimecode(
  290. progress.out_time_ns,
  291. TimeUnit.NANOSECONDS),
  292. progress.fps.doubleValue(),
  293. progress.speed));
  294. }
  295. });
  296. job.run();
  297. // 6.阻塞 call
  298. // System.out.println("设置 sate");
  299. // latch.wait();
  300. state = true;
  301. updateSql(url, output);
  302. // 删除未压缩的视频
  303. File f1 = new File(input);
  304. f1.delete();
  305. } catch (Exception e) {
  306. e.printStackTrace();
  307. }
  308. return state;
  309. }
  310. }
  311. }