/* PostgreSQL- ի երկու սերվերներ կան: * PROD - OLTP սերվեր, * STATS - երկարատև վերլուծական հարցերի սերվեր: Ընթացիկ սերվերում, prod տվյալների բազայում, կա մեծ (10Tb) աղյուսակ հետևյալ կառուցվածքով: CREATE TABLE profiles( id SERIAL, data JSONB ) Աղյուսակում կարող են լինել "հորատներ", այսինքն, որոշ `id`-ներ կարող են բաց թողնվել: Պետք է գրել ծրագիր, որը կկոպիրացնի profiles աղյուսակը PROD-ից STATS: Նախատեսվում է, որ օգտագործվելու են հետևյալ ինտերֆեյսները տվյալների բազաներով աշխատելու համար: type Row []interface{} type Database interface { // իրականացմանը կարող է վերականգնել կապերը // SaveRows կանչը էնդպոտենտ է io.Closer GetMaxID(ctx context.Context) (uint64, error) LoadRows(ctx context.Context, minID, maxID uint64) ([]Row, error) // [minID, maxID] SaveRows(ctx context.Context, rows []Row) error } func Connect(ctx context.Context, dbname string) (Database, error) // CopyTable // Եթե full=false, ապա շարունակել տվյալների փոխանցումը նախորդ սխալի վայրից // Եթե full=true, փոխանցել բոլոր տվյալները func CopyTable(fromName string, toName string, full bool) error { // ... ձեր կոդը } Եթե փոխանցվում է `full=false` տարբերակը, ապա ծրագրը պետք է շարունակել տվյալների փոխանցումը նախորդ սխալի վայրից: Եթե `full=true`, ապա պետք է փոխանցի բոլոր տվյալները: **Հիմնական մակարդակ**: - տվյալների սերիական փոխանցում մեկ հոսքով - սխալներից հետո վերականգնում (ընտրանք `full=false`) Ավելին՝ - եթե անհրաժեշտ է, կարող եք ընդլայնել ինտերֆեյսը՝ ավելացնելով ձեր մեթոդները - եթե անհրաժեշտ է, կարող եք ուղղակիորեն օգտագործել **database/sql** փաթեթը

Для решения задачи копирования большой таблицы profiles с сервера PROD на STATS с возможностью продолжения после сбоев и опцией полного копирования, можно реализовать следующий подход:

1. Определить последний успешно скопированный ID на сервере STATS. Для этого можно добавить в интерфейс Database метод, например, GetMaxCopiedID(ctx context.Context) (uint64, error), который вернёт максимальный id, уже скопированный в таблицу на сервере STATS.

2. При запуске функции CopyTable:

   • Если full=true, то начинаем копирование с id=1 (или минимального id в таблице PROD).
   • Если full=false, то запрашиваем последний скопированный id на STATS и начинаем копирование с id = lastCopiedID + 1.

3. Для копирования данных используем батчи по диапазонам id, например, по 1000 записей за раз. В цикле:

   • Получаем максимальный id в таблице PROD через GetMaxID.
   • Загружаем строки из PROD с id в диапазоне [startID, startID+batchSize-1] через LoadRows.
   • Сохраняем их на STATS через SaveRows.
   • Обновляем startID = startID + batchSize.

4. Если во время копирования происходит сбой, при следующем запуске с full=false процесс возобновится с последнего успешно сохранённого id.

Пример расширения интерфейса и реализации CopyTable:

// Расширение интерфейса Database для получения максимального скопированного id
func (db Database) GetMaxCopiedID(ctx context.Context) (uint64, error) {
    // Реализация зависит от структуры таблицы на STATS
    // Например, SELECT MAX(id) FROM profiles
}

func CopyTable(fromName string, toName string, full bool) error {
    ctx := context.Background()
    fromDB, err := Connect(ctx, fromName)
    if err != nil {
        return err
    }
    defer fromDB.Close()

    toDB, err := Connect(ctx, toName)
    if err != nil {
        return err
    }
    defer toDB.Close()

    var startID uint64 = 1
    if !full {
        startID, err = toDB.GetMaxCopiedID(ctx)
        if err != nil {
            return err
        }
        startID++
    }

    maxID, err := fromDB.GetMaxID(ctx)
    if err != nil {
        return err
    }

    batchSize := uint64(1000)

    for currentID := startID; currentID <= maxID; currentID += batchSize {
        endID := currentID + batchSize - 1
        if endID > maxID {
            endID = maxID
        }

        rows, err := fromDB.LoadRows(ctx, currentID, endID)
        if err != nil {
            return err
        }

        if len(rows) == 0 {
            continue
        }

        err = toDB.SaveRows(ctx, rows)
        if err != nil {
            return err
        }
    }

    return nil
}

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