00001
00025 package org.openmobileis.test.mock.synchro;
00026
00027 import java.io.ByteArrayOutputStream;
00028 import java.io.File;
00029 import java.io.FileOutputStream;
00030 import java.util.HashMap;
00031 import java.util.Iterator;
00032 import java.util.zip.ZipEntry;
00033 import java.util.zip.ZipOutputStream;
00034
00035 import org.openmobileis.common.util.PropertiesManager;
00036 import org.openmobileis.common.util.collection.Array;
00037 import org.openmobileis.common.util.file.FileUtilities;
00038 import org.openmobileis.synchro.client.SynchroDescriptor;
00039 import org.openmobileis.synchro.openmsp.OpenMSPException;
00040 import org.openmobileis.synchro.openmsp.client.OpenMSPSyncListener;
00041 import org.openmobileis.synchro.openmsp.client.core.OpenMSPSyncMessageFactory;
00042 import org.openmobileis.synchro.openmsp.protocol.ContainerMessage;
00043 import org.openmobileis.synchro.openmsp.protocol.Header;
00044 import org.openmobileis.synchro.openmsp.protocol.Message;
00045 import org.openmobileis.synchro.openmsp.protocol.MessageFactory;
00046 import org.openmobileis.synchro.openmsp.server.synctarget.OpenMSPSynchroTargetListener;
00047 import org.openmobileis.synchro.openmsp.server.synctarget.SyncTargetAnswer;
00048 import org.openmobileis.synchro.openmsp.server.util.ZipEntryMemoryFile;
00049 import org.openmobileis.synchro.security.auth.Credential;
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059 public final class SimpleMockSynchroManager {
00060 class ZipArrayEntryFile {
00061 ByteArrayOutputStream array;
00062
00063 ZipOutputStream zip;
00064
00065 String fileName;
00066
00067 public int hashCode() {
00068 return fileName.hashCode();
00069 }
00070
00071 public boolean equals(Object obj) {
00072 if (obj instanceof ZipArrayEntryFile) {
00073 return ((ZipArrayEntryFile) obj).fileName.equals(this.fileName);
00074 }
00075 return false;
00076 }
00077
00078 }
00079
00080 private OpenMSPSyncListener listener;
00081
00082 private OpenMSPSynchroTargetListener target;
00083
00084
00085 private String terminalSynchroDirectory;
00086
00087 private long sessionNumber = 0;
00088
00089 public SimpleMockSynchroManager(String terminalSynchroDirectory) {
00090 this.terminalSynchroDirectory = terminalSynchroDirectory;
00091 File newDir = new File(this.terminalSynchroDirectory+"/dbfiles");
00092 if (!newDir.exists()) newDir.mkdirs();
00093 File file = new File (terminalSynchroDirectory+"/filedir");
00094 if (!file.exists()) file.mkdirs();
00095 }
00096
00097 public void registerTerminalSynchroListener(OpenMSPSyncListener listener) {
00098 this.listener = listener;
00099 }
00100
00101 public OpenMSPSyncListener getTerminalSynchroListener() {
00102 return listener;
00103 }
00104
00105 public void registerServerSynchroTarget(OpenMSPSynchroTargetListener target) {
00106 this.target = target;
00107 }
00108
00109 public OpenMSPSynchroTargetListener getServerSynchroTarget() {
00110 return target;
00111 }
00112
00113 public void synchronize(String group, Credential cred, SynchroDescriptor synchrodescriptor) throws OpenMSPException {
00114 PropertiesManager.getManager().addProperty("org.openmobileis.synchro.client.openmsp.numsyncpropsfile", System.getProperty("user.dir") + File.separator +"numsync.txt");
00115 OpenMSPSyncMessageFactory messageFactory = new OpenMSPSyncMessageFactory();
00116 Array listenerList = new Array();
00117 listenerList.add(this.listener);
00118 try {
00119 messageFactory.beginSynchro(cred, synchrodescriptor);
00120 listener.startSync(cred, synchrodescriptor);
00121
00122 while (true) {
00123 Message message = messageFactory.getOpenMSPMessage(listenerList);
00124 if (message.isEmpty()) {
00125
00126 break;
00127 }
00128 Message CMLAnswer = MessageFactory.getFactory().createMessage(new Header(this.getNextSessionNumber(), target.getTargetName(), listener.getSyncName(), null, null));
00129 message.resetCursor();
00130 while (message.hasMoreMessage()) {
00131 ContainerMessage CMLRequest = message.nextMessage();
00132 SyncTargetAnswer listenerAnswer = target.processCommand(cred, CMLRequest);
00133
00134 HashMap zipFileMap = new HashMap(10);
00135 if (listenerAnswer.answerfiles != null) {
00136 for (int i=0; i<listenerAnswer.answerfiles.length; i++) {
00137 if (listenerAnswer.answerfiles[i] instanceof ZipEntryMemoryFile) {
00138 String zipFileName = ((ZipEntryMemoryFile) listenerAnswer.answerfiles[i]).getZipFileName();
00139 ZipArrayEntryFile zipEntry = (ZipArrayEntryFile) zipFileMap.get(zipFileName);
00140 if (zipEntry == null) {
00141 zipEntry = new ZipArrayEntryFile();
00142 zipEntry.array = new ByteArrayOutputStream(100000);
00143 zipEntry.zip = new ZipOutputStream(zipEntry.array);
00144 zipEntry.fileName = zipFileName;
00145 zipFileMap.put(zipFileName, zipEntry);
00146 }
00147 ZipEntry entry = new ZipEntry(FileUtilities.convertToUnixFomat(listenerAnswer.answerfiles[i].getFileName()));
00148 entry.setTime(listenerAnswer.answerfiles[i].getFileDate());
00149 entry.setMethod(ZipEntry.DEFLATED);
00150 zipEntry.zip.putNextEntry(entry);
00151 zipEntry.zip.write(listenerAnswer.answerfiles[i].getFileData());
00152 zipEntry.zip.closeEntry();
00153 } else {
00154 String path = this.terminalSynchroDirectory+listenerAnswer.answerfiles[i].getFilePath();
00155 File newDir = new File(path);
00156 if (!newDir.exists()) newDir.mkdirs();
00157 FileOutputStream out = new FileOutputStream(path+listenerAnswer.answerfiles[i].getFileName());
00158 try {
00159 out.write(listenerAnswer.answerfiles[i].getFileData());
00160 } finally {
00161 out.flush();
00162 out.close();
00163 }
00164 }
00165 }
00166
00167 Iterator iter = zipFileMap.keySet().iterator();
00168 while (iter.hasNext()) {
00169 String filename = (String) iter.next();
00170 ZipArrayEntryFile zipEntry = (ZipArrayEntryFile) zipFileMap.get(filename);
00171 zipEntry.zip.flush();
00172 zipEntry.zip.close();
00173 FileOutputStream out = new FileOutputStream(this.terminalSynchroDirectory+filename);
00174 try {
00175 out.write(zipEntry.array.toByteArray());
00176 } finally {
00177 out.flush();
00178 out.close();
00179 }
00180 }
00181 }
00182
00183 if (listenerAnswer.containerMessage != null) {
00184 int size = listenerAnswer.containerMessage.length;
00185 for (int i = 0; i < size; i++) {
00186 CMLAnswer.add(listenerAnswer.containerMessage[i]);
00187 }
00188 }
00189 }
00190
00191
00192 messageFactory.receiveOpenMSPMessage(listenerList, CMLAnswer);
00193 }
00194 listener.endSync();
00195 } catch (Throwable ex) {
00196 listener.notifySynchroFailure();
00197 throw new OpenMSPException(ex);
00198 } finally {
00199 messageFactory.endSynchro();
00200
00201 File file = new File(System.getProperty("user.dir") + File.separator +"numsync.txt");
00202 if (file.exists()) file.delete();
00203
00204 }
00205
00206 }
00207
00208 private long getNextSessionNumber() {
00209 return sessionNumber + 1;
00210 }
00211
00212 }