Golang: компонент регистрации в etcd (Service Discovery)

Для создания компонента регистрации в etcd с использованием Go, который позволяет сервису зарегистрировать себя в реестре и гарантировать, что информация будет доступна другим сервисам, можно использовать библиотеку etcd в Go.

Процесс регистрации в etcd обычно включает два этапа:

1. Регистрация сервиса с уникальным ключом (например, с использованием имени сервиса).
2. Опционально — установка TTL (время жизни), чтобы сервис автоматически удалялся через определенный промежуток времени, если он не обновит свой статус (например, при падении или перезапуске).

Ниже приведен пример компонента для регистрации в etcd с TTL для предотвращения «залипания» устаревших записей.

Пример компонента регистрации в etcd на Go

1. Установите клиент etcd для Go:
   
   Shell

   go get go.etcd.io/etcd/v3/clientv3

   1 2	go get go.etcd.io/etcd/v3/clientv3

    
2. Напишем Go код для регистрации сервиса в etcd:
   
   Go

   package main import ( "context" "fmt" "log" "time" "go.etcd.io/etcd/v3/clientv3" ) func main() { // Подключаемся к серверу etcd client, err := clientv3.New(clientv3.Config{ Endpoints: []string{"localhost:2379"}, // Адрес etcd DialTimeout: 5 * time.Second, }) if err != nil { log.Fatalf("Ошибка при подключении к etcd: %v", err) } defer client.Close() // Регистрируем сервис registerService(client, "service1", "localhost:8080", 10*time.Second) // Ожидаем некоторое время, чтобы проверить регистрацию time.Sleep(30 * time.Second) } // Регистрация сервиса в etcd с TTL func registerService(client *clientv3.Client, serviceName, serviceAddress string, ttl time.Duration) { // Создаем таймер с TTL leaseResp, err := client.Grant(context.Background(), int64(ttl.Seconds())) if err != nil { log.Fatalf("Ошибка при создании TTL в etcd: %v", err) } // Ключ для сервиса в etcd key := "services/" + serviceName value := serviceAddress // Регистрируем сервис в etcd с использованием TTL ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() // Записываем сервис в etcd с TTL _, err = client.Put(ctx, key, value, clientv3.WithLease(leaseResp.ID)) if err != nil { log.Fatalf("Ошибка при регистрации сервиса в etcd: %v", err) } // Обновляем TTL каждые 5 секунд go refreshTTL(client, leaseResp.ID, ttl) // Ожидаем, чтобы продлить жизнь сервиса fmt.Printf("Сервис %s зарегистрирован по адресу %s с TTL %v\n", serviceName, serviceAddress, ttl) } // Функция для обновления TTL func refreshTTL(client *clientv3.Client, leaseID clientv3.LeaseID, ttl time.Duration) { for { // Продлеваем TTL _, err := client.KeepAliveOnce(context.Background(), leaseID) if err != nil { log.Printf("Ошибка при продлении TTL: %v", err) return } time.Sleep(ttl / 2) // Обновляем TTL каждые пол TTL } }

   1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70

   package main   import ( "context" "fmt" "log" "time" "go.etcd.io/etcd/v3/clientv3" )   func main() { // Подключаемся к серверу etcd client, err := clientv3.New(clientv3.Config{ Endpoints: []string{"localhost:2379"}, // Адрес etcd DialTimeout: 5 * time.Second, }) if err != nil { log.Fatalf("Ошибка при подключении к etcd: %v", err) } defer client.Close()   // Регистрируем сервис registerService(client, "service1", "localhost:8080", 10*time.Second)   // Ожидаем некоторое время, чтобы проверить регистрацию time.Sleep(30 * time.Second) }   // Регистрация сервиса в etcd с TTL func registerService(client *clientv3.Client, serviceName, serviceAddress string, ttl time.Duration) { // Создаем таймер с TTL leaseResp, err := client.Grant(context.Background(), int64(ttl.Seconds())) if err != nil { log.Fatalf("Ошибка при создании TTL в etcd: %v", err) }   // Ключ для сервиса в etcd key := "services/" + serviceName value := serviceAddress   // Регистрируем сервис в etcd с использованием TTL ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel()   // Записываем сервис в etcd с TTL _, err = client.Put(ctx, key, value, clientv3.WithLease(leaseResp.ID)) if err != nil { log.Fatalf("Ошибка при регистрации сервиса в etcd: %v", err) }   // Обновляем TTL каждые 5 секунд go refreshTTL(client, leaseResp.ID, ttl)   // Ожидаем, чтобы продлить жизнь сервиса fmt.Printf("Сервис %s зарегистрирован по адресу %s с TTL %v\n", serviceName, serviceAddress, ttl) }   // Функция для обновления TTL func refreshTTL(client *clientv3.Client, leaseID clientv3.LeaseID, ttl time.Duration) { for { // Продлеваем TTL _, err := client.KeepAliveOnce(context.Background(), leaseID) if err != nil { log.Printf("Ошибка при продлении TTL: %v", err) return } time.Sleep(ttl / 2) // Обновляем TTL каждые пол TTL } }

    

Объяснение:

1. Подключение к etcd:
   • Создаем подключение к etcd с использованием конфигурации clientv3.Config. Указываем список эндпоинтов (например, localhost:2379), который должен указывать на узлы вашего etcd-кластера.
2. Регистрация сервиса:
   • В registerService сервис регистрируется в etcd с уникальным ключом, например, services/service1, и значением, которое будет хранить адрес сервиса, например, localhost:8080.
   • Мы используем TTL (время жизни), чтобы убедиться, что запись автоматически удалится, если сервис не обновит информацию о себе. В примере TTL задан в 10 секунд.
   • clientv3.Grant создает lease (аренду), которая будет действовать в течение указанного времени. Мы привязываем эту аренду к записи в etcd.
3. Обновление TTL:
   • В отдельной горутине (refreshTTL) мы обновляем TTL каждые 5 секунд, чтобы запись о сервисе не была удалена. Если сервис продолжает работать и обновляет TTL, его запись останется в etcd.
4. Пример работы:
   • После запуска, сервис будет зарегистрирован в etcd и продлять TTL, пока работает.
   • Если сервис не обновит TTL в течение установленного времени, запись будет удалена.

Дополнительные возможности:

1. Использование watch для обнаружения сервисов: Другие сервисы могут использовать watch для отслеживания изменений в реестре сервисов. Например, они могут отслеживать ключи с префиксом services/ и реагировать на добавление или удаление сервисов.
2. Удаление сервиса: Если сервис завершит работу или будет остановлен, его запись в etcd будет удалена автоматически по истечении TTL. Вы также можете явно удалить запись с помощью команды Delete.

Преимущества использования TTL:

• Автоматическая очистка: Если сервис неожиданно завершит свою работу (например, из-за сбоя), его запись будет удалена автоматически по истечении времени TTL.
• Гибкость: Вы можете настроить TTL на любое время в зависимости от требуемой частоты обновления состояния сервиса.

Заключение:

Этот компонент позволяет эффективно регистрировать сервисы в etcd с использованием TTL для предотвращения «залипания» старых записей, а также предоставляет механизм для автоматического обновления информации о сервисе.