return sequenceNumber.incrementAndGet();
    }

    /**
     * Updates the heartbeat timestamp to current time and resets failure count.
     */
    public void updateHeartbeat() {
        this.lastHeartbeat = System.currentTimeMillis();
        this.heartbeatFailures = 0;
    }

    /**
     * Increments the heartbeat failure count.
     */

    @Test
    void testHeartbeatUpdate() throws InterruptedException {
        long initialTime = registration.getLastHeartbeat();

        Thread.sleep(50); // Ensure sufficient time difference
        registration.updateHeartbeat();

        // After update, should not be expired with long timeout
        assertFalse(registration.isExpired(60000));

        // Verify heartbeat was actually updated

            WitnessHeartbeatResponse response = rpcClient.heartbeat(request);

            if (response != null && response.isSuccess()) {
                registration.updateHeartbeat();
            } else {
                log.warn("Witness heartbeat failed for: {}", registration.getRegistrationId());
                registration.incrementHeartbeatFailures();
            }

               Integer.toHexString(System.identityHashCode(this));
    }
    
    public long getNextSequenceNumber() {
        return sequenceNumber.incrementAndGet();
    }
    
    public void updateHeartbeat() {
        this.lastHeartbeat = System.currentTimeMillis();
    }
    
    public boolean isExpired(long timeoutMs) {
        return System.currentTimeMillis() - lastHeartbeat > timeoutMs;
    }