panic handler

VtanSen · Jun 25, 2018 · 0a39ded · 0a39ded
1 parent d92988d
commit 0a39ded
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/cargo/domain/service/cargo_service.go b/cargo/domain/service/cargo_service.go
@@ -28,17 +28,18 @@ func (t *CargoService) Create(cargoId string, afterDays uint) {
 
 func (t *CargoService) Delay(cargoId string, days uint) {
     cargo := t.repo.Get(cargoId)
-    if cargo != nil {
-        cargo.Delay(days)
-        t.repo.Update(cargo)
-        t.provider.Confirm(cargo)
+    if cargo == nil {
+        panic("not found cargo by cargoId")
     }
+    cargo.Delay(days)
+    t.repo.Update(cargo)
+    t.provider.Confirm(cargo)
 }
 
 func (t *CargoService) GetAfterDays(cargoId string) uint {
     cargo := t.repo.Get(cargoId)
-    if cargo != nil {
-        return cargo.GetAfterDays()
+    if cargo == nil {
+        panic("not found cargo by cargoId")
     }
-    panic("invalid cargoId")
+    return cargo.GetAfterDays()
 }