SSM下整合netty使用队列进行文件传输demo

原创文章

自己在闲暇无事的时候做的一些小demo，在这里分享出来给大家

@Service
@Scope("prototype")
public class PictureServiceImpl implements PictureService {

    @Value("${UPLOAD_PATH}")
    private String UPLOAD_PATH;

    @Value("${IMAGES_URL}")
    private String IMAGES_URL;

    @Value("${NETTY_URL}")
    private String NETTY_URL;

    @Value("${NETTY_PORT}")
    private Integer NETTY_PORT;

    @Autowired
    private FileThreadManager fileThreadManager;

    @Override
    public PictureResult uploadFile(MultipartFile file) {
        PictureResult pictureResult = new PictureResult();
        if (!file.isEmpty()) {
            try {
                String fileName = file.getOriginalFilename();
                String newname = UUID.randomUUID() + fileName.substring(fileName.lastIndexOf("."));
                new Thread(new Runnable() {
                    @Override
                    public void run() {
                        try {
                            fileThreadManager.start(NETTY_URL, NETTY_PORT);
                        } catch (InterruptedException e) {
                            e.printStackTrace();
                        }
                    }
                }).start();
                fileThreadManager.submit(file.getBytes(), newname, null);
                System.out.println("尝试获取数据");
                while (fileThreadManager.getFileInfoByName(newname) == null) {}
                FileInfo fileInfo = fileThreadManager.getFileInfoByName(newname);
                BeanUtils.copyProperties(fileInfo, pictureResult);
            } catch (Exception e) {
                e.printStackTrace();
                pictureResult.setError();
                pictureResult.setMessage("上传失败");
            }
        } else {
            pictureResult.setMessage("上传图片为空");
            pictureResult.setError();
        }
        return pictureResult;
    }

}
           

下面是主要的队列及netty的主要代码

@Component
@Scope("prototype")
public class FileThreadManager implements BeanFactoryAware{

    private BeanFactory beanFactory;

    private ExecutorService pool = Executors.newCachedThreadPool();

    private volatile ChannelFuture cf;

    private ConcurrentHashMap<String, FileInfo> map=new ConcurrentHashMap<>();

    @Autowired
    private CustomChannelHandler customChannelHandler;

    public void submit(byte[] file,String fileNewName,String message){
        while(cf==null){}
        System.out.println("提交任务");
        FileThread fileThread = beanFactory.getBean(FileThread.class);
        fileThread.setFile(file);
        fileThread.setFileNewName(fileNewName);
        fileThread.setMessage(message);
        fileThread.setCf(cf);
        pool.execute(fileThread);
    }

    public void start(String nettyUrl,Integer nettyPort) throws InterruptedException{
        if(cf!=null)
            return ;
        EventLoopGroup workgroup = new NioEventLoopGroup();
        Bootstrap b = new Bootstrap();
        b.group(workgroup);
        b.channel(NioSocketChannel.class).handler(new ChannelInitializer<SocketChannel>() {

            @Override
            protected void initChannel(SocketChannel sc) throws Exception {
                // TODO Auto-generated method stub
                sc.pipeline().addLast(MarshallingCodeCFactory.buildMarshallingDecoder());
                sc.pipeline().addLast(MarshallingCodeCFactory.buildMarshallingEncoder());
                sc.pipeline().addLast(customChannelHandler);
                customChannelHandler.setMap(map);
            }
        });
        b.option(ChannelOption.SO_KEEPALIVE, true);
        ChannelFuture cf = b.connect(nettyUrl, nettyPort).sync();
        this.cf=cf;
        cf.channel().closeFuture().sync();
        workgroup.shutdownGracefully();
    }

    public FileInfo getFileInfoByName(String fileName) throws InterruptedException{
        Set<Entry<String, FileInfo>> entrySet = map.entrySet();
        for (Entry<String, FileInfo> entry : entrySet) {
            if(entry.getKey().trim().equals(fileName.trim())){
                return entry.getValue();
            }
        }
        return null;
    }

    @Override
    public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
        // TODO Auto-generated method stub
        this.beanFactory=beanFactory;
    }
}
           

最后说一下，FileThread，FileInfo和PictureResult等类都是自己自定义的类型，根据自己的需要从而定义。