empty service

Signed-off-by: Yee Hing Tong <[email protected]>

flyteartifacts/cmd/main.go

@@ -0,0 +1,45 @@
+package main
+
+import (
+	"fmt"
+	"github.com/flyteorg/flyte/flyteartifacts/pkg/server"
+	"github.com/golang/glog"
+	"github.com/spf13/cobra"
+	"google.golang.org/grpc"
+
+	"context"
+
+	_ "net/http/pprof" // Required to serve application.
+)
+
+var serveCmd = &cobra.Command{
+	Use:   "serve",
+	Short: "Launches the Flyte artifacts server",
+	RunE: func(cmd *cobra.Command, args []string) error {
+		ctx := context.Background()
+		opts := make([]grpc.ServerOption, 0)
+		return server.Serve(ctx, opts...)
+	},
+}
+
+// RootCmd represents the base command when called without any subcommands
+var RootCmd = &cobra.Command{
+	Use:   "artifacts",
+	Short: "Fill in later",
+	Long: `
+To get started run the serve subcommand which will start a server on localhost:50051
+`,
+}
+
+func init() {
+	// Command information
+	RootCmd.AddCommand(serveCmd)
+}
+
+func main() {
+	glog.V(2).Info("Beginning Flyte Artifacts Service")
+	if err := RootCmd.Execute(); err != nil {
+		fmt.Println(err)
+		panic(err)
+	}
+}

flyteartifacts/go.mod

@@ -0,0 +1,24 @@
+module github.com/flyteorg/flyte/flyteartifacts
+
+go 1.19
+
+require (
+	github.com/flyteorg/flyteidl v0.0.0-00010101000000-000000000000
+	github.com/golang/glog v1.1.0
+	github.com/spf13/cobra v1.4.0
+	google.golang.org/grpc v1.56.1
+
+)
+
+require (
+	github.com/golang/protobuf v1.5.3 // indirect
+	github.com/inconshreveable/mousetrap v1.0.0 // indirect
+	github.com/spf13/pflag v1.0.5 // indirect
+	golang.org/x/net v0.9.0 // indirect
+	golang.org/x/sys v0.7.0 // indirect
+	golang.org/x/text v0.9.0 // indirect
+	google.golang.org/genproto v0.0.0-20230410155749-daa745c078e1 // indirect
+	google.golang.org/protobuf v1.30.0 // indirect
+)
+
+replace github.com/flyteorg/flyteidl => ../flyteidl

flyteartifacts/go.sum

@@ -0,0 +1,32 @@
+github.com/cpuguy83/go-md2man/v2 v2.0.1/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
+github.com/golang/glog v1.1.0 h1:/d3pCKDPWNnvIWe0vVUpNP32qc8U3PDVxySP/y360qE=
+github.com/golang/glog v1.1.0/go.mod h1:pfYeQZ3JWZoXTV5sFc986z3HTpwQs9At6P4ImfuP3NQ=
+github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk=
+github.com/golang/protobuf v1.5.3 h1:KhyjKVUg7Usr/dYsdSqoFveMYd5ko72D+zANwlG1mmg=
+github.com/golang/protobuf v1.5.3/go.mod h1:XVQd3VNwM+JqD3oG2Ue2ip4fOMUkwXdXDdiuN0vRsmY=
+github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
+github.com/inconshreveable/mousetrap v1.0.0 h1:Z8tu5sraLXCXIcARxBp/8cbvlwVa7Z1NHg9XEKhtSvM=
+github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.4.0 h1:y+wJpx64xcgO1V+RcnwW0LEHxTKRi2ZDPSBjWnrg88Q=
+github.com/spf13/cobra v1.4.0/go.mod h1:Wo4iy3BUC+X2Fybo0PDqwJIv3dNRiZLHQymsfxlB84g=
+github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
+github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
+golang.org/x/net v0.9.0 h1:aWJ/m6xSmxWBx+V0XRHTlrYrPG56jKsLdTFmsSsCzOM=
+golang.org/x/net v0.9.0/go.mod h1:d48xBJpPfHeWQsugry2m+kC02ZBRGRgulfHnEXEuWns=
+golang.org/x/sys v0.7.0 h1:3jlCCIQZPdOYu1h8BkNvLz8Kgwtae2cagcG/VamtZRU=
+golang.org/x/sys v0.7.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE=
+golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+google.golang.org/genproto v0.0.0-20230410155749-daa745c078e1 h1:KpwkzHKEF7B9Zxg18WzOa7djJ+Ha5DzthMyZYQfEn2A=
+google.golang.org/genproto v0.0.0-20230410155749-daa745c078e1/go.mod h1:nKE/iIaLqn2bQwXBg8f1g2Ylh6r5MN5CmZvuzZCgsCU=
+google.golang.org/grpc v1.56.1 h1:z0dNfjIl0VpaZ9iSVjA6daGatAYwPGstTjt5vkRMFkQ=
+google.golang.org/grpc v1.56.1/go.mod h1:I9bI3vqKfayGqPUAwGdOSu7kt6oIJLixfffKrpXqQ9s=
+google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw=
+google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
+google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng=
+google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=

