Question

我在DB帮助器类中有两个方法，它们对两个不同的数据库实体基本相同，我想重构它们以避免重复代码。

第一个实体：

- (void) insertOrUpdateEntityA:(NSDictionary*)data {
    sqlite3_stmt *exists_stmt;
    if(sqlite3_prepare_v2(database, RMSQLEntityAExists, -1, &exists_stmt, NULL) == SQLITE_OK) {
        [RMStoreDB bindPrimaryKey:exists_stmt data:data from:1];
        if (sqlite3_step(exists_stmt) == SQLITE_ROW) {
            int count = sqlite3_column_int(exists_stmt, 1);
            sqlite3_stmt *update_stmt;
            if (count) { // Update
                if (sqlite3_prepare_v2(database, RMSQLEntityAUpdate, -1, &update_stmt, NULL) == SQLITE_OK) {
                    int index = [RMStoreDB bindEntityA:update_stmt data:data from:1];
                    [RMStoreDB bindPrimaryKey:update_stmt data:data from:index];
                }
            } else { // Insert
                if (sqlite3_prepare_v2(database, RMSQLEntityAInsert, -1, &update_stmt, NULL) == SQLITE_OK) {
                    int index = [RMStoreDB bindPrimaryKey:update_stmt data:data from:1];
                    [RMStoreDB bindEntityA:update_stmt data:data from:index];
                }           
            }
            sqlite3_step(update_stmt);
            sqlite3_finalize(update_stmt);
        }           
    }
    sqlite3_finalize(exists_stmt);
}

第二个实体：

 - (void) insertOrUpdateEntityB:(NSDictionary*)data {
    sqlite3_stmt *exists_stmt;
    if(sqlite3_prepare_v2(database, RMSQLEntityBExists, -1, &exists_stmt, NULL) == SQLITE_OK) {
        [RMStoreDB bindPrimaryKey:exists_stmt data:data from:1];
        if (sqlite3_step(exists_stmt) == SQLITE_ROW) {
            int count = sqlite3_column_int(exists_stmt, 1);
            sqlite3_stmt *update_stmt;
            if (count) { // Update
                if (sqlite3_prepare_v2(database, RMSQLEntityBUpdate, -1, &update_stmt, NULL) == SQLITE_OK) {
                    int index = [RMStoreDB bindEntityB:update_stmt data:data from:1];
                    [RMStoreDB bindPrimaryKey:update_stmt data:data from:index];
                }
            } else { // Insert
                if (sqlite3_prepare_v2(database, RMSQLEntityBInsert, -1, &update_stmt, NULL) == SQLITE_OK) {
                    int index = [RMStoreDB bindPrimaryKey:update_stmt data:data from:1];
                    [RMStoreDB bindEntityB:update_stmt data:data from:index];
                }           
            }
            sqlite3_step(update_stmt);
            sqlite3_finalize(update_stmt);
        }           
    }
    sqlite3_finalize(exists_stmt);
}

差异是用于SQL语句的常量（RMSQLEntityAExists，RMSQLEntityBExists等）以及用于将数据绑定到SQLite语句的方法（bindEntityA和{{ 1}}）。后者是我发现特别具有挑战性的概括。

如何重构这两种方法？我应该吗？

Answer 1

首先，你不应该为此使用继承。继承用于共享接口，而不是用于共享实现。您有两种方法具有非常相似的实现，但接口不同。

其次，考虑GoF声明的主要原则之一：确定哪些变化并将其封装起来。在这种情况下，最简单的方法是提取一些有意义的方法。如果没有别的，这将使您的代码更容易阅读。你应该拍摄这样的东西（我正在使用伪代码，因为我真的不知道你的代码在做什么）：

- (void)insertOrDeleteItem:(NSDictionary *)item {
    if ([self databaseAppearsToBeWorking]]) {
        row = [self findRowForItem:item];
        if (row) {
            [self updateRow:row withItem:item];
        } else {
            [self insertItem:item];
        }
    }
}

一旦你看到的东西看起来更像那样，那么共性就会更清楚地呈现出来，或者你会发现这些方法实际上应该保持不同。

Answer 2

使用根对象中的该方法将两个实体子类化为公共对象。

可以使用在子类init方法中设置的ivars修改差异，并调用mutated方法

像这样...

@interface RootEntity : MyObject {

    NSString *statementmech1;
    NSString *statementmech2;
    NSString *statementmech3;
}

@interface Entity1 : RootEntity {

}

@interface Entity2 : RootEntity {

}

和

@implementation RootEntity

-(void)variantMethod
{
  //varies in subclasses
}

    - (void) insertOrUpdateItem:(NSDictionary*)item {
        sqlite3_stmt *exists_stmt;
        if(sqlite3_prepare_v2(database, statementmech1 , -1, &exists_stmt, NULL) == SQLITE_OK) {
            [RMStoreDB bindPrimaryKey:exists_stmt data:item from:1];
            if (sqlite3_step(exists_stmt) == SQLITE_ROW) {
                int count = sqlite3_column_int(exists_stmt, 1);
                sqlite3_stmt *update_stmt;
                if (count) { // Update
                    if (sqlite3_prepare_v2(database, statementmech2, -1, &update_stmt, NULL) == SQLITE_OK) {
                        int index = [RMStoreDB bindItem:update_stmt data:item from:1];
                        [RMStoreDB bindPrimaryKey:update_stmt data:item from:index];
                    }
                } else { // Insert
                    if (sqlite3_prepare_v2(database, statementmech3, -1, &update_stmt, NULL) == SQLITE_OK) {
                        [self variantMethod];
                    }           
                }
                sqlite3_step(update_stmt);
                sqlite3_finalize(update_stmt);
            }           
        }
        sqlite3_finalize(exists_stmt);
    }

Answer 3

我要改变的第一件事就是增加早期回报;较少缩进的代码不那么令人生畏：

- (void) insertOrUpdateItem:(NSDictionary*)item {
    sqlite3_stmt *exists_stmt;
    if (sqlite3_prepare_v2(database, RMSQLItemExists, -1, &exists_stmt, NULL) != SQLITE_OK) 
        return;
    [RMStoreDB bindPrimaryKey:exists_stmt data:item from:1];
    if (sqlite3_step(exists_stmt) != SQLITE_ROW)
        return;
    int count = sqlite3_column_int(exists_stmt, 1);
    sqlite3_stmt *update_stmt;
    if (count) { // Update
        if (sqlite3_prepare_v2(database, RMSQLItemUpdate, -1, &update_stmt, NULL) == SQLITE_OK) {
            int index = [RMStoreDB bindItem:update_stmt data:item from:1];
            [RMStoreDB bindPrimaryKey:update_stmt data:item from:index];
        }
    } else { // Insert
        if (sqlite3_prepare_v2(database, RMSQLItemInsert, -1, &update_stmt, NULL) == SQLITE_OK) {
            int index = [RMStoreDB bindPrimaryKey:update_stmt data:item from:1];
            [RMStoreDB bindItem:update_stmt data:item from:index];
        }           
    }
    sqlite3_step(update_stmt);
    sqlite3_finalize(update_stmt);
    sqlite3_finalize(exists_stmt);
}

除此之外，看起来差别在于一个用于物品，另一个用于收藏。如果您可以将项目放入集合中，然后调用insertOrUpdateCollection（），那么您就完成了。我不知道客观C;在Java中，我们将使用Collections.singleton（），正如文档所述：

返回包含的不可变集只有指定的对象。

如何重构这个Objective-C代码

3 个答案: