94 lines
2.5 KiB
Go
94 lines
2.5 KiB
Go
|
package cassandra
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/gocql/gocql"
|
|||
|
|
"github.com/testcontainers/testcontainers-go"
|
|||
|
|
"github.com/testcontainers/testcontainers-go/wait"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
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"
|
|||
|
|
}
|