2024-01-27 11:20:40 +01:00
|
|
|
package pods
|
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
|
|
|
|
|
|
|
"github.com/containers/podman/v4/pkg/bindings/containers"
|
|
|
|
"github.com/containers/podman/v4/pkg/specgen"
|
2024-01-28 15:31:58 +01:00
|
|
|
runtime_spec "github.com/opencontainers/runtime-spec/specs-go"
|
|
|
|
"github.com/spf13/viper"
|
2024-01-27 11:20:40 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
func CreateContainer() (string, error) {
|
2024-01-28 15:31:58 +01:00
|
|
|
image := viper.GetString("image")
|
2024-01-27 11:20:40 +01:00
|
|
|
conn := Socket
|
|
|
|
envmap := make(map[string]string)
|
2024-01-28 15:31:58 +01:00
|
|
|
driDevice := runtime_spec.LinuxDevice{
|
|
|
|
Path: viper.GetString("dri_node"),
|
|
|
|
}
|
|
|
|
for envvar, value := range viper.GetStringMap("envvars") {
|
|
|
|
envmap[envvar] = value.(string)
|
|
|
|
}
|
2024-01-27 11:20:40 +01:00
|
|
|
s := specgen.NewSpecGenerator(image, false)
|
|
|
|
s.Env = envmap
|
2024-01-28 15:31:58 +01:00
|
|
|
|
|
|
|
// Experimental DRI support for GPU acceleration in Container
|
|
|
|
if viper.GetBool("dri") == true {
|
|
|
|
s.Devices = []runtime_spec.LinuxDevice{driDevice}
|
|
|
|
}
|
2024-01-27 11:20:40 +01:00
|
|
|
createResponse, err := containers.CreateWithSpec(conn, s, nil)
|
2024-01-28 15:31:58 +01:00
|
|
|
log.Println("Created Container ", createResponse.ID, err, createResponse.Warnings)
|
2024-01-27 11:20:40 +01:00
|
|
|
|
|
|
|
return createResponse.ID, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func StartContainer(id string) error {
|
|
|
|
return containers.Start(Socket, id, nil)
|
|
|
|
}
|
|
|
|
|
|
|
|
func DestroyContainer(id string) error {
|
|
|
|
conn := Socket
|
2024-01-28 15:31:58 +01:00
|
|
|
log.Println("Deleting Container", id)
|
2024-01-27 11:20:40 +01:00
|
|
|
if err := containers.Kill(conn, id, nil); err != nil {
|
|
|
|
log.Println(err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if _, err := containers.Remove(conn, id, nil); err != nil {
|
|
|
|
log.Println(err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func GetContainerIP(id string) (string, error) {
|
|
|
|
conn := Socket
|
|
|
|
|
|
|
|
container, err := containers.Inspect(conn, id, nil)
|
|
|
|
if err != nil {
|
2024-01-28 15:31:58 +01:00
|
|
|
log.Println("Could not get IP of container", err)
|
2024-01-27 11:20:40 +01:00
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
ip := container.NetworkSettings.IPAddress
|
|
|
|
|
|
|
|
return ip, err
|
|
|
|
}
|