f8f17caf78
- new special internal role `initServer` used only to determine the correct ready-log-message - ready-log-messages now looked up by role and new `Intent` type (cluster-create/cluster-start/node-create/node-start), as especially for the init server there are different log messages indicating that we can proceed with the next step - moving types around: - K3s env vars now under .../types/k3s/env.go - defaults now under .../types/defaults.go - ... - improved waiting for log messages - not checking the whole log again and again in a loop - follow log with a single reader (and retry in case we see a fatal error, meaning that the K3s container will restart -> backoff after 10 tries) - BREAKING: new `*runtimeTypes.NodeLogsOpts` parameter in GetNodeLogs
90 lines
4.1 KiB
Go
90 lines
4.1 KiB
Go
/*
|
|
Copyright © 2020-2021 The k3d Author(s)
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
*/
|
|
package runtimes
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/rancher/k3d/v5/pkg/runtimes/docker"
|
|
runtimeTypes "github.com/rancher/k3d/v5/pkg/runtimes/types"
|
|
k3d "github.com/rancher/k3d/v5/pkg/types"
|
|
)
|
|
|
|
// SelectedRuntime is a runtime (pun intended) variable determining the selected runtime
|
|
var SelectedRuntime Runtime = docker.Docker{}
|
|
|
|
// Docker docker
|
|
var Docker = docker.Docker{}
|
|
|
|
// Runtimes defines a map of implemented k3d runtimes
|
|
var Runtimes = map[string]Runtime{
|
|
"docker": docker.Docker{},
|
|
}
|
|
|
|
// Runtime defines an interface that can be implemented for various container runtime environments (docker, containerd, etc.)
|
|
type Runtime interface {
|
|
ID() string
|
|
GetHost() string
|
|
CreateNode(context.Context, *k3d.Node) error
|
|
DeleteNode(context.Context, *k3d.Node) error
|
|
RenameNode(context.Context, *k3d.Node, string) error
|
|
GetNodesByLabel(context.Context, map[string]string) ([]*k3d.Node, error)
|
|
GetNode(context.Context, *k3d.Node) (*k3d.Node, error)
|
|
GetNodeStatus(context.Context, *k3d.Node) (bool, string, error)
|
|
GetNodesInNetwork(context.Context, string) ([]*k3d.Node, error)
|
|
CreateNetworkIfNotPresent(context.Context, *k3d.ClusterNetwork) (*k3d.ClusterNetwork, bool, error) // @param context, name - @return NETWORK, EXISTS, ERROR
|
|
GetKubeconfig(context.Context, *k3d.Node) (io.ReadCloser, error)
|
|
DeleteNetwork(context.Context, string) error
|
|
StartNode(context.Context, *k3d.Node) error // starts an existing container
|
|
StopNode(context.Context, *k3d.Node) error
|
|
CreateVolume(context.Context, string, map[string]string) error
|
|
DeleteVolume(context.Context, string) error
|
|
GetVolume(string) (string, error)
|
|
GetRuntimePath() string // returns e.g. '/var/run/docker.sock' for a default docker setup
|
|
ExecInNode(context.Context, *k3d.Node, []string) error
|
|
ExecInNodeGetLogs(context.Context, *k3d.Node, []string) (*bufio.Reader, error)
|
|
GetNodeLogs(context.Context, *k3d.Node, time.Time, *runtimeTypes.NodeLogsOpts) (io.ReadCloser, error)
|
|
GetImages(context.Context) ([]string, error)
|
|
CopyToNode(context.Context, string, string, *k3d.Node) error // @param context, source, destination, node
|
|
WriteToNode(context.Context, []byte, string, os.FileMode, *k3d.Node) error // @param context, content, destination, filemode, node
|
|
ReadFromNode(context.Context, string, *k3d.Node) (io.ReadCloser, error) // @param context, filepath, node
|
|
GetHostIP(context.Context, string) (net.IP, error)
|
|
ConnectNodeToNetwork(context.Context, *k3d.Node, string) error // @param context, node, network name
|
|
DisconnectNodeFromNetwork(context.Context, *k3d.Node, string) error // @param context, node, network name
|
|
Info() (*runtimeTypes.RuntimeInfo, error)
|
|
GetNetwork(context.Context, *k3d.ClusterNetwork) (*k3d.ClusterNetwork, error) // @param context, network (so we can filter by name or by id)
|
|
}
|
|
|
|
// GetRuntime checks, if a given name is represented by an implemented k3d runtime and returns it
|
|
func GetRuntime(rt string) (Runtime, error) {
|
|
if runtime, ok := Runtimes[rt]; ok {
|
|
return runtime, nil
|
|
}
|
|
return nil, fmt.Errorf("Runtime '%s' not supported", rt)
|
|
}
|