117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
package testcontainers
|
|
|
|
import "github.com/docker/docker/api/types/mount"
|
|
|
|
var (
|
|
mountTypeMapping = map[MountType]mount.Type{
|
|
MountTypeBind: mount.TypeBind,
|
|
MountTypeVolume: mount.TypeVolume,
|
|
MountTypeTmpfs: mount.TypeTmpfs,
|
|
MountTypePipe: mount.TypeNamedPipe,
|
|
}
|
|
)
|
|
|
|
// BindMounter can optionally be implemented by mount sources
|
|
// to support advanced scenarios based on mount.BindOptions
|
|
type BindMounter interface {
|
|
GetBindOptions() *mount.BindOptions
|
|
}
|
|
|
|
// VolumeMounter can optionally be implemented by mount sources
|
|
// to support advanced scenarios based on mount.VolumeOptions
|
|
type VolumeMounter interface {
|
|
GetVolumeOptions() *mount.VolumeOptions
|
|
}
|
|
|
|
// TmpfsMounter can optionally be implemented by mount sources
|
|
// to support advanced scenarios based on mount.TmpfsOptions
|
|
type TmpfsMounter interface {
|
|
GetTmpfsOptions() *mount.TmpfsOptions
|
|
}
|
|
|
|
type DockerBindMountSource struct {
|
|
*mount.BindOptions
|
|
|
|
// HostPath is the path mounted into the container
|
|
// the same host path might be mounted to multiple locations within a single container
|
|
HostPath string
|
|
}
|
|
|
|
func (s DockerBindMountSource) Source() string {
|
|
return s.HostPath
|
|
}
|
|
|
|
func (DockerBindMountSource) Type() MountType {
|
|
return MountTypeBind
|
|
}
|
|
|
|
func (s DockerBindMountSource) GetBindOptions() *mount.BindOptions {
|
|
return s.BindOptions
|
|
}
|
|
|
|
type DockerVolumeMountSource struct {
|
|
*mount.VolumeOptions
|
|
|
|
// Name refers to the name of the volume to be mounted
|
|
// the same volume might be mounted to multiple locations within a single container
|
|
Name string
|
|
}
|
|
|
|
func (s DockerVolumeMountSource) Source() string {
|
|
return s.Name
|
|
}
|
|
|
|
func (DockerVolumeMountSource) Type() MountType {
|
|
return MountTypeVolume
|
|
}
|
|
|
|
func (s DockerVolumeMountSource) GetVolumeOptions() *mount.VolumeOptions {
|
|
return s.VolumeOptions
|
|
}
|
|
|
|
type DockerTmpfsMountSource struct {
|
|
GenericTmpfsMountSource
|
|
*mount.TmpfsOptions
|
|
}
|
|
|
|
func (s DockerTmpfsMountSource) GetTmpfsOptions() *mount.TmpfsOptions {
|
|
return s.TmpfsOptions
|
|
}
|
|
|
|
// mapToDockerMounts maps the given []ContainerMount to the corresponding
|
|
// []mount.Mount for further processing
|
|
func mapToDockerMounts(containerMounts ContainerMounts) []mount.Mount {
|
|
mounts := make([]mount.Mount, 0, len(containerMounts))
|
|
|
|
for idx := range containerMounts {
|
|
m := containerMounts[idx]
|
|
|
|
var mountType mount.Type
|
|
if mt, ok := mountTypeMapping[m.Source.Type()]; ok {
|
|
mountType = mt
|
|
} else {
|
|
continue
|
|
}
|
|
|
|
containerMount := mount.Mount{
|
|
Type: mountType,
|
|
Source: m.Source.Source(),
|
|
ReadOnly: m.ReadOnly,
|
|
Target: m.Target.Target(),
|
|
}
|
|
|
|
switch typedMounter := m.Source.(type) {
|
|
case BindMounter:
|
|
containerMount.BindOptions = typedMounter.GetBindOptions()
|
|
case VolumeMounter:
|
|
containerMount.VolumeOptions = typedMounter.GetVolumeOptions()
|
|
case TmpfsMounter:
|
|
containerMount.TmpfsOptions = typedMounter.GetTmpfsOptions()
|
|
}
|
|
|
|
mounts = append(mounts, containerMount)
|
|
}
|
|
|
|
return mounts
|
|
}
|