localcli.cpp

来自「FastDb是高效的内存数据库系统」· C++ 代码 · 共 3,024 行 · 第 1/5 页

  }

  stmt->for_update = for_update;
  stmt->oid = 0;

  if (!stmt->prepared)
  {
    int tkn;
    sql_scanner scanner(stmt->sql.base());

    if (scanner.get() != tkn_select)
    {
      return cli_bad_statement;
    }

    if ((tkn = scanner.get()) == tkn_all)
    {
      tkn = scanner.get();
    }

    if (tkn == tkn_from && scanner.get() == tkn_ident)
    {
      int rc = match_columns(scanner.identifier(), stmt);

      if (rc != cli_ok)
      {
        return rc;
      }
    }
    else
    {
      return cli_bad_statement;
    }

    char* p = scanner.current_position(), *q = p;
    parameter_binding* pb = stmt->params;
    stmt->query.reset();

    while (*p != '\0')
    {
      if (*p == '\'')
      {
        do
        {
          do
          {
            p += 1;
          }
          while (*p != '\0' && *p != '\'');

          if (*p == '\0')
          {
            return cli_bad_statement;
          }
        }
        while (*++p == '\'');
      }
      else if (*p == '%')
      {
        if (p != q)
        {
          *p = '\0';
          stmt->query.append(dbQueryElement::qExpression, q);
        }

        if (pb->var_ptr == NULL)
        {
          return cli_unbound_parameter;
        }

        switch(pb->var_type)
        {

        case cli_oid:
          stmt->query.append(dbQueryElement::qVarReference, pb->var_ptr);
          break;

        case cli_bool:
          stmt->query.append(dbQueryElement::qVarBool, pb->var_ptr);
          break;

        case cli_int1:
          stmt->query.append(dbQueryElement::qVarInt1, pb->var_ptr);
          break;

        case cli_int2:
          stmt->query.append(dbQueryElement::qVarInt2, pb->var_ptr);
          break;

        case cli_int4:
          stmt->query.append(dbQueryElement::qVarInt4, pb->var_ptr);
          break;

        case cli_int8:
          stmt->query.append(dbQueryElement::qVarInt8, pb->var_ptr);
          break;

        case cli_real4:
          stmt->query.append(dbQueryElement::qVarReal4, pb->var_ptr);
          break;

        case cli_real8:
          stmt->query.append(dbQueryElement::qVarReal8, pb->var_ptr);
          break;

        case cli_asciiz:
          stmt->query.append(dbQueryElement::qVarString, pb->var_ptr);
          break;

        case cli_pasciiz:
          stmt->query.append(dbQueryElement::qVarStringPtr, pb->var_ptr);
          break;

        case cli_array_of_oid:
          stmt->query.append(dbQueryElement::qVarArrayOfRef, pb->var_ptr);
          break;

        default:
          return cli_unsupported_type;

        }

        while (isalnum((unsigned char)*++p) || *p == '_')

          ;
        q = p;

        pb = pb->next;
      }
      else
      {
        p += 1;
      }
    }

    if (p != q)
    {
      stmt->query.append(dbQueryElement::qExpression, q);
    }

    stmt->prepared = true;
  }

  stmt->cursor.setTable(stmt->table);
  stmt->cursor.reset();
  return stmt->cursor.select(stmt->query, for_update ? dbCursorForUpdate : dbCursorViewOnly);
}


