mirror of https://github.com/ouqiang/gocron
33 lines
754 B
Go
33 lines
754 B
Go
package server
|
|
|
|
import (
|
|
"golang.org/x/net/context"
|
|
"net"
|
|
"google.golang.org/grpc/grpclog"
|
|
"google.golang.org/grpc"
|
|
pb "github.com/ouqiang/gocron/modules/rpc/proto"
|
|
"github.com/ouqiang/gocron/modules/utils"
|
|
)
|
|
|
|
type Server struct {}
|
|
|
|
func (s Server) Run(ctx context.Context, req *pb.TaskRequest) (*pb.TaskResponse, error) {
|
|
output, err := utils.ExecShellWithTimeout(int(req.Timeout), req.Command)
|
|
resp := new(pb.TaskResponse)
|
|
resp.Output = output
|
|
|
|
return resp, err
|
|
}
|
|
|
|
func Start(addr string) {
|
|
l, err := net.Listen("tcp", addr)
|
|
if err != nil {
|
|
grpclog.Fatal(err)
|
|
}
|
|
s := grpc.NewServer()
|
|
pb.RegisterTaskServer(s, Server{})
|
|
grpclog.Println("listen address ", addr)
|
|
s.Serve(l)
|
|
}
|
|
|