Get latest snapshot if no id is provided

palomachain · Jul 26, 2022 · 0dffcab · 0dffcab
1 parent 2f653d8
commit 0dffcab
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/x/valset/keeper/grpc_query_get_snapshot_by_id.go b/x/valset/keeper/grpc_query_get_snapshot_by_id.go
@@ -15,8 +15,15 @@ func (k Keeper) GetSnapshotByID(goCtx context.Context, req *types.QueryGetSnapsh
 	}
 
 	ctx := sdk.UnwrapSDKContext(goCtx)
+	var snapshot *types.Snapshot
+	var err error
+
+	if req.SnapshotId == 0 {
+		snapshot, err = k.GetCurrentSnapshot(ctx)
+	} else {
+		snapshot, err = k.FindSnapshotByID(ctx, req.SnapshotId)
+	}
 
-	snapshot, err := k.FindSnapshotByID(ctx, req.SnapshotId)
 	if err != nil {
 		return nil, err
 	}