39 lines
698 B
Go
39 lines
698 B
Go
package redis
|
|
|
|
import "encoding/json"
|
|
|
|
type ShardMetadata struct {
|
|
IP string
|
|
Port int
|
|
}
|
|
|
|
const (
|
|
SHARD_SET = "shards"
|
|
)
|
|
|
|
func (r *RedisHandler) RegisterShard(shard ShardMetadata) error {
|
|
value, err := json.Marshal(shard)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return r.client.SAdd(r.ctx, SHARD_SET, value).Err()
|
|
}
|
|
|
|
func (r *RedisHandler) GetShards() []ShardMetadata {
|
|
shardData := r.client.SMembers(r.ctx, SHARD_SET).Val()
|
|
|
|
// unmarshal all shards
|
|
shards := make([]ShardMetadata, 0, len(shardData))
|
|
for _, data := range shardData {
|
|
var shard ShardMetadata
|
|
if err := json.Unmarshal([]byte(data), &shard); err != nil {
|
|
continue
|
|
}
|
|
|
|
shards = append(shards, shard)
|
|
}
|
|
|
|
return shards
|
|
}
|