避免代碼與golang中的接口重複

Question

我正在使用Google Cloud Datastore（或Google應用程序引擎數據存儲）來存儲不同的對象。大多數這些對象是相似的，因此我最終導致大量的代碼重複。

作爲示例，下面是兩個不同對象的創建方法：account和index。

func (account *Account) Create(ctx context.Context) (*Account, error) { 
    if ret, err := datastore.Put(ctx, account.newKey(ctx), account); err != nil { 
    log.Errorf(ctx, "Error while creating Account : %v\n", err) 
    return nil, err 
    } else { 
    account.Id = strconv.FormatInt(ret.IntID(), 10) 
    } 

    return account, nil 
} 

---

func (index *Index) Create(ctx context.Context) (*Index, error) { 
    if ret, err := datastore.Put(ctx, index.newKey(ctx), index); err != nil { 
    log.Errorf(ctx, "Error while creating Index : %v\n", err) 
    return nil, err 
    } else { 
    index.Id = strconv.FormatInt(ret.IntID(), 10) 
    } 

    return index, nil 
} 

正如你可以看到，這兩個片段是相同的，對於保持架型，返回類型，以及錯誤消息節選。

什麼是避免這種代碼重複的慣用方法？

Answer 1

使用界面來定義方法NewKey() & SetID()

type Entity interface { 
    SetId(id int64) 
    NewKey(c context.Context) *datastore.Key 
} 

func create(c context.Context, entity Entity) error { 
    if ret, err := datastore.Put(c, entity.NewKey(c), entity); err != nil { 
    log.Errorf(c, "Error while creating entity: %v\n", err) 
    return err 
    } else { 
    entity.SetId(ret.IntID()) 
    return nil 
    } 
} 

func (index *Index) Create(c context.Context) (*Index, error) { 
    return index, create(c, index) 
} 

func (account *Account) Create(c context.Context) (*Account, error) { 
    return account, create(c, account) 
}