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 private String terminalSynchroDirectory;
00084
00085 private long sessionNumber = 0;
00086
00087 public SimpleMockSynchroManager(String terminalSynchroDirectory) {
00088 this.terminalSynchroDirectory = terminalSynchroDirectory;
00089 File newDir = new File(this.terminalSynchroDirectory+"/dbfiles");
00090 if (!newDir.exists()) newDir.mkdirs();
00091 File file = new File (terminalSynchroDirectory+"/filedir");
00092 if (!file.exists()) file.mkdirs();
00093 }
00094
00095 public void registerTerminalSynchroListener(OpenMSPSyncListener listener) {
00096 this.listener = listener;
00097 }
00098
00099 public OpenMSPSyncListener getTerminalSynchroListener() {
00100 return listener;
00101 }
00102
00103 public void registerServerSynchroTarget(OpenMSPSynchroTargetListener target) {
00104 this.target = target;
00105 }
00106
00107 public OpenMSPSynchroTargetListener getServerSynchroTarget() {
00108 return target;
00109 }
00110
00111 public void synchronize(String group, Credential cred, SynchroDescriptor synchrodescriptor) throws OpenMSPException {
00112 PropertiesManager.getManager().addProperty("org.openmobileis.synchro.client.openmsp.numsyncpropsfile", System.getProperty("user.dir") + File.separator +"numsync.txt");
00113 OpenMSPSyncMessageFactory messageFactory = new OpenMSPSyncMessageFactory();
00114 Array listenerList = new Array();
00115 listenerList.add(this.listener);
00116 try {
00117 messageFactory.beginSynchro(cred, synchrodescriptor);
00118 listener.startSync(cred, synchrodescriptor);
00119
00120 while (true) {
00121 Message message = messageFactory.getOpenMSPMessage(listenerList);
00122 if (message.isEmpty()) {
00123
00124 break;
00125 }
00126 Message CMLAnswer = MessageFactory.getFactory().createMessage(new Header(this.getNextSessionNumber(), target.getTargetName(), listener.getSyncName(), null, null));
00127 message.resetCursor();
00128 while (message.hasMoreMessage()) {
00129 ContainerMessage CMLRequest = message.nextMessage();
00130 SyncTargetAnswer listenerAnswer = target.processCommand(cred, CMLRequest);
00131
00132 HashMap zipFileMap = new HashMap(10);
00133 if (listenerAnswer.answerfiles != null) {
00134 for (int i=0; i<listenerAnswer.answerfiles.length; i++) {
00135 if (listenerAnswer.answerfiles[i] instanceof ZipEntryMemoryFile) {
00136 String zipFileName = ((ZipEntryMemoryFile) listenerAnswer.answerfiles[i]).getZipFileName();
00137 ZipArrayEntryFile zipEntry = (ZipArrayEntryFile) zipFileMap.get(zipFileName);
00138 if (zipEntry == null) {
00139 zipEntry = new ZipArrayEntryFile();
00140 zipEntry.array = new ByteArrayOutputStream(100000);
00141 zipEntry.zip = new ZipOutputStream(zipEntry.array);
00142 zipEntry.fileName = zipFileName;
00143 zipFileMap.put(zipFileName, zipEntry);
00144 }
00145 ZipEntry entry = new ZipEntry(FileUtilities.convertToUnixFomat(listenerAnswer.answerfiles[i].getFileName()));
00146 entry.setTime(listenerAnswer.answerfiles[i].getFileDate());
00147 entry.setMethod(ZipEntry.DEFLATED);
00148 zipEntry.zip.putNextEntry(entry);
00149 zipEntry.zip.write(listenerAnswer.answerfiles[i].getFileData());
00150 zipEntry.zip.closeEntry();
00151 } else {
00152 String path = this.terminalSynchroDirectory+listenerAnswer.answerfiles[i].getFilePath();
00153 File newDir = new File(path);
00154 if (!newDir.exists()) newDir.mkdirs();
00155 FileOutputStream out = new FileOutputStream(path+listenerAnswer.answerfiles[i].getFileName());
00156 try {
00157 out.write(listenerAnswer.answerfiles[i].getFileData());
00158 } finally {
00159 out.flush();
00160 out.close();
00161 }
00162 }
00163 }
00164
00165 Iterator iter = zipFileMap.keySet().iterator();
00166 while (iter.hasNext()) {
00167 String filename = (String) iter.next();
00168 ZipArrayEntryFile zipEntry = (ZipArrayEntryFile) zipFileMap.get(filename);
00169 zipEntry.zip.flush();
00170 zipEntry.zip.close();
00171 FileOutputStream out = new FileOutputStream(this.terminalSynchroDirectory+filename);
00172 try {
00173 out.write(zipEntry.array.toByteArray());
00174 } finally {
00175 out.flush();
00176 out.close();
00177 }
00178 }
00179 }
00180
00181 if (listenerAnswer.containerMessage != null) {
00182 int size = listenerAnswer.containerMessage.length;
00183 for (int i = 0; i < size; i++) {
00184 CMLAnswer.add(listenerAnswer.containerMessage[i]);
00185 }
00186 }
00187 }
00188
00189
00190 messageFactory.receiveOpenMSPMessage(listenerList, CMLAnswer);
00191 }
00192 listener.endSync();
00193 } catch (Throwable ex) {
00194 listener.notifySynchroFailure();
00195 throw new OpenMSPException(ex);
00196 } finally {
00197 messageFactory.endSynchro();
00198
00199 File file = new File(System.getProperty("user.dir") + File.separator +"numsync.txt");
00200 if (file.exists()) file.delete();
00201
00202 }
00203
00204 }
00205
00206 private long getNextSessionNumber() {
00207 return sessionNumber + 1;
00208 }
00209
00210 }