com.scudata.parallel.RemoteFileProxyManager Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of esproc Show documentation
Show all versions of esproc Show documentation
SPL(Structured Process Language) A programming language specially for structured data computing.
package com.scudata.parallel;
import java.io.*;
import java.util.*;
import com.scudata.common.Logger;
import com.scudata.dm.*;
import com.scudata.server.unit.UnitServer;
/**
* Զ???ļ???????????
*
* @author Joancy
*
*/
public class RemoteFileProxyManager {
static ArrayList proxys = new ArrayList();
/**
* ??ȡ?ļ??????б?
* @return
*/
public static ArrayList getFileProxys(){
return proxys;
}
/**
* ִ?д???????????
* @param req ????
* @param sd ͨѶ????
* @return ??Ӧ
*/
public static Response execute(Request req, SocketData sd) {
Response res = new Response();
String fileName;
Integer partition;
FileObject fo;
int id;
RemoteFileProxy rfp;
try {
switch (req.getAction()) {
case Request.FILE_OPEN:
fileName = (String) req.getAttr(Request.OPEN_FileName);
partition = (Integer) req.getAttr(Request.OPEN_Partition);
// lf = new LocalFile(fileName, "s", partition);
id = UnitServer.nextId();
String opt = (String) req.getAttr(Request.OPEN_Opt);//?˴???Ҫ??FILE_GETPROPERTYͬ??
if(opt!=null){
opt = opt.toLowerCase();
if(opt.indexOf('t')>=0){
fileName = new File(Env.getTempPath(),fileName).getPath();
}
}
Boolean isAppend = (Boolean) req.getAttr(Request.OPEN_IsAppend);
rfp = new RemoteFileProxy(fileName, partition, id,isAppend);
addProxy(rfp);
res.setResult(new Integer(id));
break;
case Request.FILE_FROM_HANDLE:
id = (Integer) req.getAttr(Request.FROM_Handle);
long rpos = (Long) req.getAttr(Request.FROM_Pos);
rfp = getProxy(id);
rfp.setReadPosition(rpos);
res.setResult(new Integer(id));
break;
case Request.FILE_GETPROPERTY:
fileName = (String) req.getAttr(Request.GETPROPERTY_FileName);
String op = (String) req.getAttr(Request.OPEN_Opt);//?˴???Ҫ??FILE_OPENͬ????ѡ?????ĸ?·??????????ļ?λ?õIJ?ͬ
if(op!=null){
op = op.toLowerCase();
if(op.indexOf('t')>=0){
// opt??tʱ??Ŀ¼ΪEnv.getTempPath()??partition????????
fileName = new File(Env.getTempPath(),fileName).getPath();
}
}
fo = new FileObject(fileName);
HashMap properties = new HashMap();
properties.put("exists", new Boolean(fo.isExists()));
properties.put("lastModified", new Long(fo.getFile().lastModified()));
properties.put("size", new Long(fo.size()));
res.setResult(properties);
break;
case Request.FILE_DELETE:
fileName = (String) req.getAttr(Request.DELETE_FileName);
fo = new FileObject(fileName);
boolean b = fo.delete();
Logger.debug("Delete file "+fileName+": "+b);
res.setResult(new Boolean(b));
break;
case Request.FILE_READ:
id = ((Number) req.getAttr(Request.READ_Handle)).intValue();
rfp = getProxy(id);
int bufSize = ((Number) req.getAttr(Request.READ_BufferSize))
.intValue();
try {
Object result = rfp.read(bufSize);
res.setResult(result);
} catch (Exception x) {
res.setException(x);
}
break;
case Request.FILE_WRITE:
id = ((Number) req.getAttr(Request.WRITE_Handle)).intValue();
rfp = getProxy(id);
byte[] bytes = (byte[]) req.getAttr(Request.WRITE_Bytes);
try {
rfp.write(bytes);
} catch (Exception x) {
res.setException(x);
}
break;
case Request.FILE_CLOSE:
id = ((Number) req.getAttr(Request.CLOSE_Handle)).intValue();
rfp = getProxy(id);
rfp.close();
break;
case Request.FILE_POSITION:
id = ((Number) req.getAttr(Request.POSITION_Handle)).intValue();
rfp = getProxy(id);
Object result = rfp.position();
res.setResult(result);
break;
case Request.FILE_SETPOSITION:
id = ((Number) req.getAttr(Request.SETPOSITION_Handle)).intValue();
rfp = getProxy(id);
long pos = ((Number) req.getAttr(Request.SETPOSITION_Position)).longValue();
rfp.setPosition(pos);
break;
case Request.FILE_TRYLOCK:
id = ((Number) req.getAttr(Request.TRYLOCK_Handle)).intValue();
rfp = getProxy(id);
boolean isLock = rfp.tryLock();
res.setResult(new Boolean(isLock));
break;
case Request.FILE_LOCK:
id = ((Number) req.getAttr(Request.LOCK_Handle)).intValue();
rfp = getProxy(id);
boolean lock = rfp.lock();
res.setResult(new Boolean(lock));
break;
case Request.FILE_DIRECTREAD:
fileName = (String) req.getAttr(Request.DIRECTREAD_FileName);
partition = (Integer) req.getAttr(Request.DIRECTREAD_Partition);
fo = new FileObject(fileName);
fo.setPartition( partition );
InputStream is = fo.getInputStream();
byte[] fileBuf = read(is, Env.FILE_BUFSIZE);
sd.write(fileBuf);
while (fileBuf != null) {
fileBuf = read(is, Env.FILE_BUFSIZE);
sd.write(fileBuf);
}
is.close();
res.setResult(Boolean.TRUE);
break;
}
} catch (Exception x) {
Logger.debug(x);
res.setException(x);
}
return res;
}
/**
* ????????is??ȡsize?ֽڵ?????
* @param is ??????
* @param size ?ֽ???
* @return ?ֽ?????
* @throws Exception ??ȡʧ???׳??쳣
*/
public static byte[] read(InputStream is, int size) throws Exception {
byte[] buf = new byte[size];
int len = is.read(buf);
byte[] buf2;
if (len <= 0) {
buf2 = null;
} else if (len != size) {
buf2 = new byte[len];
System.arraycopy(buf, 0, buf2, 0, len);
} else {
buf2 = buf;
}
return buf2;
}
/**
* ????һ???ļ?????????
* @param rfp ?ļ?????
*/
public static synchronized void addProxy(RemoteFileProxy rfp) {
proxys.add(rfp);
}
/**
* ɾ??ָ???Ĵ???
* @param proxyID Ҫɾ???Ĵ??????
*/
public static synchronized void delProxy(int proxyID) {
for (int i = 0; i < proxys.size(); i++) {
RemoteFileProxy rfp = proxys.get(i);
if (rfp.getProxyID() == proxyID) {
proxys.remove(i);
break;
}
}
}
/**
* ??ȡָ?????proxyId?Ĵ???????
* @param proxyId ???
* @return ????????
* @throws Exception û?ҵ?ʱ?׳??쳣
*/
public static synchronized RemoteFileProxy getProxy(int proxyId)
throws Exception {
for (int i = 0; i < proxys.size(); i++) {
RemoteFileProxy rfp = proxys.get(i);
if (rfp.getProxyID() == proxyId) {
return rfp;
}
}
throw new Exception("Remote file id:" + proxyId + " is timeout.");
}
/**
* ????????ʱ
*/
public static synchronized void checkTimeOut(int proxyTimeOut) {
// ??????룬timeOut??λΪ??
for (int i = proxys.size() - 1; i >= 0; i--) {
RemoteFileProxy rft = proxys.get(i);
if (rft.checkTimeOut(proxyTimeOut)) {
proxys.remove(rft);
}
}
}
}