use grpc error codes (#101)

tcncloud · Aug 25, 2021 · 8ab7abd · 8ab7abd
1 parent 7cf27b0
commit 8ab7abd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/generator/service.go b/generator/service.go
@@ -934,10 +934,12 @@ func WriteIters(p *Printer, s *Service) (outErr error) {
             _, hasSecond := this.Next()
             if !hasFirst || hasSecond {
                 amount := "none"
+                statusCode := codes.NotFound
                 if hasSecond {
                     amount = "multiple"
+                    statusCode = codes.FailedPrecondition
                 }
-                return &Row_`, sName, `_`, camelQ(q), `{err: lib.NotFound{Msg: fmt.Sprintf("expected exactly 1 result from query '`, camelQ(q), `' found %s", amount)}}
+                return &Row_`, sName, `_`, camelQ(q), `{err: gstatus.Errorf(statusCode, "expected exactly 1 result from query '`, camelQ(q), `' found %s", amount)}
             }
             return first
         }
@@ -1154,10 +1156,12 @@ func WriteIters(p *Printer, s *Service) (outErr error) {
             _, hasSecond := this.Next()
             if !hasFirst || hasSecond {
                 amount := "none"
+                statusCode := codes.NotFound
                 if hasSecond {
                     amount = "multiple"
+                    statusCode = codes.FailedPrecondition
                 }
-                return &Row_`, sName, `_`, camelQ(q), `{err: fmt.Errorf("expected exactly 1 result from query '`, camelQ(q), `' found %s", amount)}
+                return &Row_`, sName, `_`, camelQ(q), `{err: gstatus.Errorf(statusCode, "expected exactly 1 result from query '`, camelQ(q), `' found %s", amount)}
             }
             return first
         }