LunchRestaurantGraderImpl.java

/*
 * Copyright 2014 Avanza Bank AB
 *
 * 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 lunch.grader.pu;

import lunch.api.LunchRestaurant;
import lunch.api.LunchService;
import lunch.grader.api.LunchRestaurantGrade;
import lunch.grader.api.LunchRestaurantGrader;

import org.openspaces.core.GigaSpace;
import org.springframework.beans.factory.annotation.Autowired;

import com.avanza.astrix.provider.core.AstrixServiceExport;

@AstrixServiceExport(LunchRestaurantGrader.class)
public class LunchRestaurantGraderImpl implements LunchRestaurantGrader {
	
	private final LunchService lunchService;
	private final GigaSpace gigaSpace;

	@Autowired
	public LunchRestaurantGraderImpl(LunchService lunchService, GigaSpace gigaSpace) {
		this.lunchService = lunchService;
		this.gigaSpace = gigaSpace;
	}

	@Override
	public void grade(String restaurantName, int grade) {
		LunchRestaurant lunchRestaurant = lunchService.getLunchRestaurant(restaurantName);
		if (lunchRestaurant == null) {
			throw new IllegalArgumentException("Unknown lunch restaurant: " + restaurantName);
		}
		LunchRestaurantGrade currentGrade = gigaSpace.readById(LunchRestaurantGrade.class, restaurantName);
		if (currentGrade == null) {
			currentGrade = new LunchRestaurantGrade();
			currentGrade.setRestaurantName(restaurantName);
		}
		currentGrade.addGrade(grade);
		gigaSpace.write(currentGrade);
	}

	@Override
	public Double getAvarageGrade(String restaurantName) {
		LunchRestaurantGrade currentGrade = gigaSpace.readById(LunchRestaurantGrade.class, restaurantName);
		if (currentGrade == null) {
			return null;
		}
		return currentGrade.avarageGrade();
	}

	@Override
	public LunchRestaurantGrade getHighestGrade() {
		LunchRestaurantGrade[] allGradedRestaurants = gigaSpace.readMultiple(new LunchRestaurantGrade());
		if (allGradedRestaurants.length == 0) {
			return null;
		}
		LunchRestaurantGrade result = null;
		for (LunchRestaurantGrade g : allGradedRestaurants) {
			if (result == null || g.avarageGrade() > result.avarageGrade()) {
				result = g;
			}
		}
		return result;
	}
	
	
	

}