001    /**
002     *   GRANITE DATA SERVICES
003     *   Copyright (C) 2006-2013 GRANITE DATA SERVICES S.A.S.
004     *
005     *   This file is part of Granite Data Services.
006     *
007     *   Granite Data Services is free software; you can redistribute it and/or modify
008     *   it under the terms of the GNU Library General Public License as published by
009     *   the Free Software Foundation; either version 2 of the License, or (at your
010     *   option) any later version.
011     *
012     *   Granite Data Services is distributed in the hope that it will be useful, but
013     *   WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
014     *   FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public License
015     *   for more details.
016     *
017     *   You should have received a copy of the GNU Library General Public License
018     *   along with this library; if not, see <http://www.gnu.org/licenses/>.
019     */
020    package org.granite.client.messaging.codec;
021    
022    import java.io.ByteArrayInputStream;
023    import java.io.IOException;
024    import java.io.InputStream;
025    import java.io.OutputStream;
026    
027    import org.granite.client.messaging.channel.Channel;
028    import org.granite.client.messaging.jmf.ClientSharedContext;
029    import org.granite.messaging.jmf.JMFDeserializer;
030    import org.granite.messaging.jmf.JMFSerializer;
031    import org.granite.util.ContentType;
032    
033    import flex.messaging.messages.Message;
034    
035    /**
036     * @author Franck WOLFF
037     */
038    public class JMFAMF3MessagingCodec implements MessagingCodec<Message[]> {
039            
040            private final ClientSharedContext sharedContext;
041            
042            public JMFAMF3MessagingCodec(ClientSharedContext sharedContext) {
043                    this.sharedContext = sharedContext;
044            }
045    
046            @Override
047            public ClientType getClientType() {
048                    return ClientType.JAVA;
049            }
050    
051            @Override
052            public String getContentType() {
053                    return ContentType.JMF_AMF.mimeType();
054            }
055    
056            @Override
057            public void encode(Message[] messages, OutputStream output) throws IOException {
058                    JMFSerializer serializer = new JMFSerializer(output, sharedContext);
059                    serializer.writeObject(messages);
060            }
061    
062            @Override
063            public Message[] decode(InputStream input) throws IOException {
064                    JMFDeserializer deserializer = new JMFDeserializer(input, sharedContext);
065                    
066                    Message[] messages = null;
067                    try {
068                            messages = (Message[])deserializer.readObject();
069                            if (messages != null) {
070                                    for (Message message : messages) {
071                                            if (message != null && Boolean.TRUE.equals(message.getHeader(Channel.BYTEARRAY_BODY_HEADER))) {
072                                                    byte[] body = (byte[])message.getBody();
073                                                    deserializer = new JMFDeserializer(new ByteArrayInputStream(body), sharedContext);
074                                                    message.setBody(deserializer.readObject());
075                                            }
076                                    }
077                            }
078                    }
079                    catch (ClassNotFoundException e) {
080                            throw new IOException(e);
081                    }
082                    
083                    return (messages != null ? messages : new Message[0]);
084            }
085    }