PropertyLocationHeaderStrategy.java

package net.eusashead.hateoas.header.impl;

/*
 * #[license]
 * spring-responseentitybuilder
 * %%
 * Copyright (C) 2013 Eusa's Head
 * %%
 * 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.
 * %[license]
 */

import net.eusashead.hateoas.header.LocationHeaderStrategy;

public class PropertyLocationHeaderStrategy<V> implements LocationHeaderStrategy<V> {
	
	private final String uriTemplate;
	private final String[] identity;
	
	public PropertyLocationHeaderStrategy(String uriTemplate, String... identity) {
		this.uriTemplate = uriTemplate;
		this.identity = identity;
	}

	@Override
	public LocationHeaderImpl extract(V target) {
		
		// Get the identity values from the object
		Object[] vals = new Object[identity.length];
		int i = 0;
		for (String prop : this.identity) {
			vals[i++] = PropertyUtil.getValue(target, prop);
		}
		
		return new LocationHeaderImpl(uriTemplate, vals);
	}

}