Mysql
 sql >> Base de Dados >  >> RDS >> Mysql

Criar Spark Dataframe a partir da consulta SQL


Encontrei isso aqui Migração de dados em massa por meio do Spark SQL

O parâmetro dbname pode ser qualquer consulta entre parênteses com um alias. Então, no meu caso, eu preciso fazer isso:
val query = """
  (select dl.DialogLineID, dlwim.Sequence, wi.WordRootID from Dialog as d
    join DialogLine as dl on dl.DialogID=d.DialogID
    join DialogLineWordInstanceMatch as dlwim on dlwim.DialogLineID=dl.DialogLineID
    join WordInstance as wi on wi.WordInstanceID=dlwim.WordInstanceID
    join WordRoot as wr on wr.WordRootID=wi.WordRootID
    where d.InSite=1 and dl.Active=1
    limit 100) foo
"""

val df = sqlContext.format("jdbc").
  option("url", "jdbc:mysql://localhost:3306/local_content").
  option("driver", "com.mysql.jdbc.Driver").
  option("useUnicode", "true").
  option("continueBatchOnError","true").
  option("useSSL", "false").
  option("user", "root").
  option("password", "").
  option("dbtable",query).
  load()

Como esperado, carregar cada tabela como seu próprio Dataframe e juntá-las no Spark foi muito ineficiente.