blockchain/internal/lib/cassandra/init_cassandra_container_test.go

package cassandra

import (
	"context"
	"fmt"
	"time"

	"github.com/gocql/gocql"
	"github.com/testcontainers/testcontainers-go"
	"github.com/testcontainers/testcontainers-go/wait"
)

type Container struct {
	Ctx       context.Context
	Container testcontainers.Container
	Host      string
	Port      int
}

func initCassandraContainer(version string) (Container, error) {
	ctx := context.Background()
	req := testcontainers.ContainerRequest{
		Image: fmt.Sprintf("cassandra:%s", version),
		Env: map[string]string{
			"CASSANDRA_START_RPC":       "true",
			"CASSANDRA_NUM_TOKENS":      "1",
			"CASSANDRA_ENDPOINT_SNITCH": "GossipingPropertyFileSnitch",
			"CASSANDRA_DC":              "datacenter1",
			"CASSANDRA_RACK":            "rack1",
			"MAX_HEAP_SIZE":             "256M",
			"HEAP_NEWSIZE":              "100M",
		},
		ExposedPorts: []string{"9042/tcp"},
		// 等待 Cassandra 啟動完成的指標字串，依據實際啟動 log 可調整
		WaitingFor: wait.ForLog("Created default superuser role 'cassandra'").
			WithStartupTimeout(2 * time.Minute),
	}
	cassandraContainer, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
		ContainerRequest: req,
		Started:          true,
	})
	if err != nil {
		return Container{}, err
	}

	host, err := cassandraContainer.Host(ctx)
	if err != nil {
		return Container{}, err
	}

	mappedPort, err := cassandraContainer.MappedPort(ctx, "9042")
	if err != nil {
		return Container{}, err
	}

	return Container{ctx, cassandraContainer, host, mappedPort.Int()}, nil
}

// Animal 為不實作 TableName 方法的範例 struct，則會以型別名稱轉換成 snake_case
type Animal struct {
	ID   gocql.UUID `db:"id" partition_key:"true"`
	Type string     `db:"type"`
}

func (m *Animal) TableName() string {
	return "animal"
}

// InvalidEntity 為無 partition key 的範例 struct，預期產生錯誤
type InvalidEntity struct {
	Field string `db:"field"`
}

type MonkeyEntity struct {
	ID       gocql.UUID `db:"id" partition_key:"true"`
	Name     string     `db:"name" clustering_key:"true" sai:"true"`
	UpdateAt time.Time  `db:"update_at"`
	CreateAt time.Time  `db:"create_at"`
}

func (m *MonkeyEntity) TableName() string {
	return "monkey_entity"
}

type CatEntity struct {
	ID       *gocql.UUID `db:"id" partition_key:"true"`
	Name     *string     `db:"name" partition_key:"true"`
	UpdateAt *time.Time  `db:"update_at"`
	CreateAt *time.Time  `db:"create_at" clustering_key:"true"`
}

func (m *CatEntity) TableName() string {
	return "cat_entity"
}

type Consistency struct {
	ID              gocql.UUID `db:"id" partition_key:"true"`
	ConsistencyName string     `db:"consistency_name"` // can editor
	ConsistencyType string     `db:"consistency_type"`
	LastTaskID      string     `db:"last_task_id"` // ConsistencyTask ID
	Target          string     `db:"target"`       // file name can editor
	Status          string     `db:"status"`
	ConsistencyMap  string     `db:"consistency_map"` // JSON string
	CreateAT        int64      `db:"create_at"`
	UpdateAT        int64      `db:"update_at"`
}

func (c *Consistency) TableName() string {
	return "consistency"
}