machine

package module
v0.14.2 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Mar 23, 2021 License: MIT Imports: 22 Imported by: 13

README

Go PkgGoDev GoDoc Go Report Card Codacy Badge Codacy Badge Version Badge Open in Gitpod

Machine

Machine is a library for creating data workflows. These workflows can be either very concise or quite complex, even allowing for cycles for flows that need retry or self healing mechanisms. It supports opentelemetry spans and metrics out of the box and supports building dynamic pipelines using native go plugins and hashicorp or yaegi based plugins by using the providers here.

Installation

Add the primary library to your project

  go get -u github.com/whitaker-io/machine

Foundry is a tool used to generate new projects quickly

Components is a repository of different vertex implementations

Documentation

Gopher

Docs

Example

Redis Subscription with basic receive -> process -> send Stream

  // logStore allows for running a cluster and handles communication
  var logStore machine.LogStore

  // pool is a redigo Pool for a redis cluster to read the stream from
  // see also the Google Pub/Sub, Kafka, and SQS implementations
  var pool *redigo.Pool

  // publisher is a machine.Publisher used for sending data outside of the Stream
  var publisher machine.Publisher

  redisStream := redis.New(pool, logger)
  
  // NewPipe creates a pipe in which you can run multiple streams
  // the id is the instance identifier for the cluster
  p := NewPipe(uuid.New().String(), nil, logStore, fiber.Config{
    ReadTimeout: time.Second,
    WriteTimeout: time.Second,
    BodyLimit: 4 * 1024 * 1024,
    DisableKeepalive: true,
  })

  // StreamSubscription takes an instance of machine.Subscription
  // and a time interval in which to read
  // the id here needs to be the same for all the nodes for the clustering to work
  builder := p.StreamSubscription("unique_stream_id", redisStream, 5*time.Millisecond,
    &Option{FIFO: boolP(false)},
    &Option{Injectable: boolP(true)},
    &Option{Metrics: boolP(true)},
    &Option{Span: boolP(false)},
    &Option{BufferSize: intP(0)},
  ).Builder()

  builder.Map("unique_id2", 
      func(m Data) error {
        var err error

        // ...do some processing

        return err
      },
    ).
    Publish("unique_id3", publisher)

  // Run requires a context, the port to run the fiber.App,
  // and the timeout for graceful shutdown
  if err := p.Run(context.Background(), ":5000", 10 * time.Second); err != nil {
    // Run will return an error in the case that 
    // one of the paths is not terminated (i.e. missing a Transmit)
    panic(err)
  }

🤝 Contributing

Contributions, issues and feature requests are welcome.
Feel free to check issues page if you want to contribute.
Check the contributing guide.

Author

👤 Jonathan Whitaker

Show your support

Please ⭐️ this repository if this project helped you!

License

Machine is provided under the MIT License.

The MIT License (MIT)

Copyright (c) 2020 Jonathan Whitaker

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func RegisterPluginProvider added in v0.11.0 

func RegisterPluginProvider(name string, p PluginProvider)

RegisterPluginProvider function for registering a PluginProvider to be used for loading VertexProviders

Types

type Applicative added in v0.3.0 

type Applicative func(data Data) error

Applicative is a function that is applied on an individual basis for each Packet in the payload. The data may be modified or checked for correctness. Any resulting error is combined with current errors in the wrapping Packet.

type Builder 

type Builder interface {
	Map(id string, a Applicative, options ...*Option) Builder
	FoldLeft(id string, f Fold, options ...*Option) Builder
	FoldRight(id string, f Fold, options ...*Option) Builder
	Fork(id string, f Fork, options ...*Option) (Builder, Builder)
	Loop(id string, x Fork, options ...*Option) (loop LoopBuilder, out Builder)

	Publish(id string, s Publisher, options ...*Option)
	// contains filtered or unexported methods
}

Builder is the interface provided for creating a data processing stream.

type Data added in v0.4.0 

type Data map[string]interface{}

Data is a map[string]interface{} used to hold the data flowing into the Stream it can be converted to a struct using the As method, though this is temporary and can be costly.

func (*Data) As added in v0.9.11 

func (d *Data) As(i interface{}) error

As helper function used for converting the Data into the given struct or map uses github.com/mitchellh/mapstructure under the covers and input must be a pointer to a map or struct

type Fold added in v0.5.0 

type Fold func(aggregate, next Data) Data