int dbCLI::fetch_columns(statement_desc* stmt)
{
  stmt->first_fetch = false;

  if (stmt->cursor.isEmpty())
  {
    return cli_not_found;
  }

  stmt->updated = false;

  if (stmt->record_struct != NULL)
  {
    stmt->cursor.fetch();
    return cli_ok;
  }

  char* data = (char*)stmt->session->db->getRow(stmt->cursor.currId);

  for (column_binding* cb = stmt->columns; cb != NULL; cb = cb->next)
  {
    dbFieldDescriptor* fd = cb->field;
    char* src = data + fd->dbsOffs;
    char* dst = (char*)cb->var_ptr;
    // Allow fetching of structures with one component

    if (fd->type == dbField::tpStructure && fd->components->next == NULL)
    {
      fd = fd->components;
    }

    switch (fd->type)
    {

    case dbField::tpBool:

      switch (cb->var_type)
      {

      case cli_bool:
        *(cli_bool_t*)dst = *(bool*)src;
        continue;

      case cli_int1:
        *(cli_int1_t*)dst = *(bool*)src ? 1 : 0;
        continue;

      case cli_int2:
        *(cli_int2_t*)dst = *(bool*)src ? 1 : 0;
        continue;

      case cli_int4:
        *(cli_int4_t*)dst = *(bool*)src ? 1 : 0;
        continue;

      case cli_int8:
        *(db_int8*)dst = *(bool*)src ? 1 : 0;
        continue;

      case cli_real4:
        *(cli_real4_t*)dst = (cli_real4_t)(*(bool*)src ? 1 : 0);
        continue;

      case cli_real8:
        *(cli_real8_t*)dst = *(bool*)src ? 1 : 0;
        continue;
      }

      break;

    case dbField::tpInt1:

      switch (cb->var_type)
      {

      case cli_bool:
        *(cli_bool_t*)dst = *(int1*)src != 0;
        continue;

      case cli_int1:
        *(cli_int1_t*)dst = *(int1*)src;
        continue;

      case cli_int2:
        *(cli_int2_t*)dst = *(int1*)src;
        continue;

      case cli_int4:
        *(cli_int4_t*)dst = *(int1*)src;
        continue;

      case cli_int8:
        *(db_int8*)dst = *(int1*)src;
        continue;

      case cli_real4:
        *(cli_real4_t*)dst = *(int1*)src;
        continue;

      case cli_real8:
        *(cli_real8_t*)dst = *(int1*)src;
        continue;
      }

      break;

    case dbField::tpInt2:

      switch (cb->var_type)
      {

      case cli_bool:
        *(cli_bool_t*)dst = *(int2*)src != 0;
        continue;

      case cli_int1:
        *(cli_int1_t*)dst = (int1)*(int2*)src;
        continue;

      case cli_int2:
        *(cli_int2_t*)dst = *(int2*)src;
        continue;

      case cli_int4:
        *(cli_int4_t*)dst = *(int2*)src;
        continue;

      case cli_int8:
        *(db_int8*)dst = *(int2*)src;
        continue;

      case cli_real4:
        *(cli_real4_t*)dst = *(int2*)src;
        continue;

      case cli_real8:
        *(cli_real8_t*)dst = *(int2*)src;
        continue;
      }

      break;

    case dbField::tpInt4:

      switch (cb->var_type)
      {

      case cli_bool:
        *(cli_bool_t*)dst = *(int4*)src != 0;
        continue;

      case cli_int1:
        *(cli_int1_t*)dst = (cli_int1_t)*(int4*)src;
        continue;

      case cli_int2:
        *(cli_int2_t*)dst = (cli_int2_t)*(int4*)src;
        continue;

      case cli_int4:

      case cli_autoincrement:
        *(cli_int4_t*)dst = *(int4*)src;
        continue;

      case cli_int8:
        *(db_int8*)dst = *(int4*)src;
        continue;

      case cli_real4:
        *(cli_real4_t*)dst = (cli_real4_t)*(int4*)src;
        continue;

      case cli_real8:
        *(cli_real8_t*)dst = *(int4*)src;
        continue;
      }

      break;

    case dbField::tpInt8:

      switch (cb->var_type)
      {

      case cli_bool:
        *(cli_bool_t*)dst = *(db_int8*)src != 0;
        continue;

      case cli_int1:
        *(cli_int1_t*)dst = (cli_int1_t)*(db_int8*)src;
        continue;

      case cli_int2:
        *(cli_int2_t*)dst = (cli_int2_t)*(db_int8*)src;
        continue;

      case cli_int4:
        *(cli_int4_t*)dst = (cli_int4_t)*(db_int8*)src;
        continue;

      case cli_int8:
        *(db_int8*)dst = *(db_int8*)src;
        continue;

      case cli_real4:
        *(cli_real4_t*)dst = (cli_real4_t)*(db_int8*)src;
        continue;

      case cli_real8:
        *(cli_real8_t*)dst = (cli_real8_t)*(db_int8*)src;
        continue;
      }

      break;

    case dbField::tpReal4:

      switch (cb->var_type)
      {

      case cli_bool:
        *(cli_bool_t*)dst = *(real4*)src != 0;
        continue;

      case cli_int1:
        *(cli_int1_t*)dst = (cli_int1_t)*(real4*)src;
        continue;

      case cli_int2:
        *(cli_int2_t*)dst = (cli_int2_t)*(real4*)src;
        continue;

      case cli_int4:
        *(cli_int4_t*)dst = (cli_int4_t)*(real4*)src;
        continue;

      case cli_int8:
        *(db_int8*)dst = (db_int8)*(real4*)src;
        continue;

      case cli_real4:
        *(cli_real4_t*)dst = *(real4*)src;
        continue;

      case cli_real8:
        *(cli_real8_t*)dst = *(real4*)src;
        continue;
      }

      break;

    case dbField::tpReal8:

      switch (cb->var_type)
      {

      case cli_bool:
        *(cli_bool_t*)dst = *(real8*)src != 0;
        continue;

      case cli_int1:
        *(cli_int1_t*)dst = (cli_int1_t)*(real8*)src;
        continue;

      case cli_int2:
        *(cli_int2_t*)dst = (cli_int2_t)*(real8*)src;
        continue;

      case cli_int4:
        *(cli_int4_t*)dst = (cli_int4_t)*(real8*)src;
        continue;

      case cli_int8:
        *(db_int8*)dst = (db_int8)*(real8*)src;
        continue;

      case cli_real4:
        *(cli_real4_t*)dst = (real4)*(real8*)src;
        continue;

      case cli_real8:
        *(cli_real8_t*)dst = *(real8*)src;
        continue;
      }

      break;

    case dbField::tpReference:

      if (cb->var_type == cli_oid)
      {
        *(cli_oid_t*)dst = *(oid_t*)src;
        continue;
      }

      break;

    case dbField::tpString:

      if (cb->var_type == cli_asciiz || cb->var_type == cli_pasciiz)
      {
        if (cb->var_type == cli_pasciiz)
        {
          dst = *(char**)dst;
        }

        dbVarying* v = (dbVarying*)src;
        int size = v->size;

        if (cb->set_fnc != NULL)
        {
          dst = (char*)cb->set_fnc(cli_asciiz, dst, size,
                                   cb->name, stmt->id, data + v->offs);

          if (dst != NULL)
          {
            memcpy(dst, data + v->offs, size);
          }
        }
        else
        {
          int n = size;

          if (cb->var_len != NULL)
          {
            if (n > *cb->var_len)
            {
              n = *cb->var_len;
            }

            *cb->var_len = size;
          }

          memcpy(dst, data + v->offs, n);
        }

        continue;
      }

      break;

    case dbField::tpArray:

      if (cb->var_type >= cli_array_of_oid && cb->var_type <= cli_array_of_string)
      {
        dbVarying* v = (dbVarying*)src;
        int n = v->size;
        src = data + v->offs;

        if (cb->set_fnc != NULL)
        {
          dst = (char*)cb->set_fnc(cb->var_type, dst, n,
                                   cb->name, stmt->id, src);

          if (dst == NULL)
          {
            continue;
          }
        }
        else
        {
          int size = n;

          if (cb->var_len != NULL)
          {
            if (n > *cb->var_len)
            {
              n = *cb->var_len;
            }

            *cb->var_len = size;
          }
        }

        switch (fd->components->type)
        {

        case dbField::tpBool:

          if (cb->var_type == cli_array_of_bool)
          {
            cli_bool_t* dst_elem = (cli_bool_t*)dst;
            bool*       src_elem = (bool*)src;

            while (--n >= 0)
              *dst_elem++ = *src_elem++;

            continue;
          }

          break;

        case dbField::tpInt1:

          if (cb->var_type == cli_array_of_int1)
          {
            cli_int1_t* dst_elem = (cli_int1_t*)dst;
            int1*       src_elem = (int1*)src;

            while (--n >= 0)
              *dst_elem++ = *src_elem++;

            continue;
          }

          break;

        case dbField::tpInt2:

          if (cb->var_type == cli_array_of_int2)
          {
            cli_int2_t* dst_elem = (cli_int2_t*)dst;
            int2*       src_elem = (int2*)src;

            while (--n >= 0)
              *dst_elem++ = *src_elem++;

            continue;
          }

          break;

        case dbField::tpInt4:

          if (cb->var_type == cli_array_of_int4)
          {
            cli_int4_t* dst_elem = (cli_int4_t*)dst;
            int4*       src_elem = (int4*)src;

            while (--n >= 0)
              *dst_elem++ = *src_elem++;

            continue;
          }

          break;

        case dbField::tpInt8:

          if (cb->var_type == cli_array_of_int8)
          {
            cli_int8_t* dst_elem = (cli_int8_t*)dst;
            db_int8*       src_elem = (db_int8*)src;

            while (--n >= 0)
              *dst_elem++ = *src_elem++;

            continue;
          }

          break;

        case dbField::tpReal4:

          if (cb->var_type == cli_array_of_real4)