go.chromium.org/luci@v0.0.0-20240309015107-7cdc2e660f33/resultdb/proto/v1/recorderserver_dec.go (about)

     1  // Code generated by svcdec; DO NOT EDIT.
     2  
     3  package resultpb
     4  
     5  import (
     6  	"context"
     7  
     8  	proto "github.com/golang/protobuf/proto"
     9  
    10  	emptypb "google.golang.org/protobuf/types/known/emptypb"
    11  )
    12  
    13  type DecoratedRecorder struct {
    14  	// Service is the service to decorate.
    15  	Service RecorderServer
    16  	// Prelude is called for each method before forwarding the call to Service.
    17  	// If Prelude returns an error, then the call is skipped and the error is
    18  	// processed via the Postlude (if one is defined), or it is returned directly.
    19  	Prelude func(ctx context.Context, methodName string, req proto.Message) (context.Context, error)
    20  	// Postlude is called for each method after Service has processed the call, or
    21  	// after the Prelude has returned an error. This takes the Service's
    22  	// response proto (which may be nil) and/or any error. The decorated
    23  	// service will return the response (possibly mutated) and error that Postlude
    24  	// returns.
    25  	Postlude func(ctx context.Context, methodName string, rsp proto.Message, err error) error
    26  }
    27  
    28  func (s *DecoratedRecorder) CreateInvocation(ctx context.Context, req *CreateInvocationRequest) (rsp *Invocation, err error) {
    29  	if s.Prelude != nil {
    30  		var newCtx context.Context
    31  		newCtx, err = s.Prelude(ctx, "CreateInvocation", req)
    32  		if err == nil {
    33  			ctx = newCtx
    34  		}
    35  	}
    36  	if err == nil {
    37  		rsp, err = s.Service.CreateInvocation(ctx, req)
    38  	}
    39  	if s.Postlude != nil {
    40  		err = s.Postlude(ctx, "CreateInvocation", rsp, err)
    41  	}
    42  	return
    43  }
    44  
    45  func (s *DecoratedRecorder) BatchCreateInvocations(ctx context.Context, req *BatchCreateInvocationsRequest) (rsp *BatchCreateInvocationsResponse, err error) {
    46  	if s.Prelude != nil {
    47  		var newCtx context.Context
    48  		newCtx, err = s.Prelude(ctx, "BatchCreateInvocations", req)
    49  		if err == nil {
    50  			ctx = newCtx
    51  		}
    52  	}
    53  	if err == nil {
    54  		rsp, err = s.Service.BatchCreateInvocations(ctx, req)
    55  	}
    56  	if s.Postlude != nil {
    57  		err = s.Postlude(ctx, "BatchCreateInvocations", rsp, err)
    58  	}
    59  	return
    60  }
    61  
    62  func (s *DecoratedRecorder) UpdateInvocation(ctx context.Context, req *UpdateInvocationRequest) (rsp *Invocation, err error) {
    63  	if s.Prelude != nil {
    64  		var newCtx context.Context
    65  		newCtx, err = s.Prelude(ctx, "UpdateInvocation", req)
    66  		if err == nil {
    67  			ctx = newCtx
    68  		}
    69  	}
    70  	if err == nil {
    71  		rsp, err = s.Service.UpdateInvocation(ctx, req)
    72  	}
    73  	if s.Postlude != nil {
    74  		err = s.Postlude(ctx, "UpdateInvocation", rsp, err)
    75  	}
    76  	return
    77  }
    78  
    79  func (s *DecoratedRecorder) FinalizeInvocation(ctx context.Context, req *FinalizeInvocationRequest) (rsp *Invocation, err error) {
    80  	if s.Prelude != nil {
    81  		var newCtx context.Context
    82  		newCtx, err = s.Prelude(ctx, "FinalizeInvocation", req)
    83  		if err == nil {
    84  			ctx = newCtx
    85  		}
    86  	}
    87  	if err == nil {
    88  		rsp, err = s.Service.FinalizeInvocation(ctx, req)
    89  	}
    90  	if s.Postlude != nil {
    91  		err = s.Postlude(ctx, "FinalizeInvocation", rsp, err)
    92  	}
    93  	return
    94  }
    95  
    96  func (s *DecoratedRecorder) UpdateIncludedInvocations(ctx context.Context, req *UpdateIncludedInvocationsRequest) (rsp *emptypb.Empty, err error) {
    97  	if s.Prelude != nil {
    98  		var newCtx context.Context
    99  		newCtx, err = s.Prelude(ctx, "UpdateIncludedInvocations", req)
   100  		if err == nil {
   101  			ctx = newCtx
   102  		}
   103  	}
   104  	if err == nil {
   105  		rsp, err = s.Service.UpdateIncludedInvocations(ctx, req)
   106  	}
   107  	if s.Postlude != nil {
   108  		err = s.Postlude(ctx, "UpdateIncludedInvocations", rsp, err)
   109  	}
   110  	return
   111  }
   112  
   113  func (s *DecoratedRecorder) MarkInvocationSubmitted(ctx context.Context, req *MarkInvocationSubmittedRequest) (rsp *emptypb.Empty, err error) {
   114  	if s.Prelude != nil {
   115  		var newCtx context.Context
   116  		newCtx, err = s.Prelude(ctx, "MarkInvocationSubmitted", req)
   117  		if err == nil {
   118  			ctx = newCtx
   119  		}
   120  	}
   121  	if err == nil {
   122  		rsp, err = s.Service.MarkInvocationSubmitted(ctx, req)
   123  	}
   124  	if s.Postlude != nil {
   125  		err = s.Postlude(ctx, "MarkInvocationSubmitted", rsp, err)
   126  	}
   127  	return
   128  }
   129  
   130  func (s *DecoratedRecorder) CreateTestResult(ctx context.Context, req *CreateTestResultRequest) (rsp *TestResult, err error) {
   131  	if s.Prelude != nil {
   132  		var newCtx context.Context
   133  		newCtx, err = s.Prelude(ctx, "CreateTestResult", req)
   134  		if err == nil {
   135  			ctx = newCtx
   136  		}
   137  	}
   138  	if err == nil {
   139  		rsp, err = s.Service.CreateTestResult(ctx, req)
   140  	}
   141  	if s.Postlude != nil {
   142  		err = s.Postlude(ctx, "CreateTestResult", rsp, err)
   143  	}
   144  	return
   145  }
   146  
   147  func (s *DecoratedRecorder) BatchCreateTestResults(ctx context.Context, req *BatchCreateTestResultsRequest) (rsp *BatchCreateTestResultsResponse, err error) {
   148  	if s.Prelude != nil {
   149  		var newCtx context.Context
   150  		newCtx, err = s.Prelude(ctx, "BatchCreateTestResults", req)
   151  		if err == nil {
   152  			ctx = newCtx
   153  		}
   154  	}
   155  	if err == nil {
   156  		rsp, err = s.Service.BatchCreateTestResults(ctx, req)
   157  	}
   158  	if s.Postlude != nil {
   159  		err = s.Postlude(ctx, "BatchCreateTestResults", rsp, err)
   160  	}
   161  	return
   162  }
   163  
   164  func (s *DecoratedRecorder) CreateTestExoneration(ctx context.Context, req *CreateTestExonerationRequest) (rsp *TestExoneration, err error) {
   165  	if s.Prelude != nil {
   166  		var newCtx context.Context
   167  		newCtx, err = s.Prelude(ctx, "CreateTestExoneration", req)
   168  		if err == nil {
   169  			ctx = newCtx
   170  		}
   171  	}
   172  	if err == nil {
   173  		rsp, err = s.Service.CreateTestExoneration(ctx, req)
   174  	}
   175  	if s.Postlude != nil {
   176  		err = s.Postlude(ctx, "CreateTestExoneration", rsp, err)
   177  	}
   178  	return
   179  }
   180  
   181  func (s *DecoratedRecorder) BatchCreateTestExonerations(ctx context.Context, req *BatchCreateTestExonerationsRequest) (rsp *BatchCreateTestExonerationsResponse, err error) {
   182  	if s.Prelude != nil {
   183  		var newCtx context.Context
   184  		newCtx, err = s.Prelude(ctx, "BatchCreateTestExonerations", req)
   185  		if err == nil {
   186  			ctx = newCtx
   187  		}
   188  	}
   189  	if err == nil {
   190  		rsp, err = s.Service.BatchCreateTestExonerations(ctx, req)
   191  	}
   192  	if s.Postlude != nil {
   193  		err = s.Postlude(ctx, "BatchCreateTestExonerations", rsp, err)
   194  	}
   195  	return
   196  }
   197  
   198  func (s *DecoratedRecorder) BatchCreateArtifacts(ctx context.Context, req *BatchCreateArtifactsRequest) (rsp *BatchCreateArtifactsResponse, err error) {
   199  	if s.Prelude != nil {
   200  		var newCtx context.Context
   201  		newCtx, err = s.Prelude(ctx, "BatchCreateArtifacts", req)
   202  		if err == nil {
   203  			ctx = newCtx
   204  		}
   205  	}
   206  	if err == nil {
   207  		rsp, err = s.Service.BatchCreateArtifacts(ctx, req)
   208  	}
   209  	if s.Postlude != nil {
   210  		err = s.Postlude(ctx, "BatchCreateArtifacts", rsp, err)
   211  	}
   212  	return
   213  }