com.scudata.dm.op.SwitchRemote 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.dm.op;
import com.scudata.array.IArray;
import com.scudata.array.ObjectArray;
import com.scudata.common.MessageManager;
import com.scudata.common.RQException;
import com.scudata.dm.BaseRecord;
import com.scudata.dm.Context;
import com.scudata.dm.DataStruct;
import com.scudata.dm.IndexTable;
import com.scudata.dm.Record;
import com.scudata.dm.Sequence;
import com.scudata.expression.CurrentSeq;
import com.scudata.expression.Expression;
import com.scudata.expression.Function;
import com.scudata.parallel.ClusterMemoryTable;
import com.scudata.resources.EngineMessage;
/**
* ??Զ??ά???????Ӳ????????????С??αꡢ?ܵ??????Ӳ???
* @author WangXiaoJun
*
*/
public class SwitchRemote extends Operation {
private String []fkNames; // ????ֶ???????
private Object[] datas; // ????ά??????Ⱥά??????
private Expression []exps; // ά????????????ʽ????
private String opt; // ѡ??
private int []fkIndex; // ????ֶ?????????
private Sequence []codes;
private IndexTable []indexTables; // ά????Ӧ????????????
private ClusterMemoryTable []cts; // ??Ⱥά??????
private boolean isIsect; // ????????
private boolean isDiff; // ????????
private boolean isLeft; // ?????ӣ??Ҳ???F??Ӧֵʱ???????????ݽṹ???ɿ?ֵ?????????⣩??¼??Ӧ
private DataStruct []dataStructs; // isLeftΪtrueʱʹ?ã???Ӧά???????ݽṹ
private int []keySeqs; // ά???????ֶε????
public SwitchRemote(String[] fkNames, Object[] datas, Expression[] exps, String opt) {
this(null, fkNames, datas, exps, opt);
}
public SwitchRemote(Function function, String[] fkNames,Object[] datas, Expression[] exps, String opt) {
super(function);
this.fkNames = fkNames;
this.datas = datas;
this.exps = exps;
this.opt = opt;
if (opt != null) {
if (opt.indexOf('i') != -1) {
isIsect = true;
} else if (opt.indexOf('d') != -1) {
isDiff = true;
} else if (opt.indexOf('1') != -1) {
isLeft = true;
}
}
}
/**
* ???ش˲????Ƿ??ʹ??¼??????
* @return true?????ʹ??¼?????٣?false?????????
*/
public boolean isDecrease() {
return isIsect || isDiff;
}
/**
* ???ƴ˲???
* @param ctx ??????????
* @return ???Ƶ?Switch????
*/
public Operation duplicate(Context ctx) {
Expression []newExps = dupExpressions(exps, ctx);
return new SwitchRemote(function, fkNames, datas, newExps, opt);
}
private void init(Context ctx) {
if (indexTables != null) {
return;
}
int count = datas.length;
codes = new Sequence[count];
cts = new ClusterMemoryTable[count];
indexTables = new IndexTable[count];
if (isLeft) {
dataStructs = new DataStruct[count];
keySeqs = new int[count];
}
for (int i = 0; i < count; ++i) {
if (datas[i] instanceof Sequence) {
codes[i] = (Sequence)datas[i];
Expression exp = null;
if (exps != null && exps.length > i) {
exp = exps[i];
}
if (exp == null || !(exp.getHome() instanceof CurrentSeq)) { // #
indexTables[i] = codes[i].getIndexTable(exp, ctx);
if (indexTables[i] == null) {
indexTables[i] = codes[i].newIndexTable(exp, ctx);
}
}
if (isLeft) {
dataStructs[i] = codes[i].dataStruct();
if (dataStructs[i] == null) {
MessageManager mm = EngineMessage.get();
throw new RQException(mm.getMessage("engine.needPurePmt"));
}
int key = -1;
if (exp != null) {
key = dataStructs[i].getFieldIndex(exp.getIdentifierName());
}
if (key == -1) {
int []pks = dataStructs[i].getPKIndex();
if (pks != null && pks.length == 1) {
key = pks[0];
}
}
if (key != -1) {
keySeqs[i] = key;
}
}
} else if (datas[i] instanceof ClusterMemoryTable) {
cts[i] = (ClusterMemoryTable)datas[i];
} else if (datas[i] != null) {
MessageManager mm = EngineMessage.get();
throw new RQException("switch" + mm.getMessage("function.paramTypeError"));
}
}
}
/**
* ?Դ?????????????ӣ????????ӽ??
* @param seq Ҫ??????????
* @param ctx ??????????
* @return ???ӽ??
*/
public Sequence process(Sequence data, Context ctx) {
init(ctx);
DataStruct ds = data.dataStruct();
if (ds == null) {
MessageManager mm = EngineMessage.get();
throw new RQException(mm.getMessage("engine.needPurePmt"));
}
int fkCount = fkNames.length;
if (fkIndex == null) {
fkIndex = new int[fkCount];
for (int f = 0; f < fkCount; ++f) {
fkIndex[f] = ds.getFieldIndex(fkNames[f]);
if (fkIndex[f] == -1) {
MessageManager mm = EngineMessage.get();
throw new RQException(fkNames[f] + mm.getMessage("ds.fieldNotExist"));
}
}
}
int srcLen = data.length();
Sequence result = new Sequence();
result.setMems(data.getMems());
for (int fk = 0; fk < fkCount; ++fk) {
int len = result.length();
int f = fkIndex[fk];
Sequence newSeq ;
if (cts[fk] != null) {
Sequence keyValues = result.fieldValues(f);
newSeq = cts[fk].getRows(keyValues, null, null, ctx);
} else if (indexTables[fk] != null) {
newSeq = fetch(result, f, indexTables[fk], ctx);
} else if (codes[fk] != null) {
newSeq = fetch(result, f, codes[fk], ctx);
} else {
// ָ???ֶα??ֵ
for (int i = 1; i <= len; ++i) {
Record r = (Record)result.getMem(i);
Object val = r.getNormalFieldValue(fk);
if (val instanceof Record) {
r.setNormalFieldValue(fk, ((Record)val).getPKValue());
}
}
continue;
}
if (isIsect) {
ObjectArray resultMems = new ObjectArray(len);
for (int i = 1; i <= len; ++i) {
Record r = (Record)result.getMem(i);
Object val = newSeq.getMem(i);
if (val != null) {
r.setNormalFieldValue(fk, val);
resultMems.add(r);
}
}
result.setMems(resultMems);
} else if (isDiff) {
ObjectArray resultMems = new ObjectArray(len);
for (int i = 1; i <= len; ++i) {
Record r = (Record)result.getMem(i);
Object val = newSeq.getMem(i);
// ?Ҳ???ʱ????Դֵ
if (val == null) {
resultMems.add(r);
}
}
result.setMems(resultMems);
} else {
for (int i = 1; i <= len; ++i) {
Record r = (Record)result.getMem(i);
Object val = newSeq.getMem(i);
r.setNormalFieldValue(f, val);
}
}
}
if (result.length() == 0) {
return null;
} else {
if (srcLen != result.length()) {
data.setMems(result.getMems());
}
return data;
}
}
// ??????ȡԪ??
private Sequence fetch(Sequence src, int f, IndexTable it, Context ctx) {
int len = src.length();
Sequence result = new Sequence(len);
if (isLeft) {
DataStruct ds = dataStructs[f];
int keySeq = keySeqs[f];
for (int i = 1; i <= len; ++i) {
BaseRecord r = (BaseRecord)src.getMem(i);
Object val = r.getNormalFieldValue(f);
val = it.find(val);
if (val != null) {
result.add(val);
} else {
Record record = new Record(ds);
record.setNormalFieldValue(keySeq, val);
result.add(record);
}
}
} else {
for (int i = 1; i <= len; ++i) {
BaseRecord r = (BaseRecord)src.getMem(i);
Object val = r.getNormalFieldValue(f);
val = it.find(val);
result.add(val);
}
}
return result;
}
// ?????ȡԪ??
private Sequence fetch(Sequence src, int f, Sequence code, Context ctx) {
int len = src.length();
Sequence result = new Sequence(len);
IArray codeMems = code.getMems();
int codeLen = codeMems.size();
for (int i = 1; i <= len; ++i) {
BaseRecord r = (BaseRecord)src.getMem(i);
Object val = r.getNormalFieldValue(f);
if (val instanceof Number) {
int seq = ((Number)val).intValue();
if (seq > 0 && seq <= codeLen) {
Object obj = codeMems.get(seq);
result.add(obj);
} else {
result.add(null);
}
} else {
result.add(null);
}
}
return result;
}
}