Use klog to replace log to keep them in consistence

This commit is contained in:
caiweidong 2020-05-12 13:49:10 +08:00
parent 885e1e929a
commit 5ed8fb690c

View File

@ -18,7 +18,6 @@ package devicemanager
import ( import (
"context" "context"
"log"
"net" "net"
"os" "os"
"path" "path"
@ -27,6 +26,7 @@ import (
"google.golang.org/grpc" "google.golang.org/grpc"
"k8s.io/klog"
pluginapi "k8s.io/kubelet/pkg/apis/deviceplugin/v1beta1" pluginapi "k8s.io/kubelet/pkg/apis/deviceplugin/v1beta1"
watcherapi "k8s.io/kubelet/pkg/apis/pluginregistration/v1" watcherapi "k8s.io/kubelet/pkg/apis/pluginregistration/v1"
) )
@ -108,7 +108,7 @@ func (m *Stub) Start() error {
return err return err
} }
conn.Close() conn.Close()
log.Println("Starting to serve on", m.socket) klog.Infof("Starting to serve on %v", m.socket)
return nil return nil
} }
@ -130,7 +130,7 @@ func (m *Stub) Stop() error {
// GetInfo is the RPC which return pluginInfo // GetInfo is the RPC which return pluginInfo
func (m *Stub) GetInfo(ctx context.Context, req *watcherapi.InfoRequest) (*watcherapi.PluginInfo, error) { func (m *Stub) GetInfo(ctx context.Context, req *watcherapi.InfoRequest) (*watcherapi.PluginInfo, error) {
log.Println("GetInfo") klog.Info("GetInfo")
return &watcherapi.PluginInfo{ return &watcherapi.PluginInfo{
Type: watcherapi.DevicePlugin, Type: watcherapi.DevicePlugin,
Name: m.resourceName, Name: m.resourceName,
@ -144,7 +144,7 @@ func (m *Stub) NotifyRegistrationStatus(ctx context.Context, status *watcherapi.
m.registrationStatus <- *status m.registrationStatus <- *status
} }
if !status.PluginRegistered { if !status.PluginRegistered {
log.Println("Registration failed: ", status.Error) klog.Infof("Registration failed: %v", status.Error)
} }
return &watcherapi.RegistrationStatusResponse{}, nil return &watcherapi.RegistrationStatusResponse{}, nil
} }
@ -153,11 +153,11 @@ func (m *Stub) NotifyRegistrationStatus(ctx context.Context, status *watcherapi.
func (m *Stub) Register(kubeletEndpoint, resourceName string, pluginSockDir string) error { func (m *Stub) Register(kubeletEndpoint, resourceName string, pluginSockDir string) error {
if pluginSockDir != "" { if pluginSockDir != "" {
if _, err := os.Stat(pluginSockDir + "DEPRECATION"); err == nil { if _, err := os.Stat(pluginSockDir + "DEPRECATION"); err == nil {
log.Println("Deprecation file found. Skip registration.") klog.Info("Deprecation file found. Skip registration.")
return nil return nil
} }
} }
log.Println("Deprecation file not found. Invoke registration") klog.Info("Deprecation file not found. Invoke registration")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel() defer cancel()
@ -191,13 +191,13 @@ func (m *Stub) GetDevicePluginOptions(ctx context.Context, e *pluginapi.Empty) (
// PreStartContainer resets the devices received // PreStartContainer resets the devices received
func (m *Stub) PreStartContainer(ctx context.Context, r *pluginapi.PreStartContainerRequest) (*pluginapi.PreStartContainerResponse, error) { func (m *Stub) PreStartContainer(ctx context.Context, r *pluginapi.PreStartContainerRequest) (*pluginapi.PreStartContainerResponse, error) {
log.Printf("PreStartContainer, %+v", r) klog.Infof("PreStartContainer, %+v", r)
return &pluginapi.PreStartContainerResponse{}, nil return &pluginapi.PreStartContainerResponse{}, nil
} }
// ListAndWatch lists devices and update that list according to the Update call // ListAndWatch lists devices and update that list according to the Update call
func (m *Stub) ListAndWatch(e *pluginapi.Empty, s pluginapi.DevicePlugin_ListAndWatchServer) error { func (m *Stub) ListAndWatch(e *pluginapi.Empty, s pluginapi.DevicePlugin_ListAndWatchServer) error {
log.Println("ListAndWatch") klog.Info("ListAndWatch")
s.Send(&pluginapi.ListAndWatchResponse{Devices: m.devs}) s.Send(&pluginapi.ListAndWatchResponse{Devices: m.devs})
@ -218,7 +218,7 @@ func (m *Stub) Update(devs []*pluginapi.Device) {
// Allocate does a mock allocation // Allocate does a mock allocation
func (m *Stub) Allocate(ctx context.Context, r *pluginapi.AllocateRequest) (*pluginapi.AllocateResponse, error) { func (m *Stub) Allocate(ctx context.Context, r *pluginapi.AllocateRequest) (*pluginapi.AllocateResponse, error) {
log.Printf("Allocate, %+v", r) klog.Infof("Allocate, %+v", r)
devs := make(map[string]pluginapi.Device) devs := make(map[string]pluginapi.Device)