move reserve/release container name into libkpod
Signed-off-by: Ryan Cole <rcyoalne@gmail.com>
This commit is contained in:
parent
d625e0e468
commit
49ed4ab710
7 changed files with 37 additions and 32 deletions
|
@ -2,8 +2,10 @@ package libkpod
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
|
"github.com/Sirupsen/logrus"
|
||||||
"github.com/containers/image/types"
|
"github.com/containers/image/types"
|
||||||
cstorage "github.com/containers/storage"
|
cstorage "github.com/containers/storage"
|
||||||
"github.com/docker/docker/pkg/registrar"
|
"github.com/docker/docker/pkg/registrar"
|
||||||
|
@ -101,6 +103,28 @@ func (c *ContainerServer) ContainerStateToDisk(ctr *oci.Container) error {
|
||||||
return enc.Encode(c.runtime.ContainerStatus(ctr))
|
return enc.Encode(c.runtime.ContainerStatus(ctr))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ReserveContainerName holds a name for a container that is being created
|
||||||
|
func (c *ContainerServer) ReserveContainerName(id, name string) (string, error) {
|
||||||
|
if err := c.ctrNameIndex.Reserve(name, id); err != nil {
|
||||||
|
if err == registrar.ErrNameReserved {
|
||||||
|
id, err := c.ctrNameIndex.Get(name)
|
||||||
|
if err != nil {
|
||||||
|
logrus.Warnf("conflict, ctr name %q already reserved", name)
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return "", fmt.Errorf("conflict, name %q already reserved for ctr %q", name, id)
|
||||||
|
}
|
||||||
|
return "", fmt.Errorf("error reserving ctr name %s", name)
|
||||||
|
}
|
||||||
|
return name, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReleaseContainerName releases a container name from the index so that it can
|
||||||
|
// be used by other containers
|
||||||
|
func (c *ContainerServer) ReleaseContainerName(name string) {
|
||||||
|
c.ctrNameIndex.Release(name)
|
||||||
|
}
|
||||||
|
|
||||||
type containerServerState struct {
|
type containerServerState struct {
|
||||||
containers oci.ContainerStorer
|
containers oci.ContainerStorer
|
||||||
}
|
}
|
||||||
|
|
|
@ -303,7 +303,7 @@ func (s *Server) CreateContainer(ctx context.Context, req *pb.CreateContainerReq
|
||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.releaseContainerName(containerName)
|
s.ReleaseContainerName(containerName)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
|
|
@ -42,7 +42,7 @@ func (s *Server) RemoveContainer(ctx context.Context, req *pb.RemoveContainerReq
|
||||||
return nil, fmt.Errorf("failed to delete storage for container %s: %v", c.ID(), err)
|
return nil, fmt.Errorf("failed to delete storage for container %s: %v", c.ID(), err)
|
||||||
}
|
}
|
||||||
|
|
||||||
s.releaseContainerName(c.Name())
|
s.ReleaseContainerName(c.Name())
|
||||||
|
|
||||||
if err := s.CtrIDIndex().Delete(c.ID()); err != nil {
|
if err := s.CtrIDIndex().Delete(c.ID()); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
|
|
@ -66,7 +66,7 @@ func (s *Server) generateContainerIDandNameForSandbox(sandboxConfig *pb.PodSandb
|
||||||
err error
|
err error
|
||||||
id = stringid.GenerateNonCryptoID()
|
id = stringid.GenerateNonCryptoID()
|
||||||
)
|
)
|
||||||
name, err := s.reserveContainerName(id, makeSandboxContainerName(sandboxConfig))
|
name, err := s.ReserveContainerName(id, makeSandboxContainerName(sandboxConfig))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", "", err
|
return "", "", err
|
||||||
}
|
}
|
||||||
|
@ -78,7 +78,7 @@ func (s *Server) generateContainerIDandName(sandboxMetadata *pb.PodSandboxMetada
|
||||||
err error
|
err error
|
||||||
id = stringid.GenerateNonCryptoID()
|
id = stringid.GenerateNonCryptoID()
|
||||||
)
|
)
|
||||||
name, err := s.reserveContainerName(id, makeContainerName(sandboxMetadata, containerConfig))
|
name, err := s.ReserveContainerName(id, makeContainerName(sandboxMetadata, containerConfig))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", "", err
|
return "", "", err
|
||||||
}
|
}
|
||||||
|
|
|
@ -65,7 +65,7 @@ func (s *Server) RemovePodSandbox(ctx context.Context, req *pb.RemovePodSandboxR
|
||||||
return nil, fmt.Errorf("failed to delete container %s in pod sandbox %s: %v", c.Name(), sb.ID(), err)
|
return nil, fmt.Errorf("failed to delete container %s in pod sandbox %s: %v", c.Name(), sb.ID(), err)
|
||||||
}
|
}
|
||||||
|
|
||||||
s.releaseContainerName(c.Name())
|
s.ReleaseContainerName(c.Name())
|
||||||
s.removeContainer(c)
|
s.removeContainer(c)
|
||||||
if err := s.CtrIDIndex().Delete(c.ID()); err != nil {
|
if err := s.CtrIDIndex().Delete(c.ID()); err != nil {
|
||||||
return nil, fmt.Errorf("failed to delete container %s in pod sandbox %s from index: %v", c.Name(), sb.ID(), err)
|
return nil, fmt.Errorf("failed to delete container %s in pod sandbox %s from index: %v", c.Name(), sb.ID(), err)
|
||||||
|
@ -82,7 +82,7 @@ func (s *Server) RemovePodSandbox(ctx context.Context, req *pb.RemovePodSandboxR
|
||||||
return nil, fmt.Errorf("failed to remove pod sandbox %s: %v", sb.ID(), err)
|
return nil, fmt.Errorf("failed to remove pod sandbox %s: %v", sb.ID(), err)
|
||||||
}
|
}
|
||||||
|
|
||||||
s.releaseContainerName(podInfraContainer.Name())
|
s.ReleaseContainerName(podInfraContainer.Name())
|
||||||
if err := s.CtrIDIndex().Delete(podInfraContainer.ID()); err != nil {
|
if err := s.CtrIDIndex().Delete(podInfraContainer.ID()); err != nil {
|
||||||
return nil, fmt.Errorf("failed to delete infra container %s in pod sandbox %s from index: %v", podInfraContainer.ID(), sb.ID(), err)
|
return nil, fmt.Errorf("failed to delete infra container %s in pod sandbox %s from index: %v", podInfraContainer.ID(), sb.ID(), err)
|
||||||
}
|
}
|
||||||
|
|
|
@ -149,7 +149,7 @@ func (s *Server) RunPodSandbox(ctx context.Context, req *pb.RunPodSandboxRequest
|
||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.releaseContainerName(containerName)
|
s.ReleaseContainerName(containerName)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
|
|
@ -103,14 +103,14 @@ func (s *Server) loadContainer(id string) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
name := m.Annotations[annotations.Name]
|
name := m.Annotations[annotations.Name]
|
||||||
name, err = s.reserveContainerName(id, name)
|
name, err = s.ReserveContainerName(id, name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.releaseContainerName(name)
|
s.ReleaseContainerName(name)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
@ -256,13 +256,13 @@ func (s *Server) loadSandbox(id string) error {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
cname, err := s.reserveContainerName(m.Annotations[annotations.ContainerID], m.Annotations[annotations.ContainerName])
|
cname, err := s.ReserveContainerName(m.Annotations[annotations.ContainerID], m.Annotations[annotations.ContainerName])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
defer func() {
|
defer func() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.releaseContainerName(cname)
|
s.ReleaseContainerName(cname)
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
@ -384,7 +384,7 @@ func (s *Server) update() error {
|
||||||
logrus.Warnf("bad state when getting container removed %+v", removedPodContainer)
|
logrus.Warnf("bad state when getting container removed %+v", removedPodContainer)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
s.releaseContainerName(c.Name())
|
s.ReleaseContainerName(c.Name())
|
||||||
s.removeContainer(c)
|
s.removeContainer(c)
|
||||||
if err = s.CtrIDIndex().Delete(c.ID()); err != nil {
|
if err = s.CtrIDIndex().Delete(c.ID()); err != nil {
|
||||||
return err
|
return err
|
||||||
|
@ -405,7 +405,7 @@ func (s *Server) update() error {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
podInfraContainer := sb.InfraContainer()
|
podInfraContainer := sb.InfraContainer()
|
||||||
s.releaseContainerName(podInfraContainer.Name())
|
s.ReleaseContainerName(podInfraContainer.Name())
|
||||||
s.removeContainer(podInfraContainer)
|
s.removeContainer(podInfraContainer)
|
||||||
if err = s.CtrIDIndex().Delete(podInfraContainer.ID()); err != nil {
|
if err = s.CtrIDIndex().Delete(podInfraContainer.ID()); err != nil {
|
||||||
return err
|
return err
|
||||||
|
@ -456,25 +456,6 @@ func (s *Server) releasePodName(name string) {
|
||||||
s.podNameIndex.Release(name)
|
s.podNameIndex.Release(name)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) reserveContainerName(id, name string) (string, error) {
|
|
||||||
if err := s.CtrNameIndex().Reserve(name, id); err != nil {
|
|
||||||
if err == registrar.ErrNameReserved {
|
|
||||||
id, err := s.CtrNameIndex().Get(name)
|
|
||||||
if err != nil {
|
|
||||||
logrus.Warnf("conflict, ctr name %q already reserved", name)
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
return "", fmt.Errorf("conflict, name %q already reserved for ctr %q", name, id)
|
|
||||||
}
|
|
||||||
return "", fmt.Errorf("error reserving ctr name %s", name)
|
|
||||||
}
|
|
||||||
return name, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Server) releaseContainerName(name string) {
|
|
||||||
s.CtrNameIndex().Release(name)
|
|
||||||
}
|
|
||||||
|
|
||||||
// cleanupSandboxesOnShutdown Remove all running Sandboxes on system shutdown
|
// cleanupSandboxesOnShutdown Remove all running Sandboxes on system shutdown
|
||||||
func (s *Server) cleanupSandboxesOnShutdown() {
|
func (s *Server) cleanupSandboxesOnShutdown() {
|
||||||
_, err := os.Stat(shutdownFile)
|
_, err := os.Stat(shutdownFile)
|
||||||
|
|
Loading…
Reference in a new issue