Which is the metaphor?

He swam in the sea of diamonds.
He swam in a beautiful sea.
The sea sparkled like diamonds.
The sea and diamonds both sparkle.

The metaphor in this sentence is "He swam in the sea of diamonds." The sea is being compared to a sea of diamonds, which is a metaphor.