40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package repository
|
||
|
||
import (
|
||
"chat/internal/library/cassandra"
|
||
"context"
|
||
"fmt"
|
||
)
|
||
|
||
// InitSchema 初始化 Cassandra keyspace 和表結構
|
||
func InitSchema(ctx context.Context, db *cassandra.DB, keyspace string) error {
|
||
// 建立 keyspace（如果不存在）
|
||
createKeyspaceStmt := fmt.Sprintf(
|
||
"CREATE KEYSPACE IF NOT EXISTS %s WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 1}",
|
||
keyspace,
|
||
)
|
||
session := db.GetSession()
|
||
if err := session.Query(createKeyspaceStmt, nil).Exec(); err != nil {
|
||
return fmt.Errorf("failed to create keyspace: %w", err)
|
||
}
|
||
|
||
// 建立 messages_by_room 表
|
||
createTableStmt := fmt.Sprintf(`
|
||
CREATE TABLE IF NOT EXISTS %s.messages_by_room (
|
||
room_id text,
|
||
bucket_day text,
|
||
ts bigint,
|
||
message_id text,
|
||
uid text,
|
||
content text,
|
||
PRIMARY KEY ((room_id, bucket_day), ts, message_id)
|
||
) WITH CLUSTERING ORDER BY (ts DESC, message_id DESC)
|
||
`, keyspace)
|
||
|
||
if err := session.Query(createTableStmt, nil).Exec(); err != nil {
|
||
return fmt.Errorf("failed to create messages_by_room table: %w", err)
|
||
}
|
||
|
||
return nil
|
||
}
|