I'd combine this with the primary key method:

```Ruby
def primary_key(value = nil)
  @primary_key = value if value
  @primary_key ||= default_primary_key
end
```