Creazione di una connessione SSH utilizzando SSHOperator

Per creare una nuova connessione SSH utilizzando l'interfaccia utente di Apache Airflow

1. Apri la pagina Ambienti sulla console Amazon MWAA.

2. Dall'elenco degli ambienti, scegli Open Airflow UI per il tuo ambiente.

3. Nella pagina dell'interfaccia utente di Apache Airflow, scegli Amministratore dalla barra di navigazione principale per espandere l'elenco a discesa, quindi scegli Connessioni.

4. Nella pagina Elenco connessioni, scegli + o il pulsante Aggiungi un nuovo record per aggiungere una nuova connessione.

5. Nella pagina Aggiungi connessione, aggiungi le seguenti informazioni:

   1. Per ID di connessione, immetteressh_new.

   2. Per Tipo di connessione, scegli SSH dall'elenco a discesa.

      Nota

      Se il tipo di connessione SSH non è disponibile nell'elenco, Amazon MWAA non ha installato il pacchetto richiesto. apache-airflow-providers-ssh Aggiorna il requirements.txt file per includere questo pacchetto, quindi riprova.

   3. Per Host, inserisci l'indirizzo IP dell' EC2 istanza Amazon a cui desideri connetterti. Ad esempio, 12.345.67.89.

   4. Per Nome utente, inserisci ec2-user se ti stai connettendo a un' EC2 istanza Amazon. Il tuo nome utente potrebbe essere diverso, a seconda del tipo di istanza remota a cui desideri connettere Apache Airflow.

   5. Per Extra, inserisci la seguente coppia chiave-valore in formato JSON:

      { "key_file": "/usr/local/airflow/dags/your-secret-key.pem" }

      Questa coppia chiave-valore indica ad Apache Airflow di cercare la chiave segreta nella directory locale. /dags

1. Apri un terminale e vai alla directory in cui è memorizzato il codice DAG. Ad esempio:

   cd dags

2. Copiate il contenuto del seguente esempio di codice e salvatelo localmente con nome. ssh.py

   from airflow.decorators import dag
   from datetime import datetime
   from airflow.providers.ssh.operators.ssh import SSHOperator
   
   @dag(
       dag_id="ssh_operator_example",
       schedule_interval=None,     
       start_date=datetime(2022, 1, 1),
       catchup=False,
       )
   def ssh_dag():
       task_1=SSHOperator(
           task_id="ssh_task",
           ssh_conn_id='ssh_new',
           command='hostname',
       )
   
   my_ssh_dag = ssh_dag()

3. Esegui il AWS CLI comando seguente per copiare il DAG nel bucket del tuo ambiente, quindi attiva il DAG utilizzando l'interfaccia utente di Apache Airflow.

   aws s3 cp your-dag.py s3://your-environment-bucket/dags/

4. In caso di successo, otterrete un output simile al seguente nei log delle attività del DAG: ssh_task ssh_operator_example

   [2022-01-01, 12:00:00 UTC] {{base.py:79}} INFO - Using connection to: id: ssh_new. Host: 12.345.67.89, Port: None,
   Schema: , Login: ec2-user, Password: None, extra: {'key_file': '/usr/local/airflow/dags/your-secret-key.pem'}
   [2022-01-01, 12:00:00 UTC] {{ssh.py:264}} WARNING - Remote Identification Change is not verified. This won't protect against Man-In-The-Middle attacks [2022-01-01, 12:00:00 UTC] {{ssh.py:270}} WARNING - No Host Key Verification. This won't protect against Man-In-The-Middle attacks 
   [2022-01-01, 12:00:00 UTC] {{transport.py:1819}} INFO - Connected (version 2.0, client OpenSSH_7.4) 
   [2022-01-01, 12:00:00 UTC] {{transport.py:1819}} INFO - Authentication (publickey) successful!
   [2022-01-01, 12:00:00 UTC] {{ssh.py:139}} INFO - Running command: hostname
   [2022-01-01, 12:00:00 UTC]{{ssh.py:171}} INFO - ip-123-45-67-89.us-west-2.compute.internal
   [2022-01-01, 12:00:00 UTC] {{taskinstance.py:1280}} INFO - Marking task as SUCCESS. dag_id=ssh_operator_example, task_id=ssh_task, execution_date=20220712T200914, start_date=20220712T200915, end_date=20220712T200916