org.coodex.concrete.attachments.server.UploadByFormResource.java Source code

Introduction

Here is the source code for org.coodex.concrete.attachments.server.UploadByFormResource.java

Source

/*
 * Copyright (c) 2017 coodex.org (jujus.shen@126.com)
 *
 * 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 org.coodex.concrete.attachments.server;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.coodex.concrete.attachments.AttachmentEntityInfo;
import org.coodex.concrete.attachments.AttachmentInfo;
import org.coodex.concrete.attachments.AttachmentServiceHelper;
import org.coodex.util.Common;

import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.*;
import javax.ws.rs.container.AsyncResponse;
import javax.ws.rs.container.Suspended;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import java.util.ArrayList;
import java.util.List;

/**
 * ?jaxrs2.0;apache commons-fileupload
 * post /attachments/upload/byform/{clientId}/{tokenId}
 * Created by davidoff shen on 2016-12-13.
 */
@Path("attachments/upload/byform")
public class UploadByFormResource extends AbstractUploadResource {

    @Path("/{clientId}/{tokenId}")
    @POST
    @Consumes({ MediaType.APPLICATION_FORM_URLENCODED, MediaType.MULTIPART_FORM_DATA })
    @Produces(MediaType.APPLICATION_JSON)
    public void uploadByForm(@Suspended final AsyncResponse asyncResponse,
            @Context final HttpServletRequest request, @PathParam("clientId") final String clientId,
            @PathParam("tokenId") final String tokenId) {

        Thread t = new Thread(new Runnable() {
            @Override
            public void run() {

                try {
                    ServletFileUpload uploadHandler = new ServletFileUpload(new DiskFileItemFactory());
                    List<FileItem> items = uploadHandler.parseRequest(request);
                    List<AttachmentEntityInfo> result = new ArrayList<AttachmentEntityInfo>();
                    for (FileItem item : items) {
                        if (!item.isFormField()) {
                            if (!Common.isBlank(item.getName())) {
                                AttachmentInfo attachmentInfo = new AttachmentInfo();
                                attachmentInfo.setName(item.getName());
                                attachmentInfo.setOwner(clientId);
                                attachmentInfo.setSize(item.getSize());
                                attachmentInfo.setContentType(item.getContentType());
                                result.add(saveToRepo(clientId, tokenId, attachmentInfo, item.getInputStream()));
                            }
                        }
                    }
                    asyncResponse.resume(result);
                } catch (Throwable t) {
                    asyncResponse.resume(t);
                }
            }
        });
        t.setPriority(AttachmentServiceHelper.ATTACHMENT_PROFILE.getInt("upload.priority", 5));
        t.start();
    }

}