graph/db: version MarkEdgeZombie

diff --git a/graph/builder.go b/graph/builder.go
index 2e17ef5..d63cab5 100644
--- a/graph/builder.go
+++ b/graph/builder.go
@@ -963,7 +963,7 @@ func (b *Builder) MarkZombieEdge(chanID uint64) error {
 	// node pubkeys so this edge can't be resurrected.
 	var zeroKey [33]byte
 	err := b.cfg.Graph.MarkEdgeZombie(
-		context.TODO(), chanID, zeroKey, zeroKey,
+		context.TODO(), lnwire.GossipVersion1, chanID, zeroKey, zeroKey,
 	)
 	if err != nil {
 		return fmt.Errorf("unable to mark spent chan(id=%v) as a "+
diff --git a/graph/db/graph.go b/graph/db/graph.go
index a082d5b..f564d88 100644
--- a/graph/db/graph.go
+++ b/graph/db/graph.go
@@ -566,12 +566,13 @@ func (c *ChannelGraph) FilterKnownChanIDs(ctx context.Context,
 }
 
 // MarkEdgeZombie attempts to mark a channel identified by its channel ID as a
-// zombie. This method is used on an ad-hoc basis, when channels need to be
-// marked as zombies outside the normal pruning cycle.
-func (c *ChannelGraph) MarkEdgeZombie(ctx context.Context, chanID uint64,
+// zombie for the given gossip version. This method is used on an ad-hoc basis,
+// when channels need to be marked as zombies outside the normal pruning cycle.
+func (c *ChannelGraph) MarkEdgeZombie(ctx context.Context,
+	v lnwire.GossipVersion, chanID uint64,
 	pubKey1, pubKey2 [33]byte) error {
 
-	err := c.db.MarkEdgeZombie(ctx, chanID, pubKey1, pubKey2)
+	err := c.db.MarkEdgeZombie(ctx, v, chanID, pubKey1, pubKey2)
 	if err != nil {
 		return err
 	}
diff --git a/graph/db/graph_test.go b/graph/db/graph_test.go
index 90d6748..90a2fc5 100644
--- a/graph/db/graph_test.go
+++ b/graph/db/graph_test.go
@@ -2964,11 +2964,13 @@ func TestFilterKnownChanIDsZombieRevival(t *testing.T) {
 
 	// Mark channel 1 and 2 as zombies.
 	err := graph.MarkEdgeZombie(
-		ctx, scid1.ToUint64(), [33]byte{}, [33]byte{},
+		ctx, lnwire.GossipVersion1, scid1.ToUint64(),
+		[33]byte{}, [33]byte{},
 	)
 	require.NoError(t, err)
 	err = graph.MarkEdgeZombie(
-		ctx, scid2.ToUint64(), [33]byte{}, [33]byte{},
+		ctx, lnwire.GossipVersion1, scid2.ToUint64(),
+		[33]byte{}, [33]byte{},
 	)
 	require.NoError(t, err)
 
@@ -3323,7 +3325,8 @@ func TestStressTestChannelGraphAPI(t *testing.T) {
 				}
 
 				return graph.MarkEdgeZombie(
-					ctx, channel.id.ToUint64(),
+					ctx, lnwire.GossipVersion1,
+					channel.id.ToUint64(),
 					node1.PubKeyBytes,
 					node2.PubKeyBytes,
 				)
@@ -4626,7 +4629,8 @@ func TestGraphZombieIndex(t *testing.T) {
 	// If we mark the edge as a zombie manually, then it should show up as
 	// being a zombie once again.
 	err = graph.MarkEdgeZombie(
-		ctx, edge.ChannelID, node1.PubKeyBytes, node2.PubKeyBytes,
+		ctx, lnwire.GossipVersion1, edge.ChannelID,
+		node1.PubKeyBytes, node2.PubKeyBytes,
 	)
 	require.NoError(t, err, "unable to mark edge as zombie")
 
diff --git a/graph/db/interfaces.go b/graph/db/interfaces.go
index ae07bbe..e701e61 100644
--- a/graph/db/interfaces.go
+++ b/graph/db/interfaces.go
@@ -327,10 +327,11 @@ type Store interface { //nolint:interfacebloat
 	ChannelView(ctx context.Context) ([]EdgePoint, error)
 
 	// MarkEdgeZombie attempts to mark a channel identified by its channel
-	// ID as a zombie. This method is used on an ad-hoc basis, when channels
-	// need to be marked as zombies outside the normal pruning cycle.
-	MarkEdgeZombie(ctx context.Context, chanID uint64,
-		pubKey1, pubKey2 [33]byte) error
+	// ID as a zombie for the given gossip version. This method is used on
+	// an ad-hoc basis, when channels need to be marked as zombies outside
+	// the normal pruning cycle.
+	MarkEdgeZombie(ctx context.Context, v lnwire.GossipVersion,
+		chanID uint64, pubKey1, pubKey2 [33]byte) error
 
 	// MarkEdgeLive clears an edge from our zombie index for the given
 	// gossip version, deeming it as live.
diff --git a/graph/db/kv_store.go b/graph/db/kv_store.go
index b39432d..cd8a928 100644
--- a/graph/db/kv_store.go
+++ b/graph/db/kv_store.go
@@ -4289,10 +4289,14 @@ func (c *KVStore) ChannelView(_ context.Context) ([]EdgePoint, error) {
 }
 
 // MarkEdgeZombie attempts to mark a channel identified by its channel ID as a
-// zombie. This method is used on an ad-hoc basis, when channels need to be
-// marked as zombies outside the normal pruning cycle.
-func (c *KVStore) MarkEdgeZombie(_ context.Context, chanID uint64,
-	pubKey1, pubKey2 [33]byte) error {
+// zombie for the given gossip version. This method is used on an ad-hoc basis,
+// when channels need to be marked as zombies outside the normal pruning cycle.
+func (c *KVStore) MarkEdgeZombie(_ context.Context, v lnwire.GossipVersion,
+	chanID uint64, pubKey1, pubKey2 [33]byte) error {
+
+	if v != lnwire.GossipVersion1 {
+		return ErrVersionNotSupportedForKVDB
+	}
 
 	c.cacheMu.Lock()
 	defer c.cacheMu.Unlock()
diff --git a/graph/db/sql_store.go b/graph/db/sql_store.go
index c3ac93a..c9e7384 100644
--- a/graph/db/sql_store.go
+++ b/graph/db/sql_store.go
@@ -1779,12 +1779,16 @@ func (s *SQLStore) FilterChannelRange(ctx context.Context, startHeight,
 }
 
 // MarkEdgeZombie attempts to mark a channel identified by its channel ID as a
-// zombie. This method is used on an ad-hoc basis, when channels need to be
-// marked as zombies outside the normal pruning cycle.
+// zombie for the given gossip version. This method is used on an ad-hoc basis,
+// when channels need to be marked as zombies outside the normal pruning cycle.
 //
 // NOTE: part of the Store interface.
-func (s *SQLStore) MarkEdgeZombie(ctx context.Context, chanID uint64,
-	pubKey1, pubKey2 [33]byte) error {
+func (s *SQLStore) MarkEdgeZombie(ctx context.Context, v lnwire.GossipVersion,
+	chanID uint64, pubKey1, pubKey2 [33]byte) error {
+
+	if !isKnownGossipVersion(v) {
+		return fmt.Errorf("unsupported gossip version: %d", v)
+	}
 
 	s.cacheMu.Lock()
 	defer s.cacheMu.Unlock()
@@ -1794,7 +1798,7 @@ func (s *SQLStore) MarkEdgeZombie(ctx context.Context, chanID uint64,
 	err := s.db.ExecTx(ctx, sqldb.WriteTxOpt(), func(db SQLQueries) error {
 		return db.UpsertZombieChannel(
 			ctx, sqlc.UpsertZombieChannelParams{
-				Version:  int16(lnwire.GossipVersion1),
+				Version:  int16(v),
 				Scid:     chanIDB,
 				NodeKey1: pubKey1[:],
 				NodeKey2: pubKey2[:],
@@ -1806,8 +1810,8 @@ func (s *SQLStore) MarkEdgeZombie(ctx context.Context, chanID uint64,
 			"(channel_id=%d): %w", chanID, err)
 	}
 
-	s.rejectCache.remove(lnwire.GossipVersion1, chanID)
-	s.chanCache.remove(lnwire.GossipVersion1, chanID)
+	s.rejectCache.remove(v, chanID)
+	s.chanCache.remove(v, chanID)
 
 	return nil
 }