Making basic inserts with mapping work.

This commit is contained in:
Michael Nitschinger
2012-12-31 13:51:49 +01:00
parent 92c56ac8f9
commit a7f8eb5fbe
12 changed files with 344 additions and 18 deletions

View File

@@ -24,4 +24,15 @@ package com.couchbase.spring.core;
public interface CouchbaseOperations {
/**
* Insert the object into the connected bucket.
*
* <p>
* The object is converted to a JSON representation using an instance of
* {@link CouchbaseConverter}.
* </p>
*
* @param objectToSave the object to store in the bucket.
*/
void insert(Object objectToSave);
}

View File

@@ -25,6 +25,7 @@ package com.couchbase.spring.core;
import com.couchbase.client.CouchbaseClient;
import com.couchbase.spring.core.convert.CouchbaseConverter;
import com.couchbase.spring.core.convert.MappingCouchbaseConverter;
import com.couchbase.spring.core.mapping.ConvertedCouchbaseDocument;
public class CouchbaseTemplate implements CouchbaseOperations {
@@ -47,5 +48,13 @@ public class CouchbaseTemplate implements CouchbaseOperations {
return converter;
}
@Override
public void insert(Object objectToSave) {
ConvertedCouchbaseDocument converted = new ConvertedCouchbaseDocument();
couchbaseConverter.write(objectToSave, converted);
client.set(converted.getId(), converted.getExpiry(), converted.getValue());
}
}

View File

@@ -22,6 +22,7 @@
package com.couchbase.spring.core.convert;
import com.couchbase.spring.core.mapping.ConvertedCouchbaseDocument;
import com.couchbase.spring.core.mapping.CouchbasePersistentEntity;
import com.couchbase.spring.core.mapping.CouchbasePersistentProperty;
import org.springframework.data.convert.EntityConverter;
@@ -29,8 +30,7 @@ import org.springframework.data.convert.EntityReader;
public interface CouchbaseConverter extends
EntityConverter<CouchbasePersistentEntity<?>,
CouchbasePersistentProperty, Object, Object>,
CouchbaseWriter<Object>,
EntityReader<Object, Object> {
CouchbasePersistentProperty, Object, ConvertedCouchbaseDocument>,
CouchbaseWriter<Object, ConvertedCouchbaseDocument>,
EntityReader<Object, ConvertedCouchbaseDocument> {
}

View File

@@ -24,6 +24,7 @@ package com.couchbase.spring.core.convert;
import org.springframework.data.convert.EntityWriter;
public interface CouchbaseWriter<T> extends EntityWriter<T, Object> {
public interface CouchbaseWriter<T, ConvertedCouchbaseDocument> extends
EntityWriter<T, ConvertedCouchbaseDocument> {
}

View File

@@ -22,16 +22,29 @@
package com.couchbase.spring.core.convert;
import com.couchbase.spring.core.CouchbaseFactory;
import com.couchbase.spring.core.mapping.ConvertedCouchbaseDocument;
import com.couchbase.spring.core.mapping.CouchbasePersistentEntity;
import com.couchbase.spring.core.mapping.CouchbasePersistentProperty;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.logging.Level;
import java.util.logging.Logger;
import org.codehaus.jackson.JsonEncoding;
import org.codehaus.jackson.JsonFactory;
import org.codehaus.jackson.JsonGenerationException;
import org.codehaus.jackson.JsonGenerator;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.core.convert.support.ConversionServiceFactory;
import org.springframework.data.mapping.context.MappingContext;
import org.springframework.data.mapping.model.BeanWrapper;
import org.springframework.data.mapping.model.MappingException;
import org.springframework.data.util.ClassTypeInformation;
import org.springframework.data.util.TypeInformation;
import org.springframework.core.convert.ConversionService;
import org.springframework.data.mapping.PropertyHandler;
public class MappingCouchbaseConverter extends AbstractCouchbaseConverter
implements ApplicationContextAware {
@@ -55,22 +68,79 @@ public class MappingCouchbaseConverter extends AbstractCouchbaseConverter
}
@Override
public <R> R read(Class<R> type, Object s) {
public <R> R read(Class<R> type, ConvertedCouchbaseDocument s) {
throw new UnsupportedOperationException("Not supported yet.");
}
@Override
public void write(Object source, Object target) {
public void write(Object source, ConvertedCouchbaseDocument target) {
if(source == null) {
return;
}
TypeInformation<? extends Object> type = ClassTypeInformation.from(source.getClass());
writeInternal(source, target, type);
try {
writeInternal(source, target, type);
} catch (IOException ex) {
throw new MappingException("Could not translate to JSON while converting "
+ source.getClass().getName());
}
}
protected void writeInternal(Object source, Object target, TypeInformation<?> type) {
System.out.println(type.getActualType());
protected void writeInternal(final Object source,
ConvertedCouchbaseDocument target, TypeInformation<?> type)
throws IOException {
CouchbasePersistentEntity<?> entity = mappingContext.getPersistentEntity(
source.getClass());
if(entity == null) {
throw new MappingException("No mapping metadata found for entity of type "
+ source.getClass().getName());
}
final CouchbasePersistentProperty idProperty = entity.getIdProperty();
if(idProperty == null) {
throw new MappingException("ID property required for entity of type "
+ source.getClass().getName());
}
final BeanWrapper<CouchbasePersistentEntity<Object>, Object> wrapper =
BeanWrapper.create(source, conversionService);
String id = wrapper.getProperty(idProperty, String.class, false);
target.setId(id);
JsonFactory jsonFactory = new JsonFactory();
OutputStream jsonStream = new ByteArrayOutputStream();
final JsonGenerator jsonGenerator = jsonFactory.createJsonGenerator(
jsonStream, JsonEncoding.UTF8);
jsonGenerator.writeStartObject();
entity.doWithProperties(new PropertyHandler<CouchbasePersistentProperty>() {
@Override
public void doWithPersistentProperty(CouchbasePersistentProperty prop) {
if(prop.equals(idProperty)) {
return;
}
Object propertyValue = wrapper.getProperty(prop, prop.getType(), false);
if(propertyValue != null) {
try {
jsonGenerator.writeFieldName(prop.getFieldName());
jsonGenerator.writeObject(propertyValue);
} catch (IOException ex) {
throw new MappingException("Could not translate to JSON while converting "
+ source.getClass().getName());
}
}
}
});
jsonGenerator.writeEndObject();
jsonGenerator.close();
target.setValue(jsonStream.toString());
}
@Override

View File

@@ -20,12 +20,52 @@
* IN THE SOFTWARE.
*/
package com.couchbase.spring.core;
package com.couchbase.spring.core.mapping;
import com.couchbase.client.CouchbaseClient;
public class ConvertedCouchbaseDocument {
public interface CouchbaseFactory {
private String id;
CouchbaseClient getDb();
private String value;
private int expiry;
public ConvertedCouchbaseDocument() {
this("", "", 0);
}
public ConvertedCouchbaseDocument(String id, String value) {
this(id, value, 0);
}
public ConvertedCouchbaseDocument(String id, String value, int expiry) {
this.id = id;
this.value = value;
this.expiry = expiry;
}
public void setId(String id) {
this.id = id;
}
public String getId() {
return id;
}
public String getValue() {
return value;
}
public void setValue(String value) {
this.value = value;
}
public int getExpiry() {
return expiry;
}
public void setExpiry(int expiry) {
this.expiry = expiry;
}
}