feat(compose): plan compose deployment

This commit is contained in:
Pavel Sviderski
2025-03-15 20:43:30 +10:00
parent c7355afa3d
commit 34574e41fc
4 changed files with 87 additions and 20 deletions
+6 -8
View File
@@ -12,23 +12,21 @@ type PortsSource []string
// TransformServicesPortsExtension transforms the ports extension of all services in the project by replacing a string
// representation of each port with a parsed PortSpec.
func transformServicesPortsExtension(project *types.Project) error {
for name, service := range project.Services {
func transformServicesPortsExtension(project *types.Project) (*types.Project, error) {
return project.WithServicesTransform(func(name string, service types.ServiceConfig) (types.ServiceConfig, error) {
ports, ok := service.Extensions[PortsExtensionKey].(PortsSource)
if !ok {
continue
return service, nil
}
specs, err := transformPortsExtension(ports)
if err != nil {
return err
return service, err
}
service.Extensions[PortsExtensionKey] = specs
project.Services[name] = service
}
return nil
return service, nil
})
}
func transformPortsExtension(ports PortsSource) ([]api.PortSpec, error) {
+2 -2
View File
@@ -18,7 +18,7 @@ func LoadProject(ctx context.Context, paths []string) (*types.Project, error) {
composecli.WithConfigFileEnv,
// If none was selected, get default compose.yaml file from current dir or parent folders.
composecli.WithDefaultConfigPath,
composecli.WithExtension("x-ports", PortsSource{}),
composecli.WithExtension(PortsExtensionKey, PortsSource{}),
)
if err != nil {
return nil, fmt.Errorf("create compose parser options: %w", err)
@@ -29,7 +29,7 @@ func LoadProject(ctx context.Context, paths []string) (*types.Project, error) {
return nil, err
}
if err = transformServicesPortsExtension(project); err != nil {
if project, err = transformServicesPortsExtension(project); err != nil {
return nil, err
}
+40
View File
@@ -0,0 +1,40 @@
package compose
import (
"fmt"
"github.com/compose-spec/compose-go/v2/types"
"uncloud/internal/api"
)
func ServiceSpecFromCompose(name string, service types.ServiceConfig) (api.ServiceSpec, error) {
spec := api.ServiceSpec{
Container: api.ContainerSpec{
Command: service.Command,
Image: service.Image,
Init: service.Init,
// TODO: env
// TODO: volumes
},
Name: name,
}
if ports, ok := service.Extensions[PortsExtensionKey].([]api.PortSpec); ok {
spec.Ports = ports
}
if service.Deploy != nil {
switch service.Deploy.Mode {
case "global":
spec.Mode = api.ServiceModeGlobal
case "", "replicated":
spec.Mode = api.ServiceModeReplicated
if service.Deploy.Replicas != nil {
spec.Replicas = uint(*service.Deploy.Replicas)
}
default:
return spec, fmt.Errorf("unsupported deploy mode: %s", service.Deploy.Mode)
}
}
return spec, nil
}