package example.u2ware.springfield.part4.step2;

import javax.persistence.Entity;
import javax.persistence.Id;
import javax.validation.constraints.NotNull;

import lombok.Getter;
import lombok.Setter;

import com.u2ware.springfield.config.Springfield;
import com.u2ware.springfield.config.Springfield.Strategy;

@Springfield(
        strategy=Strategy.JPA,
        topLevelMapping="/part4/setter"
)
@Entity
public class SetterBean {

        @Id
        @NotNull
        private @Getter @Setter Integer code;   
        @NotNull
        private @Getter @Setter String name;
}

package example.u2ware.springfield.part4.step2;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import com.u2ware.springfield.repository.EntityRepository;
import com.u2ware.springfield.service.EntityServiceImpl;
import com.u2ware.springfield.support.context.ContextBroker;

@Service
public class SetterBeanService extends EntityServiceImpl<SetterBean, SetterBean>{

        @Autowired @Qualifier("sessionContextBroker")
        private ContextBroker sessionContextBroker;
        
        @Autowired
        public SetterBeanService(
                @Qualifier("setterBeanRepository")EntityRepository<SetterBean, ?> r) {
                super("setterBeanRepository" , r);
        }
        
        @Override
        @Transactional
        public SetterBean create(SetterBean entity) {
                sessionContextBroker.put(entity);
                return super.create(entity);
        }
}

package example.u2ware.springfield.part4.step2;

import javax.persistence.Entity;
import javax.persistence.Id;
import javax.validation.constraints.NotNull;

import lombok.Getter;
import lombok.Setter;

import com.u2ware.springfield.config.Springfield;
import com.u2ware.springfield.config.Springfield.Strategy;

@Springfield(
        strategy=Strategy.JPA,
        topLevelMapping="/part4/getter"
)
@Entity
public class GetterBean {

        @Id
        @NotNull
        private @Getter @Setter Integer code;   
        @NotNull
        private @Getter @Setter String name;
}

package example.u2ware.springfield.part4.step2;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import com.u2ware.springfield.repository.EntityRepository;
import com.u2ware.springfield.service.EntityServiceImpl;
import com.u2ware.springfield.support.context.ContextBroker;

@Service
public class GetterBeanService extends EntityServiceImpl<GetterBean, SetterBean>{

        @Autowired @Qualifier("sessionContextBroker")
        private ContextBroker sessionContextBroker;     
        
        @Autowired
        public GetterBeanService(
                @Qualifier("getterBeanRepository")EntityRepository<GetterBean, ?> r) {
                super("getterBeanRepository", r);
        }

        @Override
        @Transactional
        public GetterBean createForm(GetterBean entity) {
                SetterBean setterBean = sessionContextBroker.get(SetterBean.class);
                if(setterBean != null){
                        entity.setCode(setterBean.getCode());
                        entity.setName(setterBean.getName());
                }
                return super.createForm(entity);
        }       
}

public interface ContextBroker {

        public <O> void put(O object);
        public <O> O get(Class<O> type);
        public <O> O get(Class<O> type, boolean throwsException);
        public <O> O remove(Class<O> type);
}