getrebuild/rebuild
1
1
Fork 0
mirror of https://github.com/getrebuild/rebuild.git synced 2024-11-10 08:55:31 +08:00
Code Issues Projects Releases Packages Wiki Activity

fix: TextFunction

Browse source
This commit is contained in:
RB 2023-11-23 16:57:08 +08:00
parent 55d2efebfb
commit b59572a96f
1 changed files with 60 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

73
src/main/java/com/rebuild/core/service/trigger/aviator/TextFunction.java
View file

@ -7,22 +7,30 @@ See LICENSE and COMMERCIAL in the project root for license information.
package com.rebuild.core.service.trigger.aviator;
import cn.devezhao.persist4j.Entity;
import cn.devezhao.persist4j.Field;
import cn.devezhao.persist4j.engine.ID;
import com.googlecode.aviator.runtime.function.AbstractFunction;
import com.googlecode.aviator.runtime.type.AviatorJavaType;
import com.googlecode.aviator.runtime.type.AviatorNil;
import com.googlecode.aviator.runtime.type.AviatorObject;
import com.googlecode.aviator.runtime.type.AviatorString;
import com.rebuild.core.Application;
import com.rebuild.core.metadata.MetadataHelper;
import com.rebuild.core.support.general.FieldValueHelper;
import com.rebuild.utils.CommonsUtils;
import com.rebuild.web.commons.MetadataGetting;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Map;
/**
* Usage: TEXT($id|$id[], [$defaultValue], [$separator])
* Usage: TEXT($id|$id[], [$defaultValue], [$separator], [$fieldName])
* Return: String
*
* @author RB
@ -37,37 +45,62 @@ public class TextFunction extends AbstractFunction {
@Override
public AviatorObject call(Map<String, Object> env, AviatorObject arg1) {
return call(env, arg1, BLANK, SEPARATOR);
return call(env, arg1, BLANK, SEPARATOR, AviatorNil.NIL);
}
@Override
public AviatorObject call(Map<String, Object> env, AviatorObject arg1, AviatorObject arg2) {
return call(env, arg1, BLANK, SEPARATOR);
return call(env, arg1, arg2, SEPARATOR, AviatorNil.NIL);
}
@Override
public AviatorObject call(Map<String, Object> env, AviatorObject arg1, AviatorObject arg2, AviatorObject arg3) {
return call(env, arg1, arg2, arg3, AviatorNil.NIL);
}
@Override
public AviatorObject call(Map<String, Object> env, AviatorObject arg1, AviatorObject arg2, AviatorObject arg3, AviatorObject arg4) {
final Object $id = arg1.getValue(env);
final Object $defaultValue = arg2.getValue(env);
final String sep = ObjectUtils.defaultIfNull(arg3.getValue(env), ", ").toString();
final String fieldName = arg4.getValue(env) == null ? null : arg4.getValue(env).toString();
// 引用 ID
if (ID.isId($id)) {
ID anyid = $id instanceof ID ? (ID) $id : ID.valueOf($id.toString());
String text = FieldValueHelper.getLabel(anyid, null);
return text == null ? arg2 : new AviatorString(text);
String text = getLabel(anyid, fieldName);
if (text == null && $defaultValue != null) text = $defaultValue.toString();
return new AviatorString(text);
}
// 多引用 ID[]
if ($id instanceof ID[]) {
List<String> texts = new ArrayList<>();
for (ID anyid : (ID[]) $id) {
String t = FieldValueHelper.getLabel(anyid, null);
if (t != null) texts.add(t);
Object idArray = $id;
if ($id instanceof Collection) {
List<ID> list = null;
for (Object o : (Collection<?>) $id) {
if (o instanceof ID) {
if (list == null) list = new ArrayList<>();
list.add((ID) o);
}
}
if (texts.isEmpty()) return arg2;
if (list != null) idArray = list.toArray(new ID[0]);
}
String sep = ObjectUtils.defaultIfNull(arg3.getValue(env), ", ").toString();
return new AviatorString(StringUtils.join(texts, sep));
if (idArray instanceof ID[]) {
List<String> text = new ArrayList<>();
for (ID anyid : (ID[]) idArray) {
String item = getLabel(anyid, fieldName);
if (item == null && $defaultValue != null) item = $defaultValue.toString();
if (item != null) text.add(item);
}
if (text.isEmpty()) return arg2;
return new AviatorString(StringUtils.join(text, sep));
}
if (arg1 instanceof AviatorJavaType) {
@ -81,6 +114,20 @@ public class TextFunction extends AbstractFunction {
return arg2;
}
// 获取字段内容
private String getLabel(ID id, String fieldName) {
if (fieldName == null) return FieldValueHelper.getLabelNotry(id);
Entity entity = MetadataHelper.getEntity(id.getEntityCode());
Field field = MetadataHelper.getLastJoinField(entity, fieldName);
Object[] o = Application.getQueryFactory().uniqueNoFilter(id, fieldName);
if (o == null || o[0] == null) return null;
Object label = FieldValueHelper.wrapFieldValue(o[0], field, true);
return label == null ? null : label.toString();
}
@Override
public String getName() {
return "TEXT";