37 lines
1.1 KiB
Go
37 lines
1.1 KiB
Go
package svc
|
||
|
||
import (
|
||
"chat/internal/config"
|
||
"chat/internal/library/cassandra"
|
||
repoImpl "chat/internal/repository"
|
||
"context"
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
)
|
||
|
||
func initCassandra(ctx context.Context, c config.CassandraConf) (*cassandra.DB, error) {
|
||
// 初始化 Cassandra DB
|
||
var cassandraOpts []cassandra.Option
|
||
cassandraOpts = append(cassandraOpts, cassandra.WithHosts(c.Hosts...))
|
||
cassandraOpts = append(cassandraOpts, cassandra.WithPort(c.Port))
|
||
cassandraOpts = append(cassandraOpts, cassandra.WithKeyspace(c.Keyspace))
|
||
if c.UseAuth {
|
||
cassandraOpts = append(cassandraOpts, cassandra.WithAuth(c.Username, c.Password))
|
||
}
|
||
|
||
cassandraDB, err := cassandra.New(cassandraOpts...)
|
||
if err != nil {
|
||
logx.Errorf("Failed to connect to Cassandra: %v", err)
|
||
return nil, err
|
||
}
|
||
logx.Infof("Connected to Cassandra at %v:%d", c.Hosts, c.Port)
|
||
|
||
// 初始化 schema（創建 keyspace 和表）
|
||
if err := repoImpl.InitSchema(ctx, cassandraDB, c.Keyspace); err != nil {
|
||
logx.Errorf("Failed to initialize Cassandra schema: %v", err)
|
||
return nil, err
|
||
}
|
||
logx.Infof("Cassandra schema initialized for keyspace: %s", c.Keyspace)
|
||
|
||
return cassandraDB, nil
|
||
}
|