wrapper.go

type Conn struct {
	dbconn *dbconn.DBConn
}

type ConnPair struct {
	src  *Conn
	dest *Conn
}

type ConnInfo struct {
	host string
	port int
	user string
}

type ConnActions interface {
	CreateSchema(ctx context.Context, name string)
}

func WithWorkerID(parent context.Context, workerID int) context.Context {
	return context.WithValue(parent, "wokerid", workerID)
}

func MakeConn(info *ConnInfo, dbName string) *Conn {
	conn := utils.NewDBConn(dbName, info.host, info.port, info.user)
	return &Conn{dbconn: conn}
}

func MakeConnPair(srcInfo *ConnInfo, srcDb string, destInfo *ConnInfo, destDb string) *ConnPair {
	s := MakeConn(srcInfo, srcDb)
	d := MakeConn(destInfo, destDb)
	return &ConnPair{src: s, dest: d}
}

func (conn *Conn) CreateSchema(ctx context.Context, name string) {
	workerID := ctx.Value("workerid").(int)

	query := fmt.Sprintf("CREATE SCHEMA \"%s\"", utils.QuoteIdent(name))
	gplog.Info("Creating schema \"%s\"", name)
	conn.dbconn.MustExec(query, workerID)
}

func doCopy(workerID int) {
	ctx := context.Background()
	ctx = WithWorkerID(ctx, workerID)

	connPair := MakeConnPair(nil, "srcDb", nil, "descDb")
	connPair.src.CreateSchema(ctx, "newSchema")
}