feat: introduce reconcile interface (#264)

* feat: introduce reconcile interface

* chore: gen set_header

* chore: add comments

* feat: return procedure_id

* feat: add `ResolveStragey` and `Parallelism` to request

* chore: apply suggestions from CR
This commit is contained in:
Weny Xu
2025-08-04 16:50:33 +08:00
committed by GitHub
parent cc56a6cdac
commit 91c3d7b97a
10 changed files with 10392 additions and 1304 deletions
+38
View File
@@ -26,6 +26,8 @@ type ProcedureServiceClient interface {
Query(ctx context.Context, in *QueryProcedureRequest, opts ...grpc.CallOption) (*ProcedureStateResponse, error)
// Submits a DDL task
Ddl(ctx context.Context, in *DdlTaskRequest, opts ...grpc.CallOption) (*DdlTaskResponse, error)
// Submits a reconcile task
Reconcile(ctx context.Context, in *ReconcileRequest, opts ...grpc.CallOption) (*ReconcileResponse, error)
// Submits a region migration task
Migrate(ctx context.Context, in *MigrateRegionRequest, opts ...grpc.CallOption) (*MigrateRegionResponse, error)
// Query all submitted procedures details
@@ -58,6 +60,15 @@ func (c *procedureServiceClient) Ddl(ctx context.Context, in *DdlTaskRequest, op
return out, nil
}
func (c *procedureServiceClient) Reconcile(ctx context.Context, in *ReconcileRequest, opts ...grpc.CallOption) (*ReconcileResponse, error) {
out := new(ReconcileResponse)
err := c.cc.Invoke(ctx, "/greptime.v1.meta.ProcedureService/reconcile", in, out, opts...)
if err != nil {
return nil, err
}
return out, nil
}
func (c *procedureServiceClient) Migrate(ctx context.Context, in *MigrateRegionRequest, opts ...grpc.CallOption) (*MigrateRegionResponse, error) {
out := new(MigrateRegionResponse)
err := c.cc.Invoke(ctx, "/greptime.v1.meta.ProcedureService/migrate", in, out, opts...)
@@ -84,6 +95,8 @@ type ProcedureServiceServer interface {
Query(context.Context, *QueryProcedureRequest) (*ProcedureStateResponse, error)
// Submits a DDL task
Ddl(context.Context, *DdlTaskRequest) (*DdlTaskResponse, error)
// Submits a reconcile task
Reconcile(context.Context, *ReconcileRequest) (*ReconcileResponse, error)
// Submits a region migration task
Migrate(context.Context, *MigrateRegionRequest) (*MigrateRegionResponse, error)
// Query all submitted procedures details
@@ -101,6 +114,9 @@ func (UnimplementedProcedureServiceServer) Query(context.Context, *QueryProcedur
func (UnimplementedProcedureServiceServer) Ddl(context.Context, *DdlTaskRequest) (*DdlTaskResponse, error) {
return nil, status.Errorf(codes.Unimplemented, "method Ddl not implemented")
}
func (UnimplementedProcedureServiceServer) Reconcile(context.Context, *ReconcileRequest) (*ReconcileResponse, error) {
return nil, status.Errorf(codes.Unimplemented, "method Reconcile not implemented")
}
func (UnimplementedProcedureServiceServer) Migrate(context.Context, *MigrateRegionRequest) (*MigrateRegionResponse, error) {
return nil, status.Errorf(codes.Unimplemented, "method Migrate not implemented")
}
@@ -156,6 +172,24 @@ func _ProcedureService_Ddl_Handler(srv interface{}, ctx context.Context, dec fun
return interceptor(ctx, in, info, handler)
}
func _ProcedureService_Reconcile_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(ReconcileRequest)
if err := dec(in); err != nil {
return nil, err
}
if interceptor == nil {
return srv.(ProcedureServiceServer).Reconcile(ctx, in)
}
info := &grpc.UnaryServerInfo{
Server: srv,
FullMethod: "/greptime.v1.meta.ProcedureService/reconcile",
}
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
return srv.(ProcedureServiceServer).Reconcile(ctx, req.(*ReconcileRequest))
}
return interceptor(ctx, in, info, handler)
}
func _ProcedureService_Migrate_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
in := new(MigrateRegionRequest)
if err := dec(in); err != nil {
@@ -207,6 +241,10 @@ var ProcedureService_ServiceDesc = grpc.ServiceDesc{
MethodName: "ddl",
Handler: _ProcedureService_Ddl_Handler,
},
{
MethodName: "reconcile",
Handler: _ProcedureService_Reconcile_Handler,
},
{
MethodName: "migrate",
Handler: _ProcedureService_Migrate_Handler,