One reason why you might be struggling to fit Einstein's equation into your analogy is that you're using the rest state simplification.

only applies to stationary bodies.