micro/runtime/options.go

177 lines
3.4 KiB
Go
Raw Normal View History

2019-09-24 20:32:35 +03:00
package runtime
2019-09-24 21:00:11 +03:00
import (
"io"
)
type Option func(o *Options)
// Options configure runtime
type Options struct {
2020-01-16 16:34:04 +03:00
// Scheduler for updates
Scheduler Scheduler
2019-11-29 14:35:00 +03:00
// 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
}
}
2020-01-16 16:34:04 +03:00
// WithScheduler specifies a scheduler for updates
func WithScheduler(n Scheduler) Option {
return func(o *Options) {
2020-01-16 16:34:04 +03:00
o.Scheduler = n
}
}
2019-11-29 14:35:00 +03:00
// 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
}
}
2019-09-24 20:32:35 +03:00
type CreateOption func(o *CreateOptions)
2019-11-29 14:35:00 +03:00
type ReadOption func(o *ReadOptions)
// CreateOptions configure runtime services
2019-09-24 20:32:35 +03:00
type CreateOptions struct {
// Command to execut
2019-09-24 20:32:35 +03:00
Command []string
// Args to pass into command
Args []string
2019-09-24 20:32:35 +03:00
// Environment to configure
Env []string
2019-09-24 21:00:11 +03:00
// Log output
Output io.Writer
2019-11-29 14:35:00 +03:00
// Type of service to create
Type string
// Retries before failing deploy
Retries int
// Specify the image to use
Image string
2019-11-29 14:35:00 +03:00
}
// ReadOptions queries runtime services
type ReadOptions struct {
// Service name
Service string
// Version queries services with given version
Version string
// Type of service
Type string
2019-09-24 20:32:35 +03:00
}
2020-01-17 17:14:47 +03:00
// 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
}
}
2019-09-24 21:00:11 +03:00
// 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 {
2019-09-24 20:32:35 +03:00
return func(o *CreateOptions) {
// set command
o.Args = args
2019-09-24 20:32:35 +03:00
}
}
// WithRetries sets the max retries attemps
func WithRetries(retries int) CreateOption {
return func(o *CreateOptions) {
o.Retries = retries
}
}
// WithEnv sets the created service environment
2019-09-24 20:32:35 +03:00
func WithEnv(env []string) CreateOption {
return func(o *CreateOptions) {
o.Env = env
}
}
2019-09-24 21:00:11 +03:00
// WithOutput sets the arg output
func WithOutput(out io.Writer) CreateOption {
return func(o *CreateOptions) {
o.Output = out
}
}
2019-11-29 14:35:00 +03:00
// ReadService returns services with the given name
func ReadService(service string) ReadOption {
return func(o *ReadOptions) {
o.Service = service
}
}
// ReadVersion confifgures service version
2019-11-29 14:35:00 +03:00
func ReadVersion(version string) ReadOption {
return func(o *ReadOptions) {
o.Version = version
}
}
2019-11-29 14:35:00 +03:00
// ReadType returns services of the given type
func ReadType(t string) ReadOption {
return func(o *ReadOptions) {
o.Type = t
}
}
// LogsOption configures runtime logging
type LogsOption func(o *LogsOptions)
// LogsOptions configure runtime logging
type LogsOptions struct {
// How many existing lines to show
Count int64
// Stream new lines?
Stream bool
}
// LogsExistingCount confiures how many existing lines to show
func LogsCount(count int64) LogsOption {
return func(l *LogsOptions) {
l.Count = count
}
}
// LogsStream configures whether to stream new lines
func LogsStream(stream bool) LogsOption {
return func(l *LogsOptions) {
l.Stream = stream
}
}