Fold is a function used to combine a payload into a single Packet. It may be used with either a Fold Left or Fold Right operation, which starts at the corresponding side and moves through the payload. The returned instance of Data is used as the aggregate in the subsequent call.

type Fork added in v0.5.0 

type Fork func(list []*Packet) (a, b []*Packet)

Fork is a function for splitting the payload into 2 separate paths. Default Forks for duplication and error checking are provided by ForkDuplicate and ForkError respectively. 

var (
	// ForkDuplicate is a Fork that creates a deep copy of the
	// payload and sends it down both branches.
	ForkDuplicate Fork = func(payload []*Packet) (a, b []*Packet) {
		payload2 := []*Packet{}
		buf := &bytes.Buffer{}
		enc, dec := gob.NewEncoder(buf), gob.NewDecoder(buf)

		_ = enc.Encode(payload)
		_ = dec.Decode(&payload2)

		for i, packet := range payload {
			payload2[i].span = packet.span
		}

		return payload, payload2
	}

	// ForkError is a Fork that splits machine.Packets based on
	// the presence of an error. Errors are sent down the
	// right side path.
	ForkError Fork = func(payload []*Packet) (s, f []*Packet) {
		s = []*Packet{}
		f = []*Packet{}

		for _, packet := range payload {
			if packet.Error != nil {
				f = append(f, packet)
			} else {
				s = append(s, packet)
			}
		}

		return s, f
	}
)

type ForkRule added in v0.5.0 

type ForkRule func(data Data) bool

ForkRule is a function that can be converted to a Fork via the Handler method allowing for Forking based on the contents of the data.

func (ForkRule) Handler added in v0.5.0 

func (r ForkRule) Handler(payload []*Packet) (t, f []*Packet)

Handler is a method for turning the ForkRule into an instance of Fork

type HealthInfo added in v0.7.0 

type HealthInfo struct {
	StreamID    string    `json:"stream_id"`
	LastPayload time.Time `json:"last_payload"`
	// contains filtered or unexported fields
}

HealthInfo is the type used for providing basic healthcheck information about last start time of payloads

type InjectionCallback added in v0.7.0 

type InjectionCallback func(logs ...*Log)

InjectionCallback is a function provided to the LogStore Join method so that the cluster may restart work that has been dropped by one of the workers. Injections will only be processed for vertices that have the Injectable option set to true, which is the default.

type Log added in v0.7.0 

type Log struct {
	OwnerID    string    `json:"owner_id"`
	StreamID   string    `json:"stream_id"`
	VertexID   string    `json:"vertex_id"`
	VertexType string    `json:"vertex_type"`
	State      string    `json:"state"`
	Packet     *Packet   `json:"packet"`
	When       time.Time `json:"when"`
}

Log type for holding the data that is recorded from the streams and sent to the LogStore instance

type LogStore added in v0.7.0 

type LogStore interface {
	Join(id string, callback InjectionCallback, streamIDs ...string) error
	Write(logs ...*Log)
	Leave(id string) error
}

LogStore is an interface for allowing a distributed cluster of workers It requires 3 methods Join, Write, and Leave.

Join is called during the run method of the Pipe and it is used for accouncing membership to the cluster with an identifier, a callback for injection, and the list of Streams that will be running in the pipe. Injection is how work can be restarted in the system and is the responsibility of the implementation to decide when and how it is reinitiated.

Write is called at the beginning of every vertex and provides the current payload about to be run. The implementation is considered the owner of that payload and may modify the data in special known circumstances if need be.

Leave is called during a graceful termination and any errors are logged.

type LoopBuilder added in v0.9.15 

type LoopBuilder interface {
	Map(id string, a Applicative, options ...*Option) LoopBuilder
	FoldLeft(id string, f Fold, options ...*Option) LoopBuilder
	FoldRight(id string, f Fold, options ...*Option) LoopBuilder
	Fork(id string, f Fork, options ...*Option) (LoopBuilder, LoopBuilder)
	Loop(id string, x Fork, options ...*Option) (loop LoopBuilder, out LoopBuilder)
	Publish(id string, s Publisher, options ...*Option)
	Done()
}

LoopBuilder is the interface provided for creating a data processing stream that loops.

type Option added in v0.2.0 

