Sample Heptio Ark plugin

main.go

package main

import (
  plugin "github.com/hashicorp/go-plugin"
  "github.com/heptio/ark/pkg/cloudprovider"
  "github.com/heptio/ark/pkg/cloudprovider/aws"
  arkplugin "github.com/heptio/ark/pkg/plugin"
  "github.com/sirupsen/logrus"
)

func main() {
  objectStore := &MyObjectStore{
    ObjectStore: aws.NewObjectStore(),
    log:         arkplugin.NewPluginLogger(),
  }

  plugin.Serve(&plugin.ServeConfig{
    HandshakeConfig: arkplugin.Handshake,
    Plugins: map[string]plugin.Plugin{
      string(arkplugin.PluginKindBlockStore): arkplugin.NewObjectStorePlugin(objectStore),
    },
    GRPCServer: plugin.DefaultGRPCServer,
  })
}

type MyObjectStore struct {
  cloudprovider.ObjectStore
  log logrus.FieldLogger
}

func (mos *MyObjectStore) Init(config map[string]string) error {
  mos.log.Info("HELLO FROM MyObjectStore!!!")
  return mos.ObjectStore.Init(config)
}