~slub.team/goobi-indexserver/3.x

public static Comparator<Doc> createComparator(final String field, final boolean asc, final boolean sortMissingLast, final boolean sortMissingFirst, final boolean sortMissingAsZero) {

970

final int mul = asc ? 1 : -1;

971

972

if (field.equals("_docid_")) {

973

return new Comparator<Doc>() {

974

public int compare(Doc o1, Doc o2) {

975

return (o1.order - o2.order) * mul;

976

}

977

};

978

}

979

980

if (field.equals("score")) {

981

return createComparator("score_f", asc, sortMissingLast, sortMissingFirst, sortMissingAsZero);

982

}

983

984

return new Comparator<Doc>() {

985

private Comparable zeroVal(Comparable template) {

986

if (template == null) return null;

987

if (template instanceof String) return null; // fast-path for string

988

if (template instanceof Integer) return 0;

989

if (template instanceof Long) return (long)0;

990

if (template instanceof Float) return (float)0;

991

if (template instanceof Double) return (double)0;

992

if (template instanceof Short) return (short)0;

993

if (template instanceof Byte) return (byte)0;

994

if (template instanceof Character) return (char)0;

995

return null;

996

}

997

998

public int compare(Doc o1, Doc o2) {

999

Comparable v1 = o1.getFirstValue(field);

1000

Comparable v2 = o2.getFirstValue(field);

1001

1002

v1 = v1 == null ? zeroVal(v2) : v1;

1003

v2 = v2 == null ? zeroVal(v1) : v2;

1004

1005

int c = 0;

1006

if (v1 == v2) {

1007

c = 0;

1008

} else if (v1 == null) {

1009

if (sortMissingLast) c = mul;

1010

else if (sortMissingFirst) c = -mul;

1011

else c = -1;

1012

} else if (v2 == null) {

1013

if (sortMissingLast) c = -mul;

1014

else if (sortMissingFirst) c = mul;

1015

else c = 1;

1016

} else {

1017

c = v1.compareTo(v2);

1018

}

1019

1020

c = c * mul;

1021

1022

return c;

1023

}

1024

};

1025

}

1026

1027

public static Comparator<Doc> createComparator(final List<Comparator<Doc>> comparators) {

1028

return new Comparator<Doc>() {

1029

public int compare(Doc o1, Doc o2) {

1030

int c = 0;

1031

for (Comparator<Doc> comparator : comparators) {

1032

c = comparator.compare(o1, o2);

1033

if (c!=0) return c;

1034

}

1035

return o1.order - o2.order;

1036

}

1037

};

1038

}

1039

1040

1041

public static String toJSON(Doc doc) {

1042

CharArr out = new CharArr();

1043

try {

1044

out.append("{\"add\":{\"doc\":{");

1045

boolean firstField = true;

1046

for (Fld fld : doc.fields) {

1047

if (firstField) firstField=false;

1048

else out.append(',');

1049

JSONUtil.writeString(fld.ftype.fname, 0, fld.ftype.fname.length(), out);

1050

out.append(':');

1051

if (fld.vals.size() > 1) {

1052

out.append('[');

1053

}

1054

boolean firstVal = true;

1055

for (Comparable val : fld.vals) {

1056

if (firstVal) firstVal=false;

1057

else out.append(',');

1058

out.append(JSONUtil.toJSON(val));

1059

}

1060

if (fld.vals.size() > 1) {

1061

out.append(']');

1062

}

1063

}

1064

out.append("}}}");

1065

} catch (IOException e) {

1066

// should never happen

1067

}

1068

return out.toString();

1069

}

1070

1071

/** Return a Map from field value to a list of document ids */

1072

Map<Comparable, List<Comparable>> invertField(Map<Comparable, Doc> model, String field) {

1073

Map<Comparable, List<Comparable>> value_to_id = new HashMap<Comparable, List<Comparable>>();

1074

1075

// invert field

1076

for (Comparable key : model.keySet()) {

1077

Doc doc = model.get(key);

1078

List<Comparable> vals = doc.getValues(field);

1079

if (vals == null) continue;

1080

for (Comparable val : vals) {

1081

List<Comparable> ids = value_to_id.get(val);

1082

if (ids == null) {

1083

ids = new ArrayList<Comparable>(2);

1084

value_to_id.put(val, ids);

1085

}

1086

ids.add(key);

1087

}

1088

}

1089

1090

return value_to_id;

1091

}

1092

1093

1094

/** Gets a resource from the context classloader as {@link File}. This method should only be used,

1095

* if a real file is needed. To get a stream, code should prefer

1096

* {@link Class#getResourceAsStream} using {@code this.getClass()}.

1097

1098

public static File getFile(String name) {

1099

try {

1100

File file = new File(name);

1101

if (!file.exists()) {

1102

file = new File(Thread.currentThread().getContextClassLoader().getResource(name).toURI());

1103

}

1104

return file;

1105

} catch (Exception e) {

1106

/* more friendly than NPE */

1107

throw new RuntimeException("Cannot find resource: " + name);

1108

}

1109

}

1110

1111

public static String TEST_HOME() {

1112

return getFile("solr/conf").getParent();

1113

}

1114

1115

public static Throwable getRootCause(Throwable t) {

1116

Throwable result = t;

1117

for (Throwable cause = t; null != cause; cause = cause.getCause()) {

1118

result = cause;

1119

}

1120

return result;

1121

}

1122

}

Older »