type Option struct {
	// DeepCopy uses encoding/gob to create a deep copy of the payload
	// before the processing to ensure concurrent map exceptions cannot
	// happen. Is fairly resource intensive so use with caution.
	// Default: false
	DeepCopy *bool
	// FIFO controls the processing order of the payloads
	// If set to true the system will wait for one payload
	// to be processed before starting the next.
	// Default: false
	FIFO *bool
	// Injectable controls whether the vertex accepts injection calls
	// if set to false the data will be logged and processing will not
	// take place.
	// Default: true
	Injectable *bool
	// BufferSize sets the buffer size on the edge channels between the
	// vertices, this setting can be useful when processing large amounts
	// of data with FIFO turned on.
	// Default: 0
	BufferSize *int
	// Span controls whether opentelemetry spans are created for tracing
	// Packets processed by the system.
	// Default: true
	Span *bool
	// Metrics controls whether opentelemetry metrics are recorded for
	// Packets processed by the system.
	// Default: true
	Metrics *bool
}

Option type for holding machine settings.

type Packet 

type Packet struct {
	ID    string `json:"id"`
	Data  Data   `json:"data"`
	Error error  `json:"error"`
	// contains filtered or unexported fields
}

Packet type that holds information traveling through the machine.

func (*Packet) As added in v0.9.11 

func (p *Packet) As(i interface{}) error

As helper function used for converting the Packet into the given struct or map uses github.com/mitchellh/mapstructure under the covers and input must be a pointer to a map or struct

type Pipe added in v0.7.0 

type Pipe struct {
	// contains filtered or unexported fields
}

Pipe is the representation of the system. It can run multiple Streams and controls the start and stop functionality of the system.

func NewPipe added in v0.7.0 

func NewPipe(id string, logger *logrus.Logger, store LogStore, config ...fiber.Config) *Pipe

NewPipe is a function for creating a new Pipe. If logger or logStore are nil then the accosiated feature will be disabled.

func (*Pipe) Add added in v0.14.2 

func (pipe *Pipe) Add(method, path string, handlers ...func(*fiber.Ctx) error) fiber.Router

Add Wraps fiber.App.Add

Add allows you to specify a HTTP method to register a route

func (*Pipe) Load added in v0.9.0 

func (pipe *Pipe) Load(streams []*StreamSerialization) error

Load method loads a stream based on the StreamSerialization

func (*Pipe) Run added in v0.7.0 

func (pipe *Pipe) Run(ctx context.Context, port string, gracePeriod time.Duration) error

Run starts the Pipe and subsequent Streams. It requires a context, a port to run an instance of fiber.App which hosts the /health endpoint and any HTTP based streams at /strea/:id, and a gracePeriod for which graceful shutdown can take place.

func (*Pipe) Stream added in v0.8.0 

func (pipe *Pipe) Stream(stream Stream) Builder

Stream is a method for adding a generic developer defined Stream. New Streams are created by the appropriately named NewStream function.

func (*Pipe) StreamHTTP added in v0.7.0 

func (pipe *Pipe) StreamHTTP(id string, opts ...*Option) Builder

StreamHTTP a method that creates a Stream at the path /stream/:id which is hosted by the Pipe's fiber.App

func (*Pipe) StreamSubscription added in v0.7.0 

func (pipe *Pipe) StreamSubscription(id string, sub Subscription, interval time.Duration, opts ...*Option) Builder

StreamSubscription is a method for creating a Stream based on the provided Subscription which has it's Read method called at the end of each interval period.

func (*Pipe) StreamWebsocket added in v0.14.0 

func (pipe *Pipe) StreamWebsocket(id string, opts ...*Option) Builder

StreamWebsocket a method that creates a Stream at the path /ws/:id which is hosted by the Pipe's fiber.App

func (*Pipe) Use added in v0.7.0 

func (pipe *Pipe) Use(args ...interface{}) fiber.Router

Use Wraps fiber.App.Use

Use registers a middleware route that will match requests with the provided prefix (which is optional and defaults to "/"). 

app.Use(func(c *fiber.Ctx) error {
   return c.Next()
})
app.Use("/api", func(c *fiber.Ctx) error {
   return c.Next()
})
app.Use("/api", handler, func(c *fiber.Ctx) error {
   return c.Next()
})

This method will match all HTTP verbs: GET, POST, PUT, HEAD etc...

type PluginDefinition added in v0.11.0 

