learning/prompts/code-generation/11_io_jdbc.md

Prompt:
Write the Python code to read data from Oracle JDBC using Apache Beam.

Response:
Your Apache Beam pipeline can read data from an Oracle Java Database Connectivity (JDBC) enabled database using the [JdbcIO connector](https://beam.apache.org/releases/pydoc/current/apache_beam.io.jdbc.html).

```python
import logging
import apache_beam as beam
import typing

from apache_beam import coders
from apache_beam import Map
from apache_beam.io.jdbc import ReadFromJdbc
from apache_beam.options.pipeline_options import PipelineOptions
from typing import NamedTuple


class ExampleRow(NamedTuple):
    id: int
    name: str


class JdbcOptions(PipelineOptions):
    @classmethod
    def _add_argparse_args(cls, parser):
        parser.add_argument(
            '--table_name',
            help='Table name'
        )
        parser.add_argument(
            '--jdbc_url',
            help='JDBC URL'
        )
        parser.add_argument(
            '--driver_class_name',
            help='Driver class name'
        )
        parser.add_argument(
            '--username',
            help='Username'
        )
        parser.add_argument(
            '--password',
            help='Password'
        )


def run():
    """
    This pipeline shows how to read from JDBC.
    """

    options = JdbcOptions()
    coders.registry.register_coder(ExampleRow, coders.RowCoder)

    with beam.Pipeline(options=options) as p:

        output = (p | 'Read from JDBC' >> ReadFromJdbc(
                      table_name=options.table_name,
                      driver_class_name=options.driver_class_name,
                      jdbc_url=options.jdbc_url,
                      username=options.username,
                      password=options.password
                    )
                    | 'Log Data' >> Map(logging.info))


if __name__ == "__main__":
    logging.getLogger().setLevel(logging.INFO)
    run()
```

The `JdbcOptions` class is a custom class that extends the `PipelineOptions` object. This code uses [pipeline options](https://beam.apache.org/documentation/patterns/pipeline-options/) to parse command line arguments.

The code creates an Apache Beam pipeline using the `JdbcOptions` class to set the JDBC connection information and the [`ReadFromJdbc` transform](https://beam.apache.org/releases/pydoc/current/apache_beam.io.jdbc.html#apache_beam.io.jdbc.ReadFromJdbc) to read data from the database via JDBC.