Scala: L'exposition d'un ResultSet JDBC par un générateur (itérables)

Question

J'ai un ensemble de lignes dans une base de données, et je voudrais fournir une interface pour filer à travers eux comme ceci:

def findAll: Iterable[MyObject] 

Où nous n'avons pas besoin d'avoir toutes les instances en mémoire à la fois. En C#, vous pouvez facilement créer des générateurs de ce type en utilisant yield, le compilateur s'occupe de convertir du code qui traverse le jeu d'enregistrements en un itérateur (sorte de l'inverser).

Mon code actuel ressemble à ceci:

def findAll: List[MyObject] = { 
    val rs = getRs 
    val values = new ListBuffer[MyObject] 
    while (rs.next()) 
    values += new valueFromResultSet(rs) 
    values.toList 
} 

Est-il possible que je pourrais convertir ce pour ne pas stocker l'ensemble dans la mémoire? Peut-être que je pourrais utiliser un pour la compréhension?

Answer 1

Essayez d'étendre Iterator à la place. Je ne l'ai pas testé, mais quelque chose comme ceci:

def findAll: Iterator[MyObject] = new Iterator[MyObject] { 
    val rs = getRs 
    override def hasNext = rs.hasNext 
    override def next = new valueFromResultSet(rs.next) 
} 

Cela devrait stocker rs quand il est appelé, et autrement juste un emballage léger pour les appels vers rs.

Si vous souhaitez enregistrer les valeurs que vous parcourez, sélectionnez Flux.

Answer 2

je suis tombé sur le même problème et sur la base des idées ci-dessus, j'ai créé la solution suivante en écrivant une classe d'adaptateur:

class RsIterator(rs: ResultSet) extends Iterator[ResultSet] { 
    def hasNext: Boolean = rs.next() 
    def next(): ResultSet = rs 
} 

Avec cela, vous pouvez par exemple effectuer carte opérations sur l'ensemble de résultats - qui était mon intention personnelle:

val x = new RsIterator(resultSet).map(x => { 
    (x.getString("column1"), x.getInt("column2")) 
}) 

un .toList Append afin de forcer l'évaluation. Cela est utile si la connexion à la base de données est fermée avant d'utiliser les valeurs. Sinon, vous obtiendrez une erreur indiquant que vous ne pouvez pas accéder au ResultSet après la fermeture de la connexion.

Answer 3

Une simple (idiomatiques) façon d'obtenir le même serait

Iterator.continually((rs.next(), rs)).takeWhile(_._1).map(r => valueFromResultSet(r._2)).toList 

Vous avez besoin de .toList pour forcer l'évaluation, sinon la collection sous-jacente sera un cours d'eau et le ResultSet peut être fermé avant l'évaluation a eu lieu .