type PluginDefinition struct {
	// Type is the name of the PluginProvider to use.
	Type string `json:"type" mapstructure:"type"`
	// Payload is the location, script, etc provided to load the plugin.
	// Depends on the PluginProvider.
	Payload string `json:"payload" mapstructure:"payload"`
	// Symbol is the name of the symbol to be loaded from the plugin.
	Symbol string `json:"symbol" mapstructure:"symbol"`
	// Attributes are a map[string]interface{} of properties to be used with the PluginProvider.
	Attributes map[string]interface{} `json:"attributes" mapstructure:"attributes"`
}

PluginDefinition type for declaring the path and symbol for a golang plugin containing the Provider

type PluginProvider added in v0.11.0 

type PluginProvider interface {
	Load(*PluginDefinition) (interface{}, error)
}

PluginProvider interface for providing a way of loading plugins must return one of the following types: 

Subscription
Retriever
Applicative
Fold
Fork
Publisher

type Publisher added in v0.12.0 

type Publisher interface {
	Send([]Data) error
}

Publisher is an interface for sending data out of the Stream

type Retriever added in v0.3.0 

type Retriever func(ctx context.Context) chan []Data

Retriever is a function that provides data to a generic Stream must stop when the context receives a done signal.

type Stream added in v0.5.0 

type Stream interface {
	ID() string
	Run(ctx context.Context, recorders ...recorder) error
	Inject(ctx context.Context, events map[string][]*Packet)
	Builder() Builder
}

Stream is a representation of a data stream and its associated logic. It may be used individually or hosted by a Pipe. Creating a new Stream is handled by the appropriately named NewStream function.

The Builder method is the entrypoint into creating the data processing flow. All branches of the Stream are required to end in either a Publish or a Link in order to be considered valid.

func NewStream added in v0.5.0 

func NewStream(id string, retriever Retriever, options ...*Option) Stream

NewStream is a function for creating a new Stream. It takes an id, a Retriever function, and a list of Options that can override the defaults and set new defaults for the subsequent vertices in the Stream.

type StreamSerialization added in v0.11.0 

type StreamSerialization struct {
	// Type type of stream to create.
	//
	// For root serializations valid values are 'http', 'subscription', or 'stream'.
	Type string `json:"type,omitempty" mapstructure:"type,omitempty"`
	// Interval is the duration in nanoseconds between pulls in a 'subscription' Type. It is only read
	// if the Type is 'subscription'.
	Interval time.Duration `json:"interval,omitempty" mapstructure:"interval,omitempty"`

	*VertexSerialization
}

StreamSerialization config based definition for a stream

func (*StreamSerialization) MarshalJSON added in v0.12.4 

func (s *StreamSerialization) MarshalJSON() ([]byte, error)

MarshalJSON implementation to marshal json

func (*StreamSerialization) MarshalYAML added in v0.11.0 

func (s *StreamSerialization) MarshalYAML() (interface{}, error)

MarshalYAML implementation to marshal yaml

func (*StreamSerialization) UnmarshalJSON added in v0.12.4 

func (s *StreamSerialization) UnmarshalJSON(data []byte) error

UnmarshalJSON implementation to unmarshal json

func (*StreamSerialization) UnmarshalYAML added in v0.11.0 

func (s *StreamSerialization) UnmarshalYAML(unmarshal func(interface{}) error) error

UnmarshalYAML implementation to unmarshal yaml

type Subscription added in v0.7.0 

type Subscription interface {
	Read(ctx context.Context) []Data
	Close() error
}

Subscription is an interface for creating a pull based stream. It requires 2 methods Read and Close.

Read is called when the interval passes and the resulting payload is sent down the Stream.

Close is called during a graceful termination and any errors are logged.

type VertexSerialization added in v0.11.0 

type VertexSerialization struct {
	// ID unique identifier for the stream.
	ID string `json:"id,omitempty" mapstructure:"id,omitempty"`
	// Provider Plugin information to load
	Provider *PluginDefinition `json:"provider,omitempty" mapstructure:"provider,omitempty"`
	// Options are a slice of machine.Option https://godoc.org/github.com/whitaker-io/machine#Option
	Options []*Option `json:"options,omitempty" mapstructure:"options,omitempty"`
	// contains filtered or unexported fields
}

VertexSerialization config based definition for a stream vertex

Source Files

View all Source files

Directories

Path Synopsis
cmd module
common module
edge
http Module
pubsub Module
telemetry module

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.