refactor(service): split service-related logic into module
This commit is contained in:
parent
6335743afd
commit
a811dd8927
2 changed files with 71 additions and 67 deletions
67
main.go
67
main.go
|
@ -1,76 +1,9 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/goccy/go-yaml"
|
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
"log"
|
|
||||||
"os"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type VolumeDefinition struct {
|
|
||||||
Name string `yaml:"name"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type VolumeConfiguration struct {
|
|
||||||
Name string `yaml:"name"`
|
|
||||||
Container string `yaml:"container"`
|
|
||||||
Host string `yaml:"host"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type ContainerDefinition struct {
|
|
||||||
Name string `yaml:"name"`
|
|
||||||
Image string `yaml:"image"`
|
|
||||||
Volumes []VolumeConfiguration `yaml:"volumes"`
|
|
||||||
ExtraArgs []string `yaml:"extra-args"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type ServiceDefinition struct {
|
|
||||||
Name string `yaml:"name"`
|
|
||||||
Volumes []VolumeDefinition `yaml:"volumes"`
|
|
||||||
Containers []ContainerDefinition `yaml:"containers"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func GetServiceDefinitionFromFile(path string) ServiceDefinition {
|
|
||||||
var definition ServiceDefinition
|
|
||||||
|
|
||||||
defData, err := os.ReadFile(path)
|
|
||||||
|
|
||||||
// TODO: Bubble up error?
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Could not parse service configuration at %s: %s", path, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err = yaml.Unmarshal(defData, &definition); err != nil {
|
|
||||||
log.Fatalf("Could not unpack service configuration: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return definition
|
|
||||||
}
|
|
||||||
|
|
||||||
func CreateService(definition ServiceDefinition) {
|
|
||||||
var err error
|
|
||||||
|
|
||||||
err = CreatePod(definition.Name)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("%s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
knownVolumes := map[string]string{}
|
|
||||||
|
|
||||||
for _, volume := range definition.Volumes {
|
|
||||||
namespacedName := definition.Name + "_" + volume.Name
|
|
||||||
CreateVolume(namespacedName, true)
|
|
||||||
knownVolumes[volume.Name] = namespacedName
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, container := range definition.Containers {
|
|
||||||
if err = CreateContainer(container, knownVolumes, definition.Name); err != nil {
|
|
||||||
log.Fatalf("%s", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var cli = &cobra.Command{
|
var cli = &cobra.Command{
|
||||||
Use: "spud",
|
Use: "spud",
|
||||||
Short: "A not-entirely-terrible-way to manage self-hosted services.",
|
Short: "A not-entirely-terrible-way to manage self-hosted services.",
|
||||||
|
|
71
service.go
Normal file
71
service.go
Normal file
|
@ -0,0 +1,71 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/goccy/go-yaml"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
type VolumeDefinition struct {
|
||||||
|
Name string `yaml:"name"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type VolumeConfiguration struct {
|
||||||
|
Name string `yaml:"name"`
|
||||||
|
Container string `yaml:"container"`
|
||||||
|
Host string `yaml:"host"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ContainerDefinition struct {
|
||||||
|
Name string `yaml:"name"`
|
||||||
|
Image string `yaml:"image"`
|
||||||
|
Volumes []VolumeConfiguration `yaml:"volumes"`
|
||||||
|
ExtraArgs []string `yaml:"extra-args"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type ServiceDefinition struct {
|
||||||
|
Name string `yaml:"name"`
|
||||||
|
Volumes []VolumeDefinition `yaml:"volumes"`
|
||||||
|
Containers []ContainerDefinition `yaml:"containers"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetServiceDefinitionFromFile(path string) ServiceDefinition {
|
||||||
|
var definition ServiceDefinition
|
||||||
|
|
||||||
|
defData, err := os.ReadFile(path)
|
||||||
|
|
||||||
|
// TODO: Bubble up error?
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Could not parse service configuration at %s: %s", path, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = yaml.Unmarshal(defData, &definition); err != nil {
|
||||||
|
log.Fatalf("Could not unpack service configuration: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return definition
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateService(definition ServiceDefinition) {
|
||||||
|
var err error
|
||||||
|
|
||||||
|
err = CreatePod(definition.Name)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("%s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
knownVolumes := map[string]string{}
|
||||||
|
|
||||||
|
for _, volume := range definition.Volumes {
|
||||||
|
namespacedName := definition.Name + "_" + volume.Name
|
||||||
|
CreateVolume(namespacedName, true)
|
||||||
|
knownVolumes[volume.Name] = namespacedName
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, container := range definition.Containers {
|
||||||
|
if err = CreateContainer(container, knownVolumes, definition.Name); err != nil {
|
||||||
|
log.Fatalf("%s", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue