use of com.google.api.expr.v1alpha1.Expr.CreateStruct in project cel-java by projectnessie.
the class Unparser method visitStructMap.
void visitStructMap(CreateStruct expr) {
List<Entry> entries = expr.getEntriesList();
str.append("{");
for (int i = 0; i < entries.size(); i++) {
if (i > 0) {
str.append(", ");
}
Entry entry = entries.get(i);
Expr k = entry.getMapKey();
visit(k);
str.append(": ");
Expr v = entry.getValue();
visit(v);
}
str.append("}");
}
use of com.google.api.expr.v1alpha1.Expr.CreateStruct in project cel-java by projectnessie.
the class AstPruner method prune.
Expr prune(Expr node) {
if (node == null) {
return null;
}
Val val = value(node.getId());
if (val != null && !isUnknownOrError(val)) {
Expr newNode = maybeCreateLiteral(node.getId(), val);
if (newNode != null) {
return newNode;
}
}
switch(node.getExprKindCase()) {
case SELECT_EXPR:
Select select = node.getSelectExpr();
Expr operand = prune(select.getOperand());
if (operand != null && operand != select.getOperand()) {
return Expr.newBuilder().setId(node.getId()).setSelectExpr(Select.newBuilder().setOperand(operand).setField(select.getField()).setTestOnly(select.getTestOnly())).build();
}
break;
case CALL_EXPR:
Call call = node.getCallExpr();
Expr newExpr = maybePruneFunction(node);
if (newExpr != null) {
newExpr = prune(newExpr);
return newExpr;
}
boolean prunedCall = false;
List<Expr> args = call.getArgsList();
List<Expr> newArgs = new ArrayList<>(args.size());
for (int i = 0; i < args.size(); i++) {
Expr arg = args.get(i);
newArgs.add(arg);
Expr newArg = prune(arg);
if (newArg != null && newArg != arg) {
prunedCall = true;
newArgs.set(i, newArg);
}
}
Call newCall = Call.newBuilder().setFunction(call.getFunction()).setTarget(call.getTarget()).addAllArgs(newArgs).build();
Expr newTarget = prune(call.getTarget());
if (newTarget != null && newTarget != call.getTarget()) {
prunedCall = true;
newCall = Call.newBuilder().setFunction(call.getFunction()).setTarget(newTarget).addAllArgs(newArgs).build();
}
if (prunedCall) {
return Expr.newBuilder().setId(node.getId()).setCallExpr(newCall).build();
}
break;
case LIST_EXPR:
CreateList list = node.getListExpr();
List<Expr> elems = list.getElementsList();
List<Expr> newElems = new ArrayList<>(elems.size());
boolean prunedList = false;
for (int i = 0; i < elems.size(); i++) {
Expr elem = elems.get(i);
newElems.add(elem);
Expr newElem = prune(elem);
if (newElem != null && newElem != elem) {
newElems.set(i, newElem);
prunedList = true;
}
}
if (prunedList) {
return Expr.newBuilder().setId(node.getId()).setListExpr(CreateList.newBuilder().addAllElements(newElems)).build();
}
break;
case STRUCT_EXPR:
boolean prunedStruct = false;
CreateStruct struct = node.getStructExpr();
List<Entry> entries = struct.getEntriesList();
String messageType = struct.getMessageName();
List<Entry> newEntries = new ArrayList<>(entries.size());
for (int i = 0; i < entries.size(); i++) {
Entry entry = entries.get(i);
newEntries.add(entry);
Expr mapKey = entry.getMapKey();
Expr newKey = mapKey != Entry.getDefaultInstance().getMapKey() ? prune(mapKey) : null;
Expr newValue = prune(entry.getValue());
if ((newKey == null || newKey == mapKey) && (newValue == null || newValue == entry.getValue())) {
continue;
}
prunedStruct = true;
Entry newEntry;
if (!messageType.isEmpty()) {
newEntry = Entry.newBuilder().setFieldKey(entry.getFieldKey()).setValue(newValue).build();
} else {
newEntry = Entry.newBuilder().setMapKey(newKey).setValue(newValue).build();
}
newEntries.set(i, newEntry);
}
if (prunedStruct) {
return Expr.newBuilder().setId(node.getId()).setStructExpr(CreateStruct.newBuilder().setMessageName(messageType).addAllEntries(entries)).build();
}
break;
case COMPREHENSION_EXPR:
Comprehension compre = node.getComprehensionExpr();
// Only the range of the comprehension is pruned since the state tracking only records
// the last iteration of the comprehension and not each step in the evaluation which
// means that the any residuals computed in between might be inaccurate.
Expr newRange = prune(compre.getIterRange());
if (newRange != null && newRange != compre.getIterRange()) {
return Expr.newBuilder().setId(node.getId()).setComprehensionExpr(Comprehension.newBuilder().setIterVar(compre.getIterVar()).setIterRange(newRange).setAccuVar(compre.getAccuVar()).setAccuInit(compre.getAccuInit()).setLoopCondition(compre.getLoopCondition()).setLoopStep(compre.getLoopStep()).setResult(compre.getResult())).build();
}
}
// allocation cost at another point. So go with the simple approach - at least for now.
return node;
}
use of com.google.api.expr.v1alpha1.Expr.CreateStruct in project cel-java by projectnessie.
the class Unparser method visitStructMsg.
void visitStructMsg(CreateStruct expr) {
List<Entry> entries = expr.getEntriesList();
str.append(expr.getMessageName());
str.append("{");
for (int i = 0; i < entries.size(); i++) {
if (i > 0) {
str.append(", ");
}
Entry entry = entries.get(i);
String f = entry.getFieldKey();
str.append(f);
str.append(": ");
Expr v = entry.getValue();
visit(v);
}
str.append("}");
}
Aggregations