112 lines
4.3 KiB
Java
Executable File
112 lines
4.3 KiB
Java
Executable File
/**
|
|
* Copyright 2018-2028 Akaxin Group
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package com.windchat.im.business.impl.tai;
|
|
|
|
import com.akaxin.proto.core.FileProto;
|
|
import com.akaxin.proto.site.ApiFileDownloadProto;
|
|
import com.akaxin.proto.site.ApiFileUploadProto;
|
|
import com.google.protobuf.ByteString;
|
|
import com.windchat.common.command.Command;
|
|
import com.windchat.common.command.CommandResponse;
|
|
import com.windchat.common.constant.ErrorCode2;
|
|
import com.windchat.common.logs.LogUtils;
|
|
import com.windchat.im.business.bean.ApiActions;
|
|
import com.windchat.im.business.impl.AbstractRequest;
|
|
import com.windchat.im.business.utils.FilePathUtils;
|
|
import com.windchat.im.business.utils.FileServerUtils;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
/**
|
|
* 文件（图片，语音）上传下载
|
|
*
|
|
* @author Sam{@link an.guoyue254@gmail.com}
|
|
* @since 2018-01-25 17:52:08
|
|
*/
|
|
@ApiActions(action = "api.file")
|
|
public class ApiFileService extends AbstractRequest {
|
|
private static Logger logger = LoggerFactory.getLogger(ApiFileService.class);
|
|
|
|
@ApiActions(action = ".upload")
|
|
public CommandResponse upload(Command command) {
|
|
CommandResponse commandResponse = new CommandResponse();
|
|
ErrorCode2 errCode = ErrorCode2.ERROR;
|
|
try {
|
|
ApiFileUploadProto.ApiFileUploadRequest request = ApiFileUploadProto.ApiFileUploadRequest
|
|
.parseFrom(command.getParams());
|
|
FileProto.File file = request.getFile();
|
|
FileProto.FileDesc fileDesc = request.getFileDesc();
|
|
FileProto.FileType fileType = file.getFileType();
|
|
byte[] content = file.getFileContent().toByteArray();
|
|
LogUtils.requestDebugLog(logger, command, request.toString());
|
|
|
|
String fileId = FileServerUtils.saveFile(content, FilePathUtils.getPicPath(), fileType, fileDesc);
|
|
ApiFileUploadProto.ApiFileUploadResponse response = ApiFileUploadProto.ApiFileUploadResponse.newBuilder()
|
|
.setFileId(fileId).build();
|
|
commandResponse.setParams(response.toByteArray());
|
|
errCode = ErrorCode2.SUCCESS;
|
|
} catch (Exception e) {
|
|
errCode = ErrorCode2.ERROR_SYSTEMERROR;
|
|
LogUtils.requestErrorLog(logger, command, e);
|
|
}
|
|
return commandResponse.setErrCode2(errCode);
|
|
}
|
|
|
|
@ApiActions(action = ".download")
|
|
public CommandResponse download(Command command) {
|
|
CommandResponse commandResponse = new CommandResponse();
|
|
ErrorCode2 errCode = ErrorCode2.ERROR;
|
|
try {
|
|
ApiFileDownloadProto.ApiFileDownloadRequest request = ApiFileDownloadProto.ApiFileDownloadRequest
|
|
.parseFrom(command.getParams());
|
|
// String fileId = request.getFileId();
|
|
String reqFileId = request.getFileId();
|
|
LogUtils.requestDebugLog(logger, command, request.toString());
|
|
|
|
if (StringUtils.isNotBlank(reqFileId) && !"null".equals(reqFileId)) {
|
|
String fileId = reqFileId;
|
|
if (fileId.startsWith("AKX-") || fileId.startsWith("akx-")) {
|
|
fileId = fileId.substring(4, fileId.length());
|
|
}
|
|
|
|
byte[] imageBytes = FileServerUtils.fileToBinary(FilePathUtils.getPicPath(), fileId);
|
|
|
|
if (imageBytes != null && imageBytes.length > 0) {
|
|
FileProto.File file = FileProto.File.newBuilder().setFileId(reqFileId)
|
|
.setFileContent(ByteString.copyFrom(imageBytes)).build();
|
|
|
|
ApiFileDownloadProto.ApiFileDownloadResponse response = ApiFileDownloadProto.ApiFileDownloadResponse
|
|
.newBuilder().setFile(file).build();
|
|
|
|
commandResponse.setParams(response.toByteArray());
|
|
errCode = ErrorCode2.SUCCESS;
|
|
} else {
|
|
// 获取文件资源失败，文件可能不存在，此时需要抛出异常
|
|
errCode = ErrorCode2.ERROR2_FILE_DOWNLOAD;
|
|
}
|
|
} else {
|
|
errCode = ErrorCode2.ERROR_PARAMETER;
|
|
}
|
|
} catch (Exception e) {
|
|
errCode = ErrorCode2.ERROR_SYSTEMERROR;
|
|
LogUtils.requestErrorLog(logger, command, e);
|
|
}
|
|
return commandResponse.setErrCode2(errCode);
|
|
}
|
|
|
|
}
|