Skip to content

GitLab

Commit f3b08103 authored by Zhou Yaochen's avatar Zhou Yaochen
Browse files

rename worker

parent 89800e6e
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 381 additions and 235 deletions
makefile
View file @ f3b08103
...@@ -2,3 +2,9 @@ IMAGE_NAME:=hub.cstcloud.cn/scalebox/goactuator ...@@ -2,3 +2,9 @@ IMAGE_NAME:=hub.cstcloud.cn/scalebox/goactuator
build: build:
docker build --network=host -f DockerFile -t $(IMAGE_NAME) . docker build --network=host -f DockerFile -t $(IMAGE_NAME) .
push:
docker push $(IMAGE_NAME)
clean:
docker rmi $(IMAGE_NAME)
src/actuator/actuatotFeatures.go
View file @ f3b08103
...@@ -57,7 +57,7 @@ func GetRunnableDaemonList() ([]*pb.Command,error) { ...@@ -57,7 +57,7 @@ func GetRunnableDaemonList() ([]*pb.Command,error) {
defer conn.Close() defer conn.Close()
c := pb.NewBoxServiceClient(conn) c := pb.NewBoxServiceClient(conn)
res,err:=c.GetRunnableDaemonList(context.Background(),&empty.Empty{}) res,err:=c.GetRunnableWorkerList(context.Background(),&empty.Empty{})
if err!=nil{ if err!=nil{
log.Printf("getRunnableDaemonList grpc error:%v",err) log.Printf("getRunnableDaemonList grpc error:%v",err)
logger.Printf("getRunnableDaemonList grpc error:%v",err) logger.Printf("getRunnableDaemonList grpc error:%v",err)
...@@ -80,7 +80,7 @@ func GetTerminatedDaemonList() ([]*pb.Command,error) { ...@@ -80,7 +80,7 @@ func GetTerminatedDaemonList() ([]*pb.Command,error) {
defer conn.Close() defer conn.Close()
c := pb.NewBoxServiceClient(conn) c := pb.NewBoxServiceClient(conn)
res,err:=c.GetTerminableDaemonList(context.Background(),&empty.Empty{}) res,err:=c.GetTerminableWorkerList(context.Background(),&empty.Empty{})
if err!=nil{ if err!=nil{
log.Printf("getTerminatedDaemonList grpc error:%v",err) log.Printf("getTerminatedDaemonList grpc error:%v",err)
logger.Printf("getTerminatedDaemonList grpc error:%v",err) logger.Printf("getTerminatedDaemonList grpc error:%v",err)
...@@ -134,7 +134,7 @@ func SetRunnableDaemonInitialized(cmdPkg *pb.Command)error { ...@@ -134,7 +134,7 @@ func SetRunnableDaemonInitialized(cmdPkg *pb.Command)error {
} }
defer conn.Close() defer conn.Close()
c := pb.NewBoxServiceClient(conn) c := pb.NewBoxServiceClient(conn)
retCode,err:=c.SetDaemonInitialized(context.Background(),&wrps.Int32Value{Value: cmdPkg.Id}) retCode,err:=c.SetWorkerInitialized(context.Background(),&wrps.Int32Value{Value: cmdPkg.Id})
if err!=nil{ if err!=nil{
log.Printf("setDaemonInit grpc error:%v",err) log.Printf("setDaemonInit grpc error:%v",err)
logger.Printf("setDaemonInit grpc error:%v",err) logger.Printf("setDaemonInit grpc error:%v",err)
...@@ -158,7 +158,7 @@ func SetRunnableDaemonTerminated(cmdPkg *pb.Command)error{ ...@@ -158,7 +158,7 @@ func SetRunnableDaemonTerminated(cmdPkg *pb.Command)error{
} }
defer conn.Close() defer conn.Close()
c := pb.NewBoxServiceClient(conn) c := pb.NewBoxServiceClient(conn)
retCode,err:=c.SetDaemonTerminated(context.Background(),&wrps.Int32Value{Value: cmdPkg.Id}) retCode,err:=c.SetWorkerTerminated(context.Background(),&wrps.Int32Value{Value: cmdPkg.Id})
if err!=nil{ if err!=nil{
log.Printf("setDaemonTerminated grpc error:%v",err) log.Printf("setDaemonTerminated grpc error:%v",err)
logger.Printf("setDaemonTerminated grpc error:%v",err) logger.Printf("setDaemonTerminated grpc error:%v",err)
......
src/main/main
View file @ f3b08103
No preview for this file type
src/main/main.go
View file @ f3b08103
...@@ -78,7 +78,7 @@ func main() { ...@@ -78,7 +78,7 @@ func main() {
go func() { go func() {
err:=Instantiate(cmdPkg) err:=Instantiate(cmdPkg)
if err!=nil{ if err!=nil{
logger.Printf("Launch container failure:%v",err) logger.Printf("Terminate container failure:%v",err)
return return
} }
err = SetRunnableDaemonTerminated(cmdPkg) err = SetRunnableDaemonTerminated(cmdPkg)
...@@ -90,7 +90,7 @@ func main() { ...@@ -90,7 +90,7 @@ func main() {
} }
} }
} }
log.Println("Going to sleep for 2m") log.Println("Going to sleep for 2min")
time.Sleep(time.Second*120) time.Sleep(time.Second*120)
} }
......
src/scalebox/agent.pb.go
View file @ f3b08103
This diff is collapsed.
src/scalebox/agent.proto
View file @ f3b08103
...@@ -48,21 +48,21 @@ service BoxService { ...@@ -48,21 +48,21 @@ service BoxService {
// slot: 'READY' -> 'ON' // slot: 'READY' -> 'ON'
rpc SetSlotInitialized(google.protobuf.Int32Value) returns (google.protobuf.Int32Value); rpc SetSlotInitialized(google.protobuf.Int32Value) returns (google.protobuf.Int32Value);
// job:'RUNNING' && daemon:'PAUSED' // job:'RUNNING' && worker:'PAUSED'
rpc GetRunnableDaemonList(google.protobuf.Empty) returns (CommandList); rpc GetRunnableWorkerList(google.protobuf.Empty) returns (CommandList);
// input : daemon id // input : worker id
// return : ret_code : 0(OK), -1(daemon NOT FOUND), -2(daemon WRONG STATUS) // return : ret_code : 0(OK), -1(worker NOT FOUND), -2(worker WRONG STATUS)
// daemon: 'PAUSED' -> 'RUNNING' // worker: 'PAUSED' -> 'RUNNING'
rpc SetDaemonInitialized(google.protobuf.Int32Value) returns (google.protobuf.Int32Value); rpc SetWorkerInitialized(google.protobuf.Int32Value) returns (google.protobuf.Int32Value);
// job:'PAUSED' && daemon:'RUNNING' // job:'PAUSED' && worker:'RUNNING'
rpc GetTerminableDaemonList(google.protobuf.Empty) returns (CommandList); rpc GetTerminableWorkerList(google.protobuf.Empty) returns (CommandList);
// input : daemon id, return ret_code; // input : worker id, return ret_code;
// return : ret_code : 0(OK), -1(daemon NOT FOUND), -2(daemon WRONG STATUS) // return : ret_code : 0(OK), -1(worker NOT FOUND), -2(worker WRONG STATUS)
// daemon : 'RUNNING' -> 'PAUSED' // worker : 'RUNNING' -> 'PAUSED'
rpc SetDaemonTerminated(google.protobuf.Int32Value) returns (google.protobuf.Int32Value); rpc SetWorkerTerminated(google.protobuf.Int32Value) returns (google.protobuf.Int32Value);
// ////////////////////////////////////////////////////////////////// // //////////////////////////////////////////////////////////////////
// app client , called by user app. // app client , called by user app.
...@@ -72,6 +72,8 @@ service BoxService { ...@@ -72,6 +72,8 @@ service BoxService {
// return : ret_code : 0(OK), -1(job NOT FOUND) // return : ret_code : 0(OK), -1(job NOT FOUND)
// task_status_code : 'INITIAL'/-9 // task_status_code : 'INITIAL'/-9
rpc SendToNextJob(JobKey) returns (google.protobuf.Int32Value); rpc SendToNextJob(JobKey) returns (google.protobuf.Int32Value);
rpc SendMessagesToNext(JobKeys) returns (google.protobuf.Int32Value);
} }
message JobKey { message JobKey {
...@@ -79,6 +81,15 @@ message JobKey { ...@@ -79,6 +81,15 @@ message JobKey {
string job_name=1; string job_name=1;
string key_text=2; string key_text=2;
int32 slot_id=3; int32 slot_id=3;
// OR worker_id ? job_id ?
}
message JobKeys {
// qualified name of job
string next_job_name=1;
repeated string key_texts=2;
int32 current_job_id=3;
// OR worker_id ?
} }
message TaskItem { message TaskItem {
...@@ -87,7 +98,7 @@ message TaskItem { ...@@ -87,7 +98,7 @@ message TaskItem {
} }
message Command { message Command {
// primary key of slot/daemon table // primary key of slot/worker table
int32 id=1; int32 id=1;
string host=2; string host=2;
string command_text=3; string command_text=3;
......
src/scalebox/agent_grpc.pb.go
View file @ f3b08103
...@@ -44,22 +44,23 @@ type BoxServiceClient interface { ...@@ -44,22 +44,23 @@ type BoxServiceClient interface {
// return : ret_code : 0(OK), -1(slot NOT FOUND), -2(slot WRONG STATUS) // return : ret_code : 0(OK), -1(slot NOT FOUND), -2(slot WRONG STATUS)
// slot: 'READY' -> 'ON' // slot: 'READY' -> 'ON'
SetSlotInitialized(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error) SetSlotInitialized(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error)
// job:'RUNNING' && daemon:'PAUSED' // job:'RUNNING' && worker:'PAUSED'
GetRunnableDaemonList(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (*CommandList, error) GetRunnableWorkerList(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (*CommandList, error)
// input : daemon id // input : worker id
// return : ret_code : 0(OK), -1(daemon NOT FOUND), -2(daemon WRONG STATUS) // return : ret_code : 0(OK), -1(worker NOT FOUND), -2(worker WRONG STATUS)
// daemon: 'PAUSED' -> 'RUNNING' // worker: 'PAUSED' -> 'RUNNING'
SetDaemonInitialized(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error) SetWorkerInitialized(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error)
// job:'PAUSED' && daemon:'RUNNING' // job:'PAUSED' && worker:'RUNNING'
GetTerminableDaemonList(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (*CommandList, error) GetTerminableWorkerList(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (*CommandList, error)
// input : daemon id, return ret_code; // input : worker id, return ret_code;
// return : ret_code : 0(OK), -1(daemon NOT FOUND), -2(daemon WRONG STATUS) // return : ret_code : 0(OK), -1(worker NOT FOUND), -2(worker WRONG STATUS)
// daemon : 'RUNNING' -> 'PAUSED' // worker : 'RUNNING' -> 'PAUSED'
SetDaemonTerminated(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error) SetWorkerTerminated(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error)
// send task-key to next job in current pipeline // send task-key to next job in current pipeline
// return : ret_code : 0(OK), -1(job NOT FOUND) // return : ret_code : 0(OK), -1(job NOT FOUND)
// task_status_code : 'INITIAL'/-9 // task_status_code : 'INITIAL'/-9
SendToNextJob(ctx context.Context, in *JobKey, opts ...grpc.CallOption) (*wrappers.Int32Value, error) SendToNextJob(ctx context.Context, in *JobKey, opts ...grpc.CallOption) (*wrappers.Int32Value, error)
SendMessagesToNext(ctx context.Context, in *JobKeys, opts ...grpc.CallOption) (*wrappers.Int32Value, error)
} }
type boxServiceClient struct { type boxServiceClient struct {
...@@ -124,36 +125,36 @@ func (c *boxServiceClient) SetSlotInitialized(ctx context.Context, in *wrappers. ...@@ -124,36 +125,36 @@ func (c *boxServiceClient) SetSlotInitialized(ctx context.Context, in *wrappers.
return out, nil return out, nil
} }
func (c *boxServiceClient) GetRunnableDaemonList(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (*CommandList, error) { func (c *boxServiceClient) GetRunnableWorkerList(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (*CommandList, error) {
out := new(CommandList) out := new(CommandList)
err := c.cc.Invoke(ctx, "/scalebox.BoxService/GetRunnableDaemonList", in, out, opts...) err := c.cc.Invoke(ctx, "/scalebox.BoxService/GetRunnableWorkerList", in, out, opts...)
if err != nil { if err != nil {
return nil, err return nil, err
} }
return out, nil return out, nil
} }
func (c *boxServiceClient) SetDaemonInitialized(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error) { func (c *boxServiceClient) SetWorkerInitialized(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error) {
out := new(wrappers.Int32Value) out := new(wrappers.Int32Value)
err := c.cc.Invoke(ctx, "/scalebox.BoxService/SetDaemonInitialized", in, out, opts...) err := c.cc.Invoke(ctx, "/scalebox.BoxService/SetWorkerInitialized", in, out, opts...)
if err != nil { if err != nil {
return nil, err return nil, err
} }
return out, nil return out, nil
} }
func (c *boxServiceClient) GetTerminableDaemonList(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (*CommandList, error) { func (c *boxServiceClient) GetTerminableWorkerList(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (*CommandList, error) {
out := new(CommandList) out := new(CommandList)
err := c.cc.Invoke(ctx, "/scalebox.BoxService/GetTerminableDaemonList", in, out, opts...) err := c.cc.Invoke(ctx, "/scalebox.BoxService/GetTerminableWorkerList", in, out, opts...)
if err != nil { if err != nil {
return nil, err return nil, err
} }
return out, nil return out, nil
} }
func (c *boxServiceClient) SetDaemonTerminated(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error) { func (c *boxServiceClient) SetWorkerTerminated(ctx context.Context, in *wrappers.Int32Value, opts ...grpc.CallOption) (*wrappers.Int32Value, error) {
out := new(wrappers.Int32Value) out := new(wrappers.Int32Value)
err := c.cc.Invoke(ctx, "/scalebox.BoxService/SetDaemonTerminated", in, out, opts...) err := c.cc.Invoke(ctx, "/scalebox.BoxService/SetWorkerTerminated", in, out, opts...)
if err != nil { if err != nil {
return nil, err return nil, err
} }
...@@ -169,6 +170,15 @@ func (c *boxServiceClient) SendToNextJob(ctx context.Context, in *JobKey, opts . ...@@ -169,6 +170,15 @@ func (c *boxServiceClient) SendToNextJob(ctx context.Context, in *JobKey, opts .
return out, nil return out, nil
} }
func (c *boxServiceClient) SendMessagesToNext(ctx context.Context, in *JobKeys, opts ...grpc.CallOption) (*wrappers.Int32Value, error) {
out := new(wrappers.Int32Value)
err := c.cc.Invoke(ctx, "/scalebox.BoxService/SendMessagesToNext", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
// BoxServiceServer is the server API for BoxService service. // BoxServiceServer is the server API for BoxService service.
// All implementations must embed UnimplementedBoxServiceServer // All implementations must embed UnimplementedBoxServiceServer
// for forward compatibility // for forward compatibility
...@@ -197,22 +207,23 @@ type BoxServiceServer interface { ...@@ -197,22 +207,23 @@ type BoxServiceServer interface {
// return : ret_code : 0(OK), -1(slot NOT FOUND), -2(slot WRONG STATUS) // return : ret_code : 0(OK), -1(slot NOT FOUND), -2(slot WRONG STATUS)
// slot: 'READY' -> 'ON' // slot: 'READY' -> 'ON'
SetSlotInitialized(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) SetSlotInitialized(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error)
// job:'RUNNING' && daemon:'PAUSED' // job:'RUNNING' && worker:'PAUSED'
GetRunnableDaemonList(context.Context, *empty.Empty) (*CommandList, error) GetRunnableWorkerList(context.Context, *empty.Empty) (*CommandList, error)
// input : daemon id // input : worker id
// return : ret_code : 0(OK), -1(daemon NOT FOUND), -2(daemon WRONG STATUS) // return : ret_code : 0(OK), -1(worker NOT FOUND), -2(worker WRONG STATUS)
// daemon: 'PAUSED' -> 'RUNNING' // worker: 'PAUSED' -> 'RUNNING'
SetDaemonInitialized(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) SetWorkerInitialized(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error)
// job:'PAUSED' && daemon:'RUNNING' // job:'PAUSED' && worker:'RUNNING'
GetTerminableDaemonList(context.Context, *empty.Empty) (*CommandList, error) GetTerminableWorkerList(context.Context, *empty.Empty) (*CommandList, error)
// input : daemon id, return ret_code; // input : worker id, return ret_code;
// return : ret_code : 0(OK), -1(daemon NOT FOUND), -2(daemon WRONG STATUS) // return : ret_code : 0(OK), -1(worker NOT FOUND), -2(worker WRONG STATUS)
// daemon : 'RUNNING' -> 'PAUSED' // worker : 'RUNNING' -> 'PAUSED'
SetDaemonTerminated(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) SetWorkerTerminated(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error)
// send task-key to next job in current pipeline // send task-key to next job in current pipeline
// return : ret_code : 0(OK), -1(job NOT FOUND) // return : ret_code : 0(OK), -1(job NOT FOUND)
// task_status_code : 'INITIAL'/-9 // task_status_code : 'INITIAL'/-9
SendToNextJob(context.Context, *JobKey) (*wrappers.Int32Value, error) SendToNextJob(context.Context, *JobKey) (*wrappers.Int32Value, error)
SendMessagesToNext(context.Context, *JobKeys) (*wrappers.Int32Value, error)
mustEmbedUnimplementedBoxServiceServer() mustEmbedUnimplementedBoxServiceServer()
} }
...@@ -238,21 +249,24 @@ func (UnimplementedBoxServiceServer) GetRunnableSlotList(context.Context, *empty ...@@ -238,21 +249,24 @@ func (UnimplementedBoxServiceServer) GetRunnableSlotList(context.Context, *empty
func (UnimplementedBoxServiceServer) SetSlotInitialized(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) { func (UnimplementedBoxServiceServer) SetSlotInitialized(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) {
return nil, status.Errorf(codes.Unimplemented, "method SetSlotInitialized not implemented") return nil, status.Errorf(codes.Unimplemented, "method SetSlotInitialized not implemented")
} }
func (UnimplementedBoxServiceServer) GetRunnableDaemonList(context.Context, *empty.Empty) (*CommandList, error) { func (UnimplementedBoxServiceServer) GetRunnableWorkerList(context.Context, *empty.Empty) (*CommandList, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetRunnableDaemonList not implemented") return nil, status.Errorf(codes.Unimplemented, "method GetRunnableWorkerList not implemented")
} }
func (UnimplementedBoxServiceServer) SetDaemonInitialized(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) { func (UnimplementedBoxServiceServer) SetWorkerInitialized(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) {
return nil, status.Errorf(codes.Unimplemented, "method SetDaemonInitialized not implemented") return nil, status.Errorf(codes.Unimplemented, "method SetWorkerInitialized not implemented")
} }
func (UnimplementedBoxServiceServer) GetTerminableDaemonList(context.Context, *empty.Empty) (*CommandList, error) { func (UnimplementedBoxServiceServer) GetTerminableWorkerList(context.Context, *empty.Empty) (*CommandList, error) {
return nil, status.Errorf(codes.Unimplemented, "method GetTerminableDaemonList not implemented") return nil, status.Errorf(codes.Unimplemented, "method GetTerminableWorkerList not implemented")
} }
func (UnimplementedBoxServiceServer) SetDaemonTerminated(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) { func (UnimplementedBoxServiceServer) SetWorkerTerminated(context.Context, *wrappers.Int32Value) (*wrappers.Int32Value, error) {
return nil, status.Errorf(codes.Unimplemented, "method SetDaemonTerminated not implemented") return nil, status.Errorf(codes.Unimplemented, "method SetWorkerTerminated not implemented")
} }
func (UnimplementedBoxServiceServer) SendToNextJob(context.Context, *JobKey) (*wrappers.Int32Value, error) { func (UnimplementedBoxServiceServer) SendToNextJob(context.Context, *JobKey) (*wrappers.Int32Value, error) {
return nil, status.Errorf(codes.Unimplemented, "method SendToNextJob not implemented") return nil, status.Errorf(codes.Unimplemented, "method SendToNextJob not implemented")
} }
func (UnimplementedBoxServiceServer) SendMessagesToNext(context.Context, *JobKeys) (*wrappers.Int32Value, error) {
return nil, status.Errorf(codes.Unimplemented, "method SendMessagesToNext not implemented")
}
func (UnimplementedBoxServiceServer) mustEmbedUnimplementedBoxServiceServer() {} func (UnimplementedBoxServiceServer) mustEmbedUnimplementedBoxServiceServer() {}
// UnsafeBoxServiceServer may be embedded to opt out of forward compatibility for this service. // UnsafeBoxServiceServer may be embedded to opt out of forward compatibility for this service.
...@@ -374,74 +388,74 @@ func _BoxService_SetSlotInitialized_Handler(srv interface{}, ctx context.Context ...@@ -374,74 +388,74 @@ func _BoxService_SetSlotInitialized_Handler(srv interface{}, ctx context.Context
return interceptor(ctx, in, info, handler) return interceptor(ctx, in, info, handler)
} }
func _BoxService_GetRunnableDaemonList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { func _BoxService_GetRunnableWorkerList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(empty.Empty) in := new(empty.Empty)
if err := dec(in); err != nil { if err := dec(in); err != nil {
return nil, err return nil, err
} }
if interceptor == nil { if interceptor == nil {
return srv.(BoxServiceServer).GetRunnableDaemonList(ctx, in) return srv.(BoxServiceServer).GetRunnableWorkerList(ctx, in)
} }
info := &grpc.UnaryServerInfo{ info := &grpc.UnaryServerInfo{
Server: srv, Server: srv,
FullMethod: "/scalebox.BoxService/GetRunnableDaemonList", FullMethod: "/scalebox.BoxService/GetRunnableWorkerList",
} }
handler := func(ctx context.Context, req interface{}) (interface{}, error) { handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(BoxServiceServer).GetRunnableDaemonList(ctx, req.(*empty.Empty)) return srv.(BoxServiceServer).GetRunnableWorkerList(ctx, req.(*empty.Empty))
} }
return interceptor(ctx, in, info, handler) return interceptor(ctx, in, info, handler)
} }
func _BoxService_SetDaemonInitialized_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { func _BoxService_SetWorkerInitialized_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(wrappers.Int32Value) in := new(wrappers.Int32Value)
if err := dec(in); err != nil { if err := dec(in); err != nil {
return nil, err return nil, err
} }
if interceptor == nil { if interceptor == nil {
return srv.(BoxServiceServer).SetDaemonInitialized(ctx, in) return srv.(BoxServiceServer).SetWorkerInitialized(ctx, in)
} }
info := &grpc.UnaryServerInfo{ info := &grpc.UnaryServerInfo{
Server: srv, Server: srv,
FullMethod: "/scalebox.BoxService/SetDaemonInitialized", FullMethod: "/scalebox.BoxService/SetWorkerInitialized",
} }
handler := func(ctx context.Context, req interface{}) (interface{}, error) { handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(BoxServiceServer).SetDaemonInitialized(ctx, req.(*wrappers.Int32Value)) return srv.(BoxServiceServer).SetWorkerInitialized(ctx, req.(*wrappers.Int32Value))
} }
return interceptor(ctx, in, info, handler) return interceptor(ctx, in, info, handler)
} }
func _BoxService_GetTerminableDaemonList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { func _BoxService_GetTerminableWorkerList_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(empty.Empty) in := new(empty.Empty)
if err := dec(in); err != nil { if err := dec(in); err != nil {
return nil, err return nil, err
} }
if interceptor == nil { if interceptor == nil {
return srv.(BoxServiceServer).GetTerminableDaemonList(ctx, in) return srv.(BoxServiceServer).GetTerminableWorkerList(ctx, in)
} }
info := &grpc.UnaryServerInfo{ info := &grpc.UnaryServerInfo{
Server: srv, Server: srv,
FullMethod: "/scalebox.BoxService/GetTerminableDaemonList", FullMethod: "/scalebox.BoxService/GetTerminableWorkerList",
} }
handler := func(ctx context.Context, req interface{}) (interface{}, error) { handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(BoxServiceServer).GetTerminableDaemonList(ctx, req.(*empty.Empty)) return srv.(BoxServiceServer).GetTerminableWorkerList(ctx, req.(*empty.Empty))
} }
return interceptor(ctx, in, info, handler) return interceptor(ctx, in, info, handler)
} }
func _BoxService_SetDaemonTerminated_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) { func _BoxService_SetWorkerTerminated_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(wrappers.Int32Value) in := new(wrappers.Int32Value)
if err := dec(in); err != nil { if err := dec(in); err != nil {
return nil, err return nil, err
} }
if interceptor == nil { if interceptor == nil {
return srv.(BoxServiceServer).SetDaemonTerminated(ctx, in) return srv.(BoxServiceServer).SetWorkerTerminated(ctx, in)
} }
info := &grpc.UnaryServerInfo{ info := &grpc.UnaryServerInfo{
Server: srv, Server: srv,
FullMethod: "/scalebox.BoxService/SetDaemonTerminated", FullMethod: "/scalebox.BoxService/SetWorkerTerminated",
} }
handler := func(ctx context.Context, req interface{}) (interface{}, error) { handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(BoxServiceServer).SetDaemonTerminated(ctx, req.(*wrappers.Int32Value)) return srv.(BoxServiceServer).SetWorkerTerminated(ctx, req.(*wrappers.Int32Value))
} }
return interceptor(ctx, in, info, handler) return interceptor(ctx, in, info, handler)
} }
...@@ -464,6 +478,24 @@ func _BoxService_SendToNextJob_Handler(srv interface{}, ctx context.Context, dec ...@@ -464,6 +478,24 @@ func _BoxService_SendToNextJob_Handler(srv interface{}, ctx context.Context, dec
return interceptor(ctx, in, info, handler) return interceptor(ctx, in, info, handler)
} }
func _BoxService_SendMessagesToNext_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(JobKeys)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(BoxServiceServer).SendMessagesToNext(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/scalebox.BoxService/SendMessagesToNext",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(BoxServiceServer).SendMessagesToNext(ctx, req.(*JobKeys))
}
return interceptor(ctx, in, info, handler)
}
// BoxService_ServiceDesc is the grpc.ServiceDesc for BoxService service. // BoxService_ServiceDesc is the grpc.ServiceDesc for BoxService service.
// It's only intended for direct use with grpc.RegisterService, // It's only intended for direct use with grpc.RegisterService,
// and not to be introspected or modified (even as a copy) // and not to be introspected or modified (even as a copy)
...@@ -496,25 +528,29 @@ var BoxService_ServiceDesc = grpc.ServiceDesc{ ...@@ -496,25 +528,29 @@ var BoxService_ServiceDesc = grpc.ServiceDesc{
Handler: _BoxService_SetSlotInitialized_Handler, Handler: _BoxService_SetSlotInitialized_Handler,
}, },
{ {
MethodName: "GetRunnableDaemonList", MethodName: "GetRunnableWorkerList",
Handler: _BoxService_GetRunnableDaemonList_Handler, Handler: _BoxService_GetRunnableWorkerList_Handler,
}, },
{ {
MethodName: "SetDaemonInitialized", MethodName: "SetWorkerInitialized",
Handler: _BoxService_SetDaemonInitialized_Handler, Handler: _BoxService_SetWorkerInitialized_Handler,
}, },
{ {
MethodName: "GetTerminableDaemonList", MethodName: "GetTerminableWorkerList",
Handler: _BoxService_GetTerminableDaemonList_Handler, Handler: _BoxService_GetTerminableWorkerList_Handler,
}, },
{ {
MethodName: "SetDaemonTerminated", MethodName: "SetWorkerTerminated",
Handler: _BoxService_SetDaemonTerminated_Handler, Handler: _BoxService_SetWorkerTerminated_Handler,
}, },
{ {
MethodName: "SendToNextJob", MethodName: "SendToNextJob",
Handler: _BoxService_SendToNextJob_Handler, Handler: _BoxService_SendToNextJob_Handler,
}, },
{
MethodName: "SendMessagesToNext",
Handler: _BoxService_SendMessagesToNext_Handler,
},
}, },
Streams: []grpc.StreamDesc{}, Streams: []grpc.StreamDesc{},
Metadata: "agent.proto", Metadata: "agent.proto",
......
src/shells/makefile
View file @ f3b08103
...@@ -9,3 +9,5 @@ down: ...@@ -9,3 +9,5 @@ down:
docker stop test-box-server docker stop test-box-server
docker stop test-box-db docker stop test-box-db
docker rm test-box-server docker rm test-box-server
generate:
protoc --go_out=. --go-grpc_out=. ./agent.proto
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment