ShortVisitor.java

/**
 * Copyright (C) 2010 Michael Mosmann <michael@mosmann.de>
 *
 * 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 de.flapdoodle.mongoom.mapping.types.truncations;

import java.util.Set;

import com.google.common.collect.Sets;

import de.flapdoodle.mongoom.mapping.ITransformation;
import de.flapdoodle.mongoom.mapping.properties.PropertyReference;
import de.flapdoodle.mongoom.mapping.types.AbstractPrimitiveTransformation;


public class ShortVisitor extends AbstractSmallerTypeVisitor<Short,Integer>  {

	
	public ShortVisitor() {
		super(Integer.class, short.class,Short.class);
	}
	
	@Override
	protected ITransformation<Short, Integer> newTransformation() {
		return new ShortTransformation();
	}

	static class ShortTransformation extends AbstractPrimitiveTransformation<Short, Integer> {

		
		public ShortTransformation() {
			super(Short.class);
		}
		
		@Override
		public Integer asObject(Short value) {
			return value!=null ? value.intValue() : null;
		}

		@Override
		public Short asEntity(Integer object) {
			return object!=null ? object.shortValue() : null;
		}
		
	}
}