flyteartifacts/pkg/server/server.go

@@ -0,0 +1,66 @@
+package server
+
+import (
+	"github.com/flyteorg/flyteidl/gen/pb-go/flyteidl/artifact"
+	"google.golang.org/grpc"
+	"net"
+
+	"context"
+
+	_ "net/http/pprof" // Required to serve application.
+)
+
+type ArtifactService struct {
+	artifact.UnimplementedArtifactRegistryServer
+}
+
+func NewArtifactService() *ArtifactService {
+	return &ArtifactService{}
+}
+
+func Serve(ctx context.Context, opts ...grpc.ServerOption) error {
+	var serverOpts []grpc.ServerOption
+
+	serverOpts = append(serverOpts, grpc.MaxRecvMsgSize(3000000))
+	serverOpts = append(serverOpts, opts...)
+	grpcServer := grpc.NewServer(serverOpts...)
+	server := NewArtifactService()
+
+	artifact.RegisterArtifactRegistryServer(grpcServer, server)
+
+	lis, err := net.Listen("tcp", "localhost:50051")
+	if err != nil {
+		return err
+	}
+	err = grpcServer.Serve(lis)
+
+	return err
+}
+
+func (a *ArtifactService) CreateArtifact(ctx context.Context, request *artifact.CreateArtifactRequest) (*artifact.CreateArtifactResponse, error) {
+	return &artifact.CreateArtifactResponse{}, nil
+}
+
+func (a *ArtifactService) GetArtifact(ctx context.Context, request *artifact.GetArtifactRequest) (*artifact.GetArtifactResponse, error) {
+	return &artifact.GetArtifactResponse{}, nil
+}
+
+func (a *ArtifactService) CreateTrigger(ctx context.Context, request *artifact.CreateTriggerRequest) (*artifact.CreateTriggerResponse, error) {
+	return &artifact.CreateTriggerResponse{}, nil
+}
+
+func (a *ArtifactService) DeleteTrigger(ctx context.Context, request *artifact.DeleteTriggerRequest) (*artifact.DeleteTriggerResponse, error) {
+	return &artifact.DeleteTriggerResponse{}, nil
+}
+
+func (a *ArtifactService) AddTag(ctx context.Context, request *artifact.AddTagRequest) (*artifact.AddTagResponse, error) {
+	return &artifact.AddTagResponse{}, nil
+}
+
+func (a *ArtifactService) RegisterProducer(ctx context.Context, request *artifact.RegisterProducerRequest) (*artifact.RegisterResponse, error) {
+	return &artifact.RegisterResponse{}, nil
+}
+
+func (a *ArtifactService) RegisterConsumer(ctx context.Context, request *artifact.RegisterConsumerRequest) (*artifact.RegisterResponse, error) {
+	return &artifact.RegisterResponse{}, nil
+}