coffeepot.bean.wr.writer.AbstractWriter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of coffeepot-bean-wr Show documentation
Show all versions of coffeepot-bean-wr Show documentation
Coffeepot-bean-wr is a simple Java framework for marshalling Java beans to flat file and unmarshal flat file to Java Beans
/*
* Copyright 2015 Jeandeson O. Merelis.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package coffeepot.bean.wr.writer;
/*
* #%L
* coffeepot-bean-wr
* %%
* Copyright (C) 2013 - 2015 Jeandeson O. Merelis
* %%
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* #L%
*/
import coffeepot.bean.wr.mapper.Callback;
import coffeepot.bean.wr.mapper.FieldConditionModel;
import coffeepot.bean.wr.mapper.FieldModel;
import coffeepot.bean.wr.mapper.Metadata;
import coffeepot.bean.wr.mapper.ObjectMapper;
import coffeepot.bean.wr.mapper.ObjectMapperFactory;
import coffeepot.bean.wr.mapper.RecordModel;
import coffeepot.bean.wr.mapper.UnresolvedObjectMapperException;
import coffeepot.bean.wr.types.Align;
import lombok.Getter;
import lombok.Setter;
import java.io.IOException;
import java.io.Writer;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.Collection;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
/**
*
* @author Jeandeson O. Merelis
*/
public abstract class AbstractWriter implements ObjectWriter {
@Getter @Setter protected Writer writer;
@Getter @Setter protected int autoFlush = 0;
protected int recordCount = 0;
@Getter @Setter protected Callback callback;
@Getter protected int version = 0;
@Override public void setVersion(int version) {
this.version = version;
this.metadata.__setVersion(version);
}
protected final Metadata metadata = new Metadata(0);
protected abstract void writeRecord(List values) throws IOException;
@Override
public abstract ObjectMapperFactory getObjectMapperFactory();
@Override
public void clearMappers() {
getObjectMapperFactory().getMappers().clear();
}
@Override
public void createMapper(Class clazz) throws UnresolvedObjectMapperException, NoSuchFieldException, Exception {
getObjectMapperFactory().create(clazz, null, callback);
}
@Override
public void createMapper(Class clazz, String recordGroupId) throws UnresolvedObjectMapperException, NoSuchFieldException, Exception {
getObjectMapperFactory().create(clazz, recordGroupId, callback);
}
@Override
public void flush() throws IOException {
if (writer != null) {
writer.flush();
}
}
@Override
public void close() throws IOException {
if (writer != null) {
writer.close();
}
}
@Override
public void write(Object obj) throws IOException {
write(obj, null);
}
private ObjectMapper getObjectMapper(Object obj, String recordGroupId) {
ObjectMapper op = getObjectMapperFactory().getMappers().get(obj.getClass());
if (op == null) {
try {
createMapper(obj.getClass(), recordGroupId);
} catch (Exception ex) {
throw new RuntimeException(ex);
}
op = getObjectMapperFactory().getMappers().get(obj.getClass());
if (op == null) {
throw new RuntimeException("Mapper for class has not been set.");
}
}
return op;
}
@Override
public void write(Object obj, String recordGroupId) throws IOException {
ObjectMapper op;
if (obj instanceof Collection) {
Collection c = (Collection) obj;
if (c.isEmpty()) {
return;
}
Iterator it = c.iterator();
Object o = it.next();
op = getObjectMapper(o, recordGroupId);
marshal(o, op);
while (it.hasNext()) {
marshal(it.next(), op);
}
return;
}
op = getObjectMapper(obj, recordGroupId);
marshal(obj, op);
}
private void marshal(Object obj, ObjectMapper op) throws IOException {
List fieldsValue = marshal(obj, null, op);
if (fieldsValue != null && !fieldsValue.isEmpty()) {
writeRecord(fieldsValue);
}
}
private List marshal(Object obj, List fieldsValue, ObjectMapper op) throws IOException {
if (obj == null) {
return fieldsValue;
}
return marshal(obj, fieldsValue, op.getFields(), op.getRootClass(), op);
}
private List marshal(Object obj, List fieldsValue, List fields, Class clazz, ObjectMapper op) throws IOException {
for (FieldModel f : fields) {
if (version < f.getMinVersion() || version > f.getMaxVersion()) {
continue;
}
if (f.isIgnoreOnWrite()) {
continue;
}
FieldConditionModel condition = f.getWriteAsNull();
if (condition != null && condition.isActive()) {
if (condition.isAlways() || (version >= condition.getMinVersion() && version <= condition.getMaxVersion())) {
if (fieldsValue == null) {
fieldsValue = new LinkedList<>();
}
String s = process(null, f);
fieldsValue.add(s);
continue;
}
}
condition = f.getConditionForWriteAs();
if (condition != null && condition.isActive()) {
if (condition.isAlways() || (version >= condition.getMinVersion() && version <= condition.getMaxVersion())) {
if (fieldsValue == null) {
fieldsValue = new LinkedList<>();
}
String s = process(f.getWriteAs(), f);
fieldsValue.add(s);
continue;
}
}
if (f.getConstantValue() != null && !"".equals(f.getConstantValue())) {
if (fieldsValue == null) {
fieldsValue = new LinkedList<>();
}
String s = process(f.getConstantValue(), f);
fieldsValue.add(s);
continue;
}
fieldsValue = marshalField(obj, fieldsValue, clazz, f, op);
}
return fieldsValue;
}
private List marshalField(final Object obj, List fieldsValue, Class clazz, final FieldModel f, ObjectMapper op) throws IOException {
try {
Object o = null;
if (obj != null) {
final java.lang.reflect.Field declaredField;
if (f.getGetterMethod() != null) {
o = AccessController.doPrivileged(new PrivilegedAction() {
@Override
public Object run() {
boolean wasAccessible = f.getGetterMethod().isAccessible();
try {
f.getGetterMethod().setAccessible(true);
return f.getGetterMethod().invoke(obj);
} catch (Exception ex) {
throw new IllegalStateException("Cannot invoke method get", ex);
} finally {
f.getGetterMethod().setAccessible(wasAccessible);
}
}
});
} else {
declaredField = clazz.getDeclaredField(f.getName());
o = AccessController.doPrivileged(new PrivilegedAction() {
@Override
public Object run() {
boolean wasAccessible = declaredField.isAccessible();
try {
declaredField.setAccessible(true);
return declaredField.get(obj);
} catch (Exception ex) {
throw new IllegalStateException("Cannot invoke method get", ex);
} finally {
declaredField.setAccessible(wasAccessible);
}
}
});
}
if (f.isCollection()) {
writeRecord(fieldsValue);
fieldsValue = null;
fieldsValue = marshalCollection(o, fieldsValue, f, op);
writeRecord(fieldsValue);
fieldsValue = null;
return fieldsValue;
}
if (f.getTypeHandler() == null) {
if (f.isNestedObject()) {
writeRecord(fieldsValue);
fieldsValue = null;
}
ObjectMapper parser = getObjectMapperFactory().getMappers().get(f.getClassType());
if (parser != null) {
fieldsValue = marshal(o, fieldsValue, parser);
if (f.isNestedObject()) {
writeRecord(fieldsValue);
fieldsValue = null;
}
} else {
throw new RuntimeException("Parser not found for class: " + f.getClassType().getName());
}
return fieldsValue;
}
}
if (f.isNestedObject()) {
writeRecord(fieldsValue);
fieldsValue = null;
}
String s = process(f.getTypeHandler().toString(o, metadata.__setVersion(version).__setFieldModel(f)), f);
if (fieldsValue == null) {
fieldsValue = new LinkedList<>();
}
fieldsValue.add(s);
//FIXME: Exceptions
} catch (IllegalArgumentException ex) {
Logger.getLogger(ObjectMapper.class.getName()).log(Level.SEVERE, null, ex);
} catch (NoSuchFieldException ex) {
Logger.getLogger(ObjectMapper.class.getName()).log(Level.SEVERE, null, ex);
} catch (SecurityException ex) {
Logger.getLogger(ObjectMapper.class.getName()).log(Level.SEVERE, null, ex);
}
return fieldsValue;
}
private List marshalCollection(Object obj, List fieldsValue, FieldModel field, ObjectMapper op) throws IOException {
if (obj == null) {
return fieldsValue;
}
if (!Collection.class.isAssignableFrom(obj.getClass())) {
return fieldsValue;
}
ObjectMapper parser = getObjectMapperFactory().getMappers().get(field.getClassType());
if (parser != null) {
Collection c = (Collection) obj;
Iterator it = c.iterator();
while (it.hasNext()) {
Class cl;
List fi;
fi = parser.getFields();
cl = parser.getRootClass();
fieldsValue = marshal(it.next(), fieldsValue, fi, cl, parser);
writeRecord(fieldsValue);
fieldsValue = null;
}
}
return fieldsValue;
}
private String process(String s, FieldModel field) {
if (s == null && !field.isPaddingIfNullOrEmpty()) {
return "";
}
if (s == null) {
s = "";
}
if (field.isTrim()) {
s = s.trim();
}
if (s.isEmpty() && !field.isPaddingIfNullOrEmpty()) {
return s;
}
if (field.getMaxLength() > 0 && s.length() > field.getMaxLength()) {
if (field.getAlign().equals(Align.LEFT)) {
s = s.substring(0, field.getMaxLength());
} else {
s = s.substring(s.length() - field.getMaxLength(), s.length());
}
}
if (field.getMinLength() > 0) {
if (s.length() < field.getMinLength()) {
StringBuilder sb = new StringBuilder();
int i = field.getMinLength() - s.length();
if (field.getAlign().equals(Align.LEFT)) {
sb.append(s);
}
for (int j = 0; j < i; j++) {
sb.append(field.getPadding());
}
if (!field.getAlign().equals(Align.LEFT)) {
sb.append(s);
}
s = sb.toString();
}
}
return s;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy