152 lines
2.9 KiB
Go
152 lines
2.9 KiB
Go
package runtime
|
|
|
|
import (
|
|
"io"
|
|
)
|
|
|
|
type Option func(o *Options)
|
|
|
|
// Options configure runtime
|
|
type Options struct {
|
|
// Scheduler for updates
|
|
Scheduler Scheduler
|
|
// Service type to manage
|
|
Type string
|
|
// Source of the services repository
|
|
Source string
|
|
// Base image to use
|
|
Image string
|
|
}
|
|
|
|
// WithSource sets the base image / repository
|
|
func WithSource(src string) Option {
|
|
return func(o *Options) {
|
|
o.Source = src
|
|
}
|
|
}
|
|
|
|
// WithScheduler specifies a scheduler for updates
|
|
func WithScheduler(n Scheduler) Option {
|
|
return func(o *Options) {
|
|
o.Scheduler = n
|
|
}
|
|
}
|
|
|
|
// WithType sets the service type to manage
|
|
func WithType(t string) Option {
|
|
return func(o *Options) {
|
|
o.Type = t
|
|
}
|
|
}
|
|
|
|
// WithImage sets the image to use
|
|
func WithImage(t string) Option {
|
|
return func(o *Options) {
|
|
o.Image = t
|
|
}
|
|
}
|
|
|
|
type CreateOption func(o *CreateOptions)
|
|
|
|
type ReadOption func(o *ReadOptions)
|
|
|
|
// CreateOptions configure runtime services
|
|
type CreateOptions struct {
|
|
// Command to execut
|
|
Command []string
|
|
// Args to pass into command
|
|
Args []string
|
|
// Environment to configure
|
|
Env []string
|
|
// Log output
|
|
Output io.Writer
|
|
// Type of service to create
|
|
Type string
|
|
// Retries before failing deploy
|
|
Retries int
|
|
// Specify the image to use
|
|
Image string
|
|
}
|
|
|
|
// ReadOptions queries runtime services
|
|
type ReadOptions struct {
|
|
// Service name
|
|
Service string
|
|
// Version queries services with given version
|
|
Version string
|
|
// Type of service
|
|
Type string
|
|
}
|
|
|
|
// CreateType sets the type of service to create
|
|
func CreateType(t string) CreateOption {
|
|
return func(o *CreateOptions) {
|
|
o.Type = t
|
|
}
|
|
}
|
|
|
|
// CreateImage sets the image to use
|
|
func CreateImage(img string) CreateOption {
|
|
return func(o *CreateOptions) {
|
|
o.Image = img
|
|
}
|
|
}
|
|
|
|
// WithCommand specifies the command to execute
|
|
func WithCommand(cmd ...string) CreateOption {
|
|
return func(o *CreateOptions) {
|
|
// set command
|
|
o.Command = cmd
|
|
}
|
|
}
|
|
|
|
// WithArgs specifies the command to execute
|
|
func WithArgs(args ...string) CreateOption {
|
|
return func(o *CreateOptions) {
|
|
// set command
|
|
o.Args = args
|
|
}
|
|
}
|
|
|
|
// WithRetries sets the max retries attemps
|
|
func WithRetries(retries int) CreateOption {
|
|
return func(o *CreateOptions) {
|
|
o.Retries = retries
|
|
}
|
|
}
|
|
|
|
// WithEnv sets the created service environment
|
|
func WithEnv(env []string) CreateOption {
|
|
return func(o *CreateOptions) {
|
|
o.Env = env
|
|
}
|
|
}
|
|
|
|
// WithOutput sets the arg output
|
|
func WithOutput(out io.Writer) CreateOption {
|
|
return func(o *CreateOptions) {
|
|
o.Output = out
|
|
}
|
|
}
|
|
|
|
// ReadService returns services with the given name
|
|
func ReadService(service string) ReadOption {
|
|
return func(o *ReadOptions) {
|
|
o.Service = service
|
|
}
|
|
}
|
|
|
|
// ReadVersion confifgures service version
|
|
func ReadVersion(version string) ReadOption {
|
|
return func(o *ReadOptions) {
|
|
o.Version = version
|
|
}
|
|
}
|
|
|
|
// ReadType returns services of the given type
|
|
func ReadType(t string) ReadOption {
|
|
return func(o *ReadOptions) {
|
|
o.Type = t
|
|
}
|
|
}
|