petstoreapplication.java

来自「Java EE 5 许多新功能都包含经过修补的 EJB 架构」· Java 代码 · 共 100 行

/*
 * Copyright 2006 Borys Burnayev
 * 
 * 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 com.rdacorp.petstore.web.action;

import java.util.Collection;

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.servlet.ServletContext;

import com.rdacorp.petstore.business.PetStoreService;
import com.rdacorp.petstore.domain.Category;
import com.rdacorp.petstore.domain.CreditCardCompany;
import com.rdacorp.petstore.domain.Language;
import com.rdacorp.petstore.domain.State;

/**
 * @author Borys Burnayev
 */
public class PetStoreApplication {
   private PetStoreService petStoreService;

   private Collection<State> states;

   private Collection<Category> categories;

   private Collection<Language> languages;

   private Collection<CreditCardCompany> ccTypes;

   public PetStoreApplication(ServletContext servletContext) {
      petStoreService = (PetStoreService) servletContext.getAttribute("petStoreService");
      if (petStoreService == null) {
         try {
            Context initialContext = new InitialContext();
            petStoreService = (PetStoreService) initialContext.lookup("petstore/PetStoreService/local");
            servletContext.setAttribute("petStoreService", petStoreService);
         }
         catch (NamingException e) {
            throw new RuntimeException("NamingException occured", e);
         }
      }
      states = (Collection) servletContext.getAttribute("states");
      if (states == null) {
         states = petStoreService.getStates();
         servletContext.setAttribute("states", states);
      }
      categories = (Collection) servletContext.getAttribute("categories");
      if (categories == null) {
         categories = petStoreService.getCategories();
         servletContext.setAttribute("categories", categories);
      }
      languages = (Collection) servletContext.getAttribute("languages");
      if (languages == null) {
         languages = petStoreService.getLanguages();
         servletContext.setAttribute("languages", languages);
      }
      ccTypes = (Collection) servletContext.getAttribute("ccTypes");
      if (ccTypes == null) {
         ccTypes = petStoreService.getCreditCardCompanies();
         servletContext.setAttribute("ccTypes", ccTypes);
      }
   }

   public PetStoreService getPetStoreService() {
      return petStoreService;
   }

   public Collection<State> getStates() {
      return states;
   }

   public Collection<Category> getCategories() {
      return categories;
   }

   public Collection<Language> getLanguages() {
      return languages;
   }

   public Collection<CreditCardCompany> getCreditCardCompanies() {
      return ccTypes;
   }
}