Как я могу передать класс в качестве параметра и вернуть универсальную коллекцию в Java? - PullRequest
Новая
       32

Как я могу передать класс в качестве параметра и вернуть универсальную коллекцию в Java?

35 голосов
/ 03 августа 2010

Я проектирую простой объект доступа к данным для моего Java-приложения. У меня есть несколько классов (записей), которые представляют одну строку в таблицах, как User и Fruit.

Я хотел бы иметь один метод для получения всех записей определенного типа.

На данный момент у меня так: 

public List<User> getAllUsers() {
 ...
}

public List<Fruit> getAllFruits() {
 ...
}

....

Но я хотел бы иметь один полиморфный метод, подобный этому (неверный): 

public List<T> getAllRecords(Class<T> type) {
    if(type instanceof User) {
        // Use JDBC and SQL SELECT * FROM user
    } else if(type instanceof Fruit) {
        // Use JDBC and SQL SELECT * FROM fruit
    }
    return collection;
}

Пример использования: 

List<Fruit> fruits = myDataAccessObject.getAllRecrods(Fruit.class);
List<User> users = myDataAccessObject.getAllRecords(User.class);

Как я могу сделать это на Java?

Ответы [ 7 ]

27 голосов
/ 03 августа 2010

Похоже, вы хотите адаптировать то, что Джош Блох называет шаблоном Typesafe Heterogenous Container : вы передаете токен типа Class<T> и хотите получить обратно List<T>.

Обычный старый THC может сопоставить Class<T> с T типобезопасным способом, но поскольку вместо этого вы действительно хотите List<T>, то вы хотите использовать то, что Нил Гафтер называет токенами супертипа .

Следующий фрагмент адаптирован из кода Сумасшедшего Боба Ли, опубликованного в блоге Нила Гафтера: 

public abstract class TypeReference<T> {
    private final Type type;

    protected TypeReference() {
        Type superclass = getClass().getGenericSuperclass();
        if (superclass instanceof Class<?>) {
            throw new RuntimeException("Missing type parameter.");
        }
        this.type = ((ParameterizedType) superclass).getActualTypeArguments()[0];
    }
    public Type getType() {
        return this.type;
    }
}

Теперь вы можете создать токен супертипа , подобный следующему: 

    TypeReference<String> stringTypeRef =
            new TypeReference<String>(){};

    TypeReference<Integer> integerTypeRef =
            new TypeReference<Integer>(){};

    TypeReference<List<Boolean>> listBoolTypeRef =
            new TypeReference<List<Boolean>>(){};

По сути вы передаете TypeReference<T> вместо Class<T>.Разница в том, что List<String>.class нет, но вы можете сделать TypeReference<List<String>>.

Так что теперь мы можем сделать наш контейнер следующим образом (следующее адаптировано из исходного кода Джоша Блоха): 

public class Favorites {
    private Map<Type, Object> favorites =
        new HashMap<Type, Object>();

    public <T> void setFavorite(TypeReference<T> ref, T thing) {
        favorites.put(ref.getType(), thing);
    }
    public <T> T getFavorite(TypeReference<T> ref) {
        @SuppressWarnings("unchecked")
        T ret = (T) favorites.get(ref.getType());
        return ret;
    }
}

Теперь мы можем соединить их вместе: 

    Favorites f = new Favorites();
    f.setFavorite(stringTypeRef, "Java");
    f.setFavorite(integerTypeRef, 42);
    f.setFavorite(listBoolTypeRef, Arrays.asList(true, true));

    String s = f.getFavorite(stringTypeRef);
    int i = f.getFavorite(integerTypeRef);
    List<Boolean> list = f.getFavorite(listBoolTypeRef);

    System.out.println(s);    // "Java"
    System.out.println(i);    // "42"
    System.out.println(list); // "[true, true]"

Нил Гафтер утверждал в своем блоге, что с некоторыми дополнительными прибавками TypeReference для жетонов супертипа сделает достойное включение вJDK.

Вложения

Ссылки

22 голосов
/ 05 августа 2010

Поскольку вы говорите, что вам не нужны методы доступа к данным в разных классах (в комментарии к ответу Аниша), я подумал, почему бы не попробовать что-то подобное. 

public class Records {

    public interface RecordFetcher<T>{
        public List<T> getRecords();
    }
    static RecordFetcher<Fruit> Fruit=new RecordFetcher<Fruit>(){
        public List<Fruit> getRecords() {
            ...
        }
    };


    static RecordFetcher<User> User=new RecordFetcher<User>(){
        public List<User> getRecords() {
            ...
        }   
    };

    public static void main(String[] args) {
        List<Fruit> fruitRecords=Records.Fruit.getRecords();
        List<User> userRecords=Records.User.getRecords();

    }
}

EDIT:

Я хотел бы добавить еще одну мою реализацию. 

public class Test 
{ 
    public static void main(String[] args) 
    { 
       Test dataAccess=new Test();
       List<Fruit> FruitList=dataAccess.getAllRecords(Fruit.myType);
       List<User> UserList=dataAccess.getAllRecords(User.myType);
    } 
    <T> List<T> getAllRecords(T cl)
    {
        List<T> list=new ArrayList<T>();
        if(cl instanceof Fruit)
        {
             // Use JDBC and SQL SELECT * FROM fruit
        }
        else if(cl instanceof User)
        {
            // Use JDBC and SQL SELECT * FROM user
        }
        return list;
    }
}
class Fruit
{
    static final Fruit myType;
    static {myType=new Fruit();}
}
class User
{
    static final User myType;
    static {myType=new User();}
}

EDIT

Я думаю, что эта реализация такая же, как вы просили 

public class Test 
{ 
    public static void main(String[] args) throws InstantiationException, IllegalAccessException 
    { 
       Test dataAccess=new Test();

       List<Fruit> FruitList=dataAccess.getAllRecords(Fruit.class);

       List<User> UserList=dataAccess.getAllRecords(User.class);

    } 
    <T> List<T> getAllRecords(Class<T> cl) throws InstantiationException, IllegalAccessException
    {
        T inst=cl.newInstance();
        List<T> list=new ArrayList<T>();
        if(inst instanceof Fruit)
        {
             // Use JDBC and SQL SELECT * FROM user
        }
        else if(inst instanceof User)
        {
            // Use JDBC and SQL SELECT * FROM fruit
        }
        return list;
    }
}
12 голосов
/ 03 августа 2010

Вы довольно близко. 

public <T> LinkedList<T> getAllRecords(List<T> list) {
 ...
}

Это называется Универсальный метод .

Вы хотите указать такой параметр, как List<T>. Затем, основываясь на типе передаваемого вами списка, Java выведет общий тип для возврата.

Edit:

Ответ Поли очень хороший. Вам должно быть достаточно легко выполнить следующие действия и не создавать класс TypeReference. 

List<Fruit> fruit = myDataAccessObject.getAllRecrods(new LinkedList<Fruit>());
List<User> users = myDataAccessObject.getAllRecords(new LinkedList<User>());
4 голосов
/ 03 августа 2010

В зависимости от того, как вы на самом деле извлекаете свои данные, вы можете сделать что-то вроде этого: 

private static <T> List<T> getAll(Class<T> cls){
  List<T> fromSql = (List<T>) sql.query("SELECT * FROM objects WHERE type="+cls.getName());
  return fromSql;
}

Для этого требуется, чтобы ваш sql объект возвращал правильный тип списка, что делают O / R-сопоставители, как iBatis.

Если вам нужно различить переданные типы, вы все равно можете сделать переключатель / регистр на cls.

3 голосов
/ 03 августа 2010

Ну, я действительно не знаю, нужно ли вам это так.Но вот полиморфный подход.Это может как-то помочь.

Создание разных объектов для разных таблиц, реализующих общий интерфейс.Это означает, что вы представляете каждую таблицу как объект. 

import java.util.LinkedList;

public class DataAccessTest 
{

    /**
     * @param args
     */
    public static void main(String[] args) 
    {
        DataAccess myDataAccessObject = new DataAccess();
        Type type1 = new Fruit();
        Type type2 = new User();
        LinkedList<Type> list1 = myDataAccessObject.getAllRecords(type1);
        LinkedList<Type> list2 = myDataAccessObject.getAllRecords(type2);
        LinkedList<Type> list3 = myDataAccessObject.getAllRecords(new Fruit());
        LinkedList<Type> list4 = myDataAccessObject.getAllRecords(new User());
    }
}

class DataAccess
{
    public LinkedList<Type> getAllRecords(Type type)
    {
        return type.getAllRecords();
    }
}

interface Type
{
    public LinkedList<Type> getAllRecords();
}

class Fruit implements Type
{
    public LinkedList<Type> getAllRecords()
    {
        LinkedList<Type> list = new LinkedList<Type>();
        list.add(new Fruit());
        return list;
    }
}

class User implements Type
{
    public LinkedList<Type> getAllRecords() 
    {
        LinkedList<Type> list = new LinkedList<Type>();
        list.add(new User());
        return list;
    }
}
1 голос
/ 10 августа 2014

Я верю, что то, что вы пытаетесь сделать, возможно с помощью немного магии дженериков. Я должен был решить ту же проблему только сейчас, и вот что я сделал: 

public class ListArrayUtils{
   @SuppressWarnings("unchecked") // It is checked. 
   public static <T,E> List<T> filterByType(List<E> aList, Class<T> aClass){
      List<T> ans = new ArrayList<>();
      for(E e: aList){
         if(aClass.isAssignableFrom(e.getClass())){
            ans.add((T)e);
         }
      }
      return ans;
   }       
}

И юнит-тесты: 

public class ListArrayUtilsTest{
   interface IfA{/*nothing*/}
   interface IfB{/*nothing*/}
   class A implements IfA{/*nothing*/}
   class B implements IfB{/*nothing*/}
   class C extends A implements IfB{/*nothing*/}

   @Test
   public void testFilterByType(){
      List<Object> data = new ArrayList<>();
      A a = new A();
      B b = new B();
      C c = new C();
      data.add(a);
      data.add(b);
      data.add(c);

      List<IfB> ans = ListArrayUtils.filterByType(data, IfB.class);

      assertEquals(2, ans.size());
      assertSame(b, ans.get(0));
      assertSame(c, ans.get(1));
   }
}
0 голосов
/ 20 ноября 2014

Я на самом деле сделал это в общей библиотеке доступа к данным. См. Норма . Полный исходный код на Github.

...