server

Documentation

Overview

Package server is the SQLFlow grpc server which connects to database and parse, submit or execute the training and predicting codes.

To generate grpc protobuf code, run the below command:

Index

• func SubmitWorkflow(sqlProgram string, modelDir string, session *pb.Session) *pipe.Reader
• func WaitPortReady(addr string, timeout time.Duration)
• type Server
  • func NewServer(run func(string, string, *pb.Session) *pipe.Reader, modelDir string) *Server
  • func (s *Server) Fetch(ctx context.Context, job *pb.FetchRequest) (*pb.FetchResponse, error)
  • func (s *Server) Run(req *pb.Request, stream pb.SQLFlow_RunServer) error

Functions

func SubmitWorkflow

func SubmitWorkflow(sqlProgram string, modelDir string, session *pb.Session) *pipe.Reader

SubmitWorkflow submits an Argo workflow

TODO(wangkuiyi): Make SubmitWorkflow return an error in addition to *pipe.Reader, and remove the calls to log.Printf.

func WaitPortReady

func WaitPortReady(addr string, timeout time.Duration)

WaitPortReady will wait until the tcp address like ip:port can be connected from client.

Types

type Server

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

Server is the instance will be used to connect to DB and execute training

func NewServer

func NewServer(run func(string, string, *pb.Session) *pipe.Reader,
	modelDir string) *Server

NewServer returns a server instance

func (*Server) Fetch

func (s *Server) Fetch(ctx context.Context, job *pb.FetchRequest) (*pb.FetchResponse, error)

Fetch implements `rpc Fetch (Job) returns(JobStatus)`

func (*Server) Run

func (s *Server) Run(req *pb.Request, stream pb.SQLFlow_RunServer) error

Run implements `rpc Run (Request) returns (stream Response)`