Динамический sortBy с Slick 3

Question

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

trait Model { 
    type ATable <: AbstractTable[_] 

    def tableQuery: TableQuery[ATable] 

    def sortMap: Map[String, Rep[_]] 

    private def sortKey[T](e: ATable, sort: (String, SortOrder)): ColumnOrdered[_] = sort match { 
     case (field, SortOrder.Asc) => ColumnOrdered(sortMap.getOrElse(field, throw new ClientException(s"Can't sort by $field")), Ordering(Ordering.Asc)) 
     case (field, SortOrder.Desc) => ColumnOrdered(sortMap.getOrElse(field, throw new ClientException(s"Can't sort by $field")), Ordering(Ordering.Desc)) 
    } 
    def all(sort: (String, SortOrder)) = tableQuery.sortBy(sortKey(_, sort)).result 
} 

object User extends Model { 
    type ATable = Tables.User 
    val tableQuery = Tables.User 

    val sortMap = Map("id" -> tableQuery.id) 
} 

Но работает db.run(User.all(("id", SortOrder.Asc)) выдает следующее сообщение об ошибке:

slick.SlickException: No type for symbol name found in Vector[t2<@t3<UnassignedType>>] 

Кто-нибудь знает лучшего решения или там, где я неправильно?

Answer 1

Путем я решил это было изменить мой sortMap, чтобы содержать тип таблицы с типом столбца неявной (от Map[String, Rep[_]] до Map[String, ATable => Rep[_]]):

trait Model { 
    type ATable <: AbstractTable[_] 

    def tableQuery: TableQuery[ATable] 

    def sortMap: Map[String, ATable => Rep[_]] 

    private def sortKey(baseQ: Query[ATable, ATable#TableElementType, Seq], sort: (String, SortOrder)) = { 
     val rep = sortMap.getOrElse(sort._1, throw new ClientException(s"Can't sort by ${sort._1}")) 
     val orderedRep = sort._2 match { 
      case SortOrder.Asc => (t: ATable) => ColumnOrdered(rep(t), slick.ast.Ordering(slick.ast.Ordering.Asc)) 
      case SortOrder.Desc => (t: ATable) => ColumnOrdered(rep(t), slick.ast.Ordering(slick.ast.Ordering.Desc)) 
     } 
     q.sortBy(orderedRep) 
    } 

    def all(sort: (String, SortOrder)) = sortKey(tableQuery, sort).result 
} 

object User extends Model { 
    type ATable = Tables.User 
    val tableQuery = Tables.User 

    val sortMap = Map("id" -> { (t: ATable) => t.id }) 
} 

Единственный неоптимальным аспект этого решения заключается в том, что, если новый столбец добавляется в базу данных, которую необходимо добавить вручную в sortKey. Я изучаю эти карты в сценариях генерации кода таблицы.

Answer 2

Вам просто нужно включить в поколение что-то вроде

val columns = Map("id" -> { (t: User) => t.id }) 

и использовать его так:

class UserDAO{ 
//... 
baseQuery = filterContext.sort.foldLeft[Query[User, UserRow, Seq]](User) ((_, s) => UserColumnFilter.getSort(s,User.baseTableRow.columns)); 
//... 
} 

case class Sort(columnName:String,value: String) {} 

case class FilterContext(filters:List[Filter],page:Int,pageSize:Int,sort:List[Sort]) 
    object UserColumnSort extends ColumnSort[User,UserRow](User) 
    class ColumnSort[T<:Table[C],C](query:TableQuery[T]) 
    {  

     def getSort(s:Sort,columns:Map[String, T => Rep[_]]):Query[T, C, Seq] = { 
     val aux = columns.getOrElse(s.columnName, throw new Exception(s"Can't sort by ${s.columnName}")); 
      val orderedRep = s.value match { 
       case "asc" => (t: T) => ColumnOrdered(aux(t), slick.ast.Ordering(slick.ast.Ordering.Asc)) 
       case _ => (t: T) => ColumnOrdered(aux(t), slick.ast.Ordering(slick.ast.Ordering.Desc)) 
      } 
      return query.sortBy(orderedRep); 

     } 
    }