Подключение к базе данных Yandex Managed Service for YDB из функции Cloud Functions на Python

В консоли управления выберите каталог, в котором хотите создать сервисный аккаунт.

Выберите вкладку Сервисные аккаунты.

Нажмите кнопку Создать сервисный аккаунт.

Введите имя сервисного аккаунта, например sa-function. Требования к имени:

• Длина — от 3 до 63 символов.
• Может содержать строчные буквы латинского алфавита, цифры и дефисы.
• Первый символ — буква. Последний символ — не дефис.

Нажмите Добавить роль и выберите editor.

Нажмите кнопку Создать.

В консоли управления выберите каталог, в котором хотите создать базу данных.

В списке сервисов выберите Managed Service for YDB.

Нажмите кнопку Создать базу данных.

Введите имя базы. Требования к имени:

• Длина — от 3 до 63 символов.
• Может содержать строчные буквы латинского алфавита, цифры и дефисы.
• Первый символ — буква. Последний символ — не дефис.

В блоке Тип базы данных выберите опцию Serverless.

Нажмите кнопку Создать базу данных.

Дождитесь запуска базы данных. В процессе создания база будет иметь статус Provisioning. Когда база станет готова к использованию, статус сменится на Running.

Нажмите на имя созданной БД.

В блоке YDB эндпоинт найдите поля Эндпоинт и База данных и сохраните их значения. Они понадобятся на следующем шаге.

В консоли управления выберите каталог, в котором хотите создать функцию.

В списке сервисов выберите Cloud Functions.

Нажмите кнопку Создать функцию.

Введите имя и описание функции. Требования к имени:

• Длина — от 3 до 63 символов.
• Может содержать строчные буквы латинского алфавита, цифры и дефисы.
• Первый символ — буква. Последний символ — не дефис.

Нажмите кнопку Создать.

В блоке Редактор выберите среду выполнения Python и нажмите Продолжить.

В блоке Код функции очистите содержимое файла index.py и вставьте в него следующий код:

import os
import ydb

# Create driver in global space.
driver = ydb.Driver(endpoint=os.getenv('YDB_ENDPOINT'), database=os.getenv('YDB_DATABASE'))
# Wait for the driver to become active for requests.
driver.wait(fail_fast=True, timeout=5)
# Create the session pool instance to manage YDB sessions.
pool = ydb.SessionPool(driver)

def execute_query(session):
  # Create the transaction and execute query.
  return session.transaction().execute(
    'select 1 as cnt;',
    commit_tx=True,
    settings=ydb.BaseRequestSettings().with_timeout(3).with_operation_timeout(2)
  )

def handler(event, context):
  # Execute query with the retry_operation helper.
  result = pool.retry_operation_sync(execute_query)
  return {
    'statusCode': 200,
    'body': str(result[0].rows[0].cnt == 1),
  }

В блоке Код функции создайте файл requirements.txt и вставьте в него следующий текст:

ydb

В качестве точки входа укажите index.handler.

Выберите сервисный аккаунт, например sa-function.

Настройте переменные окружения:

• YDB_ENDPOINT — введите строку вида <протокол>://<эндпоинт базы данных>.

  Например, если протокол grpcs, а эндпоинт ydb.serverless.yandexcloud.net:2135, введите grpcs://ydb.serverless.yandexcloud.net:2135.

• YDB_DATABASE — введите сохраненное ранее значение поля База данных, например /ru-central1/b1gia87mbaomkfvsleds/etn02j1mlm4vgjhij03e.

В верхнем правом углу блока Редактор нажмите кнопку Создать версию.

Перейдите на вкладку Тестирование.

Нажмите кнопку Запустить тест и посмотрите результат тестирования.

При успешном подключении к БД и выполнении запроса состояние функции изменится на Выполнена, а ответ функции будет содержать следующий текст:

{
  "statusCode": 200,
  "body": "True"
}