sqlite使用blob類型存儲/訪問 結構體

/* open fire host and slora report data database */
int open_report_db(void)
{
    int               rv = -1;
    char              sql[SQL_COMMAND_LEN];
    char              *errmsg = NULL;
    const char        *db_file = REPORT_DB_FILE;

    /* If report database already exist, then we will just open it */
    if( 0==access(db_file, F_OK) )
    {
        if( SQLITE_OK != sqlite3_open(db_file, &s_report_db) )
        {
            log_err("open report database file '%s' failure\n", db_file);
             return -2;
        }
        log_nrml("open report database file '%s' ok\n", db_file);
        return 0;
    }

    pthread_mutex_lock(&s_reportdb_lock);

    /*  If report database file not exist, then we create it now */
    if( SQLITE_OK != sqlite3_open(db_file, &s_report_db) )
    {
        log_err("create report database file '%s' failure\n", db_file);
        pthread_mutex_unlock(&s_reportdb_lock);
        return -2;
    }
    log_nrml("open report database file '%s' ok\n", db_file);


    /*  Auto increase/decrease  */
    snprintf(sql, sizeof(sql), "pragma auto_vacuum=2;");
    if( SQLITE_OK != sqlite3_exec(s_report_db, sql, NULL, NULL, &errmsg) )
    {
        log_err("set pragma auto_vacuum=2 failure: %s\n", db_file, errmsg);
        rv = -4;
        goto FAILED;
    }

    /*  Create firehost table in the database */
    strncpy(sql, "CREATE TABLE firehost(id INTEGER PRIMARY KEY autoincrement, sent BOOL DEFAULT 0, packet BLOB);", sizeof(sql));
    if( SQLITE_OK != sqlite3_exec(s_report_db, sql, NULL, NULL, &errmsg) )
    {                                       
        log_err("create firehost table in report database file '%s' failure: %s\n", db_file, errmsg);
        rv = -5;                            
        goto FAILED;                        
    }                                       
    
    /*  Create elora table in the database */
    strncpy(sql, "CREATE TABLE elora(id INTEGER PRIMARY KEY autoincrement, sent BOOL DEFAULT 0, packet BLOB);", sizeof(sql));
    if( SQLITE_OK != sqlite3_exec(s_report_db, sql, NULL, NULL, &errmsg) )
    {                                       
        log_err("create elora table in report database file '%s' failure: %s\n", db_file, errmsg);
        rv = -5;                            
        goto FAILED;                        
    }                                       
 
    log_nrml("create report database file '%s' ok\n", db_file);
    pthread_mutex_unlock(&s_reportdb_lock);
    return 0; 

FAILED:
    log_err("Create report database '%s' failure and remove it\n", db_file);
    sqlite3_close(s_report_db);
    sqlite3_free(errmsg);
    unlink(db_file);
    pthread_mutex_unlock(&s_reportdb_lock);
    return rv;
}

int add_reportdb_firehost(firehost_pack_t *pack)
{
    int                      rv;
    sqlite3_stmt             *stat = NULL;

    if( !pack )
    {
        log_err("invalid input arguments\n");
        return -1;
    }

    if(!s_report_db)
    {
        log_nrml("report database not opened, try to open it now\n");
        if( open_report_db() < 0)
            return -2;
    }

    pthread_mutex_lock(&s_reportdb_lock);

    rv = sqlite3_prepare_v2(s_report_db, "insert into firehost(packet)values(?)", -1, &stat, NULL);
    if(SQLITE_OK!=rv || !stat)
    {
        log_err("firehost sqlite3_prepare_v2 failure\n");
        rv = -3;
        goto out;
    }

    cp_log_dump(LOG_LEVEL_NRML, (char *)pack, sizeof(*pack));

    if( SQLITE_OK != sqlite3_bind_blob(stat, 1, pack, sizeof(*pack), NULL) )
    {
        log_err("firehost sqlite3_bind_blob failure\n");
        rv = -4;
        goto out;
    }

    rv = sqlite3_step(stat);
    if( SQLITE_DONE!=rv && SQLITE_ROW!=rv )
    {
        log_err("firehost sqlite3_step failure\n");
        rv = -5;
        goto out;
    }

    sqlite3_finalize(stat);

    pthread_mutex_unlock(&s_reportdb_lock);
    log_nrml("add new firehost report data into database ok\n");

    return 0;

out:
    pthread_mutex_unlock(&s_reportdb_lock);
    log_nrml("add new firehost report data into database failure, rv=%d\n", rv);
    return rv;
}

int mark_reportdb_firehost_sent(int record_id)
{
    char                     sql[SQL_COMMAND_LEN];
    char                     *errmsg = NULL;

    if(record_id <= 0)
    {
        log_err("invalid input arguments\n");
        return -1;
    }

    if(!s_report_db)
    {
        log_nrml("report database not opened, try to open it now\n");
        if( open_report_db() < 0)
            return -2;
    }

    pthread_mutex_lock(&s_reportdb_lock);

    memset(sql, 0, sizeof(sql));
    snprintf(sql, sizeof(sql), "update firehost set sent=1 where id=%d;", record_id);
    if( SQLITE_OK != sqlite3_exec(s_report_db, sql, NULL, NULL, &errmsg) )
    {
        log_err("mark firehost packet[%d] already sent from database failure: %s\n", record_id, errmsg);
        sqlite3_free(errmsg);
        pthread_mutex_unlock(&s_reportdb_lock);
        return -2;
    }

    pthread_mutex_unlock(&s_reportdb_lock);
    log_nrml("mark firehost packet[%d] already sent from database ok\n", record_id);
    return 0;
}


int query_reportdb_firehost_unsent(int *record_id, firehost_pack_t *pack)
{
    int                      rv;
    sqlite3_stmt             *stat = NULL;
    const void               *blob_ptr;

    if( !record_id || !pack )
    {
        log_err("invalid input arguments\n");
        return -1;
    }

    if(!s_report_db)
    {
        log_nrml("report database not opened, try to open it now\n");
        if( open_report_db() < 0)
            return -2;
    }

    pthread_mutex_lock(&s_reportdb_lock);

    /* Only query the first unsent packet record */
    rv = sqlite3_prepare_v2(s_report_db, "select id,packet from firehost where sent=0 limit 0,1;", -1, &stat, NULL);
    if(SQLITE_OK!=rv || !stat)
    {
        log_err("firehost sqlite3_prepare_v2 failure\n");
        rv = -3;
        goto out;
    }

    rv = sqlite3_step(stat);
    if( SQLITE_DONE!=rv && SQLITE_ROW!=rv )
    {
        log_err("firehost sqlite3_step failure\n");
        rv = -5;
        goto out;
    }

    /* 2rd argument<0> means first segement is id */
    *record_id = sqlite3_column_int(stat, 0); 

    /* 2rd argument<1> means first segement is id */
    //blob_bytes = sqlite3_column_bytes(stat, 1 );
    blob_ptr = sqlite3_column_blob(stat, 1);

    memcpy(pack, blob_ptr, sizeof(*pack));

    log_trace("Read firehost packet from database:\n");
    cp_log_dump(LOG_LEVEL_TRACE, (char *)pack, sizeof(*pack));

    sqlite3_finalize(stat);

    pthread_mutex_unlock(&s_reportdb_lock);
    log_nrml("query new firehost report data into database ok\n");
    return 0;

out:
    pthread_mutex_unlock(&s_reportdb_lock);
    log_nrml("query new firehost report data into database failure, rv=%d\n", rv);
    return rv;
}