Skip to content

Commit 048c853

Browse files
authored
Merge pull request kubernetes#127382 from bart0sh/PR158-dra-fix-golangci-lint-findings
Kubelet: DRA: fix golangci-lint findings
2 parents 9134c17 + bba7864 commit 048c853

File tree

3 files changed

+26
-17
lines changed

3 files changed

+26
-17
lines changed

pkg/kubelet/cm/dra/claiminfo.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -115,12 +115,12 @@ func (info *ClaimInfo) isPrepared() bool {
115115
func newClaimInfoCache(stateDir, checkpointName string) (*claimInfoCache, error) {
116116
checkpointer, err := state.NewCheckpointer(stateDir, checkpointName)
117117
if err != nil {
118-
return nil, fmt.Errorf("could not initialize checkpoint manager, please drain node and remove dra state file, err: %+v", err)
118+
return nil, fmt.Errorf("could not initialize checkpoint manager, please drain node and remove dra state file, err: %w", err)
119119
}
120120

121121
checkpoint, err := checkpointer.GetOrCreate()
122122
if err != nil {
123-
return nil, fmt.Errorf("error calling GetOrCreate() on checkpoint state: %v", err)
123+
return nil, fmt.Errorf("error calling GetOrCreate() on checkpoint state: %w", err)
124124
}
125125

126126
cache := &claimInfoCache{
@@ -182,9 +182,9 @@ func (cache *claimInfoCache) delete(claimName, namespace string) {
182182
// that is referenced by the pod with the given UID
183183
// This function is used indirectly by the status manager
184184
// to check if pod can enter termination status
185-
func (cache *claimInfoCache) hasPodReference(UID types.UID) bool {
185+
func (cache *claimInfoCache) hasPodReference(uid types.UID) bool {
186186
for _, claimInfo := range cache.claimInfo {
187-
if claimInfo.hasPodReference(UID) {
187+
if claimInfo.hasPodReference(uid) {
188188
return true
189189
}
190190
}

pkg/kubelet/cm/dra/manager.go

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ type ManagerImpl struct {
7070
func NewManagerImpl(kubeClient clientset.Interface, stateFileDirectory string, nodeName types.NodeName) (*ManagerImpl, error) {
7171
claimInfoCache, err := newClaimInfoCache(stateFileDirectory, draManagerStateFileName)
7272
if err != nil {
73-
return nil, fmt.Errorf("failed to create claimInfo cache: %+v", err)
73+
return nil, fmt.Errorf("failed to create claimInfo cache: %w", err)
7474
}
7575

7676
// TODO: for now the reconcile period is not configurable.
@@ -158,7 +158,7 @@ func (m *ManagerImpl) PrepareResources(ctx context.Context, pod *v1.Pod) error {
158158
logger.V(3).Info("Processing resource", "pod", klog.KObj(pod), "podClaim", podClaim.Name)
159159
claimName, mustCheckOwner, err := resourceclaim.Name(pod, podClaim)
160160
if err != nil {
161-
return fmt.Errorf("prepare resource claim: %v", err)
161+
return fmt.Errorf("prepare resource claim: %w", err)
162162
}
163163

164164
if claimName == nil {
@@ -172,7 +172,7 @@ func (m *ManagerImpl) PrepareResources(ctx context.Context, pod *v1.Pod) error {
172172
*claimName,
173173
metav1.GetOptions{})
174174
if err != nil {
175-
return fmt.Errorf("failed to fetch ResourceClaim %s referenced by pod %s: %+v", *claimName, pod.Name, err)
175+
return fmt.Errorf("failed to fetch ResourceClaim %s referenced by pod %s: %w", *claimName, pod.Name, err)
176176
}
177177

178178
if mustCheckOwner {
@@ -489,10 +489,10 @@ func (m *ManagerImpl) unprepareResources(ctx context.Context, podUID types.UID,
489489

490490
// PodMightNeedToUnprepareResources returns true if the pod might need to
491491
// unprepare resources
492-
func (m *ManagerImpl) PodMightNeedToUnprepareResources(UID types.UID) bool {
492+
func (m *ManagerImpl) PodMightNeedToUnprepareResources(uid types.UID) bool {
493493
m.cache.Lock()
494494
defer m.cache.Unlock()
495-
return m.cache.hasPodReference(UID)
495+
return m.cache.hasPodReference(uid)
496496
}
497497

498498
// GetContainerClaimInfos gets Container's ClaimInfo

pkg/kubelet/cm/dra/manager_test.go

Lines changed: 17 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ import (
3838
"k8s.io/apimachinery/pkg/util/sets"
3939
"k8s.io/client-go/kubernetes/fake"
4040
"k8s.io/dynamic-resource-allocation/resourceclaim"
41+
"k8s.io/klog/v2"
4142
drapb "k8s.io/kubelet/pkg/apis/dra/v1alpha4"
4243
"k8s.io/kubernetes/pkg/kubelet/cm/dra/plugin"
4344
"k8s.io/kubernetes/pkg/kubelet/cm/dra/state"
@@ -118,7 +119,7 @@ type fakeDRAServerInfo struct {
118119
teardownFn tearDown
119120
}
120121

121-
func setupFakeDRADriverGRPCServer(shouldTimeout bool, pluginClientTimeout *time.Duration, prepareResourcesResponse *drapb.NodePrepareResourcesResponse, unprepareResourcesResponse *drapb.NodeUnprepareResourcesResponse) (fakeDRAServerInfo, error) {
122+
func setupFakeDRADriverGRPCServer(ctx context.Context, shouldTimeout bool, pluginClientTimeout *time.Duration, prepareResourcesResponse *drapb.NodePrepareResourcesResponse, unprepareResourcesResponse *drapb.NodeUnprepareResourcesResponse) (fakeDRAServerInfo, error) {
122123
socketDir, err := os.MkdirTemp("", "dra")
123124
if err != nil {
124125
return fakeDRAServerInfo{
@@ -133,7 +134,10 @@ func setupFakeDRADriverGRPCServer(shouldTimeout bool, pluginClientTimeout *time.
133134

134135
teardown := func() {
135136
close(stopCh)
136-
os.RemoveAll(socketName)
137+
if err := os.Remove(socketName); err != nil {
138+
logger := klog.FromContext(ctx)
139+
logger.Error(err, "failed to remove socket file", "path", socketName)
140+
}
137141
}
138142

139143
l, err := net.Listen("unix", socketName)
@@ -159,11 +163,16 @@ func setupFakeDRADriverGRPCServer(shouldTimeout bool, pluginClientTimeout *time.
159163

160164
drapb.RegisterNodeServer(s, fakeDRADriverGRPCServer)
161165

162-
go func() {
163-
go s.Serve(l)
166+
go func(ctx context.Context) {
167+
go func() {
168+
if err := s.Serve(l); err != nil {
169+
logger := klog.FromContext(ctx)
170+
logger.Error(err, "failed to serve gRPC")
171+
}
172+
}()
164173
<-stopCh
165174
s.GracefulStop()
166-
}()
175+
}(ctx)
167176

168177
return fakeDRAServerInfo{
169178
server: fakeDRADriverGRPCServer,
@@ -565,7 +574,7 @@ func TestPrepareResources(t *testing.T) {
565574
pluginClientTimeout = &timeout
566575
}
567576

568-
draServerInfo, err := setupFakeDRADriverGRPCServer(test.wantTimeout, pluginClientTimeout, test.resp, nil)
577+
draServerInfo, err := setupFakeDRADriverGRPCServer(tCtx, test.wantTimeout, pluginClientTimeout, test.resp, nil)
569578
if err != nil {
570579
t.Fatal(err)
571580
}
@@ -702,7 +711,7 @@ func TestUnprepareResources(t *testing.T) {
702711
pluginClientTimeout = &timeout
703712
}
704713

705-
draServerInfo, err := setupFakeDRADriverGRPCServer(test.wantTimeout, pluginClientTimeout, nil, test.resp)
714+
draServerInfo, err := setupFakeDRADriverGRPCServer(tCtx, test.wantTimeout, pluginClientTimeout, nil, test.resp)
706715
if err != nil {
707716
t.Fatal(err)
708717
}
@@ -872,7 +881,7 @@ func TestParallelPrepareUnprepareResources(t *testing.T) {
872881
tCtx := ktesting.Init(t)
873882

874883
// Setup and register fake DRA driver
875-
draServerInfo, err := setupFakeDRADriverGRPCServer(false, nil, nil, nil)
884+
draServerInfo, err := setupFakeDRADriverGRPCServer(tCtx, false, nil, nil, nil)
876885
if err != nil {
877886
t.Fatal(err)
878887
}

0 commit comments

Comments
 (0)