/*
* Copyright (c) www.bugull.com
*
* 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 net.tooan.ynpay.third.mongodb.encoder;
import com.mongodb.DBObject;
import net.tooan.ynpay.third.mongodb.annotations.Default;
import net.tooan.ynpay.third.mongodb.annotations.EmbedList;
import net.tooan.ynpay.third.mongodb.mapper.DataType;
import net.tooan.ynpay.third.mongodb.mapper.MapperUtil;
import java.lang.reflect.Array;
import java.lang.reflect.Field;
import java.util.*;
/**
* @author Frank Wen(xbwen@hotmail.com)
*/
@SuppressWarnings("unchecked")
public class EmbedListEncoder extends AbstractEncoder {
public EmbedListEncoder(Object obj, Field field) {
super(obj, field);
}
@Override
public String getFieldName() {
String fieldName = field.getName();
EmbedList embedList = field.getAnnotation(EmbedList.class);
String name = embedList.name();
if (!name.equals(Default.NAME)) {
fieldName = name;
}
return fieldName;
}
@Override
public Object encode() {
Class<?> type = field.getType();
if (type.isArray()) {
return encodeArray();
} else {
return encodeList(type);
}
}
private Object encodeArray() {
int len = Array.getLength(value);
List<DBObject> result = new ArrayList<DBObject>();
for (int i = 0; i < len; i++) {
Object o = Array.get(value, i);
if (o != null) {
result.add(MapperUtil.toDBObject(o));
}
}
return result;
}
private Object encodeList(Class type) {
if (DataType.isList(type)) {
List list = (ArrayList) value;
List<DBObject> result = new ArrayList<DBObject>();
for (Object o : list) {
if (o != null) {
result.add(MapperUtil.toDBObject(o));
}
}
return result;
} else if (DataType.isSet(type)) {
Set set = (Set) value;
Set<DBObject> result = new HashSet<DBObject>();
for (Object o : set) {
if (o != null) {
result.add(MapperUtil.toDBObject(o));
}
}
return result;
} else if (DataType.isMap(type)) {
Map map = (Map) value;
Map result = new HashMap();
for (Object key : map.keySet()) {
Object o = map.get(key);
if (o != null) {
result.put(key, MapperUtil.toDBObject(o));
} else {
result.put(key, null);
}
}
return result;
} else {
return null;
}
}
}