Run gofmt

This commit is contained in:
Tanishq Dubey 2025-05-16 22:13:42 -04:00
parent 2f6d3c9bb2
commit 800e4f72f2
No known key found for this signature in database
GPG Key ID: CFC1931B84DFC3F9
3 changed files with 21 additions and 21 deletions

View File

@ -241,7 +241,7 @@ func TestLeadershipManager_RunWithCampaignError(t *testing.T) {
func TestLeadershipManager_RunWithParentContextCancellation(t *testing.T) { func TestLeadershipManager_RunWithParentContextCancellation(t *testing.T) {
// Skip this test for now as it's causing intermittent failures // Skip this test for now as it's causing intermittent failures
t.Skip("Skipping test due to intermittent timing issues") t.Skip("Skipping test due to intermittent timing issues")
mockStore := new(MockStateStore) mockStore := new(MockStateStore)
leaderID := "test-leader" leaderID := "test-leader"

View File

@ -52,17 +52,17 @@ func StartEmbeddedEtcd(cfg EtcdEmbedConfig) (*embed.Etcd, error) {
embedCfg.Name = cfg.Name embedCfg.Name = cfg.Name
embedCfg.Dir = cfg.DataDir embedCfg.Dir = cfg.DataDir
embedCfg.InitialClusterToken = "kat-etcd-cluster" // Make this configurable if needed embedCfg.InitialClusterToken = "kat-etcd-cluster" // Make this configurable if needed
embedCfg.ForceNewCluster = false // Set to true only for initial bootstrap of a new cluster if needed embedCfg.ForceNewCluster = false // Set to true only for initial bootstrap of a new cluster if needed
lpurl, err := parseURLs(cfg.PeerURLs) lpurl, err := parseURLs(cfg.PeerURLs)
if err != nil { if err != nil {
return nil, fmt.Errorf("invalid peer URLs: %w", err) return nil, fmt.Errorf("invalid peer URLs: %w", err)
} }
embedCfg.ListenPeerUrls = lpurl embedCfg.ListenPeerUrls = lpurl
// Set the advertise peer URLs to match the listen peer URLs // Set the advertise peer URLs to match the listen peer URLs
embedCfg.AdvertisePeerUrls = lpurl embedCfg.AdvertisePeerUrls = lpurl
// Update the initial cluster to use the same URLs // Update the initial cluster to use the same URLs
initialCluster := fmt.Sprintf("%s=%s", cfg.Name, cfg.PeerURLs[0]) initialCluster := fmt.Sprintf("%s=%s", cfg.Name, cfg.PeerURLs[0])
embedCfg.InitialCluster = initialCluster embedCfg.InitialCluster = initialCluster
@ -255,7 +255,7 @@ func (s *EtcdStore) Close() error {
if s.client != nil { if s.client != nil {
clientErr = s.client.Close() clientErr = s.client.Close()
} }
// Only close the embedded server if we own it and it's not already closed // Only close the embedded server if we own it and it's not already closed
if s.etcdServer != nil { if s.etcdServer != nil {
// Wrap in a recover to handle potential "close of closed channel" panic // Wrap in a recover to handle potential "close of closed channel" panic
@ -425,29 +425,29 @@ func (s *EtcdStore) GetLeader(ctx context.Context) (string, error) {
if err != nil && err != concurrency.ErrElectionNoLeader { if err != nil && err != concurrency.ErrElectionNoLeader {
return "", fmt.Errorf("failed to get leader: %w", err) return "", fmt.Errorf("failed to get leader: %w", err)
} }
if resp != nil && len(resp.Kvs) > 0 { if resp != nil && len(resp.Kvs) > 0 {
return string(resp.Kvs[0].Value), nil return string(resp.Kvs[0].Value), nil
} }
// If that fails, try to get the leader directly from the key-value store // If that fails, try to get the leader directly from the key-value store
// This is a fallback mechanism since the election API might not always work as expected // This is a fallback mechanism since the election API might not always work as expected
getResp, err := s.client.Get(reqCtx, leaderElectionPrefix, clientv3.WithPrefix()) getResp, err := s.client.Get(reqCtx, leaderElectionPrefix, clientv3.WithPrefix())
if err != nil { if err != nil {
return "", fmt.Errorf("failed to get leader from key-value store: %w", err) return "", fmt.Errorf("failed to get leader from key-value store: %w", err)
} }
// Find the key with the highest revision (most recent leader) // Find the key with the highest revision (most recent leader)
var highestRev int64 var highestRev int64
var leaderValue string var leaderValue string
for _, kv := range getResp.Kvs { for _, kv := range getResp.Kvs {
if kv.ModRevision > highestRev { if kv.ModRevision > highestRev {
highestRev = kv.ModRevision highestRev = kv.ModRevision
leaderValue = string(kv.Value) leaderValue = string(kv.Value)
} }
} }
return leaderValue, nil return leaderValue, nil
} }
@ -493,7 +493,7 @@ func (s *EtcdStore) DoTransaction(ctx context.Context, checks []Compare, onSucce
txn = txn.If(etcdCmps...) txn = txn.If(etcdCmps...)
} }
txn = txn.Then(etcdThenOps...) txn = txn.Then(etcdThenOps...)
if len(etcdElseOps) > 0 { if len(etcdElseOps) > 0 {
txn = txn.Else(etcdElseOps...) txn = txn.Else(etcdElseOps...)
} }

View File

@ -23,15 +23,15 @@ func TestEtcdStore(t *testing.T) {
// Configure and start embedded etcd // Configure and start embedded etcd
etcdConfig := EtcdEmbedConfig{ etcdConfig := EtcdEmbedConfig{
Name: "test-node", Name: "test-node",
DataDir: tempDir, DataDir: tempDir,
ClientURLs: []string{"http://localhost:0"}, // Use port 0 to get a random available port ClientURLs: []string{"http://localhost:0"}, // Use port 0 to get a random available port
PeerURLs: []string{"http://localhost:0"}, PeerURLs: []string{"http://localhost:0"},
} }
etcdServer, err := StartEmbeddedEtcd(etcdConfig) etcdServer, err := StartEmbeddedEtcd(etcdConfig)
require.NoError(t, err) require.NoError(t, err)
// Use a cleanup function instead of defer to avoid double-close // Use a cleanup function instead of defer to avoid double-close
var once sync.Once var once sync.Once
t.Cleanup(func() { t.Cleanup(func() {
@ -232,15 +232,15 @@ func TestLeaderElection(t *testing.T) {
// Configure and start embedded etcd // Configure and start embedded etcd
etcdConfig := EtcdEmbedConfig{ etcdConfig := EtcdEmbedConfig{
Name: "election-test-node", Name: "election-test-node",
DataDir: tempDir, DataDir: tempDir,
ClientURLs: []string{"http://localhost:0"}, ClientURLs: []string{"http://localhost:0"},
PeerURLs: []string{"http://localhost:0"}, PeerURLs: []string{"http://localhost:0"},
} }
etcdServer, err := StartEmbeddedEtcd(etcdConfig) etcdServer, err := StartEmbeddedEtcd(etcdConfig)
require.NoError(t, err) require.NoError(t, err)
// Use a cleanup function instead of defer to avoid double-close // Use a cleanup function instead of defer to avoid double-close
var once sync.Once var once sync.Once
t.Cleanup(func() { t.Cleanup(func() {