Function GetStarRocksContainer

backend/common/testcontainer/testcontainer.go:286–326

GetStarRocksContainer creates a StarRocks (allin1) container for testing. The allin1 image bundles the FE and BE; information_schema is served by the BE, so this waits until the backend actually serves queries rather than only the FE port being open. NOTE: requires an amd64 host — the StarRocks BE

(ctx context.Context)

// come alive under emulation). bytebase CI runs on amd64; locally, run without -short on an
// amd64 machine, or the test is skipped via testing.Short().
func GetStarRocksContainer(ctx context.Context) (retC *Container, retErr error) {
	req := testcontainers.ContainerRequest{
		Image:        "starrocks/allin1-ubuntu:3.4.10",
		ExposedPorts: []string{"9030/tcp"},
		WaitingFor:   wait.ForListeningPort("9030/tcp").WithStartupTimeout(5 * time.Minute),
	}
	c, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
		ContainerRequest: req,
		Started:          true,
	})
	if err != nil {
		return nil, err
	}
	host, err := c.Host(ctx)
	if err != nil {
		return nil, err
	}
	port, err := c.MappedPort(ctx, "9030/tcp")
	if err != nil {
		return nil, err
	}
	dsn := fmt.Sprintf("root@tcp(%s:%s)/?multiStatements=true", host, port.Port())
	db, err := sql.Open("mysql", dsn)
	if err != nil {
		return nil, err
	}
	defer func() {
		if retErr != nil {
			db.Close()
		}
	}()
	if err := waitStarRocksReady(ctx, db); err != nil {
		return nil, err
	}
	return &Container{
		container: c,
		host:      host,
		port:      port.Port(),
		db:        db,
	}, nil
}

// GetTestStarRocksContainer creates a StarRocks container and fails the test on error. It
// skips on non-amd64 hosts: the StarRocks all-in-one BE has no working arm64 build (it never