fix review.

pkg/agent/config/config.go

@@ -161,6 +161,7 @@ func get(envInfo *cmds.Agent) (*config.Node, error) {
 		ContainerRuntimeEndpoint: envInfo.ContainerRuntimeEndpoint,
 	}
 	nodeConfig.LocalAddress = localAddress(controlConfig)
+	nodeConfig.Images = "/var/lib/rancher/k3s/agent/images"
 	nodeConfig.AgentConfig.NodeIP = nodeIP
 	nodeConfig.AgentConfig.NodeName = nodeName
 	nodeConfig.AgentConfig.ClusterDNS = controlConfig.ClusterDNS

pkg/agent/containerd/containerd.go

@@ -1,7 +1,6 @@
 package containerd
 
 import (
-	"bytes"
 	"context"
 	"fmt"
 	"github.com/containerd/containerd"
@@ -148,13 +147,13 @@ func Run(ctx context.Context, cfg *config.Node) error {
 			for _, fileInfo := range fileInfos {
 				if !fileInfo.IsDir() {
 					filePath := filepath.Join(imageDir, fileInfo.Name())
-					fileContent, err := ioutil.ReadFile(filePath)
+					file, err := os.Open(filePath)
 					if err != nil {
 						logrus.Errorf("Unable to read %s: %v", filePath, err)
 						continue
 					}
 					logrus.Debugf("Import %s", filePath)
-					_, err = client.Import(ctxContainerD, bytes.NewReader(fileContent))
+					_, err = client.Import(ctxContainerD, file)
 					if err != nil {
 						logrus.Errorf("Unable to import %s: %v", filePath, err)
 					}

pkg/daemons/config/types.go

@@ -16,6 +16,7 @@ type Node struct {
 	FlannelConf              string
 	LocalAddress             string
 	Containerd               Containerd
+	Images                   string
 	AgentConfig              Agent
 	CACerts                  []byte
 	ServerAddress            string