You've already forked mariadb-columnstore-engine
							
							
				mirror of
				https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
				synced 2025-11-03 17:13:17 +03:00 
			
		
		
		
	* fix(plugin): MCOL-4740: This fixes update rows counter for multi-table update For UPDATEs involving a single table, the server call to handler::direct_update_rows() is used to correctly set the count for the number of updated rows in the UPDATE statement. However, for UPDATEs involving multi-tables, the server does not call handler::direct_update_rows(). This patch adds support to correctly report the number of updated rows to the client by setting multi_update::updated and multi_update::found in handler::rnd_end(). * fix(plugin): MCOL-4740: this is to addres the original patch QA found in the original patch --------- Co-authored-by: Roman Nozdrin <rnozdrin@mariadb.com> Co-authored-by: drrtuy <roman.nozdrin@mariadb.com>
		
			
				
	
	
		
			213 lines
		
	
	
		
			7.0 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			213 lines
		
	
	
		
			7.0 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/* Copyright (C) 2014 InfiniDB, Inc.
 | 
						|
   Copyright (C) 2016 MariaDB Corporation
 | 
						|
 | 
						|
   This program is free software; you can redistribute it and/or
 | 
						|
   modify it under the terms of the GNU General Public License
 | 
						|
   as published by the Free Software Foundation; version 2 of
 | 
						|
   the License.
 | 
						|
 | 
						|
   This program is distributed in the hope that it will be useful,
 | 
						|
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
   GNU General Public License for more details.
 | 
						|
 | 
						|
   You should have received a copy of the GNU General Public License
 | 
						|
   along with this program; if not, write to the Free Software
 | 
						|
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 | 
						|
   MA 02110-1301, USA. */
 | 
						|
 | 
						|
/***********************************************************************
 | 
						|
 *   $Id: ha_view.cpp 9642 2013-06-24 14:57:42Z rdempsey $
 | 
						|
 *
 | 
						|
 *
 | 
						|
 ***********************************************************************/
 | 
						|
 | 
						|
#define PREFER_MY_CONFIG_H
 | 
						|
#include <my_config.h>
 | 
						|
#include "idb_mysql.h"
 | 
						|
 | 
						|
#include <string>
 | 
						|
using namespace std;
 | 
						|
 | 
						|
#include <boost/algorithm/string/case_conv.hpp>
 | 
						|
using namespace boost;
 | 
						|
 | 
						|
#include "errorids.h"
 | 
						|
using namespace logging;
 | 
						|
 | 
						|
#include "parsetree.h"
 | 
						|
#include "simplefilter.h"
 | 
						|
#include "calpontsystemcatalog.h"
 | 
						|
using namespace execplan;
 | 
						|
 | 
						|
#include "ha_subquery.h"
 | 
						|
#include "ha_view.h"
 | 
						|
#include "ha_mcs_common.h"
 | 
						|
 | 
						|
namespace cal_impl_if
 | 
						|
{
 | 
						|
extern uint32_t buildJoin(gp_walk_info& gwi, List<TABLE_LIST>& join_list,
 | 
						|
                          std::stack<execplan::ParseTree*>& outerJoinStack);
 | 
						|
extern string getViewName(TABLE_LIST* table_ptr);
 | 
						|
 | 
						|
CalpontSystemCatalog::TableAliasName& View::viewName()
 | 
						|
{
 | 
						|
  return fViewName;
 | 
						|
}
 | 
						|
 | 
						|
void View::viewName(execplan::CalpontSystemCatalog::TableAliasName& viewName)
 | 
						|
{
 | 
						|
  fViewName = viewName;
 | 
						|
}
 | 
						|
 | 
						|
void View::transform()
 | 
						|
{
 | 
						|
  CalpontSelectExecutionPlan* csep = new CalpontSelectExecutionPlan();
 | 
						|
  csep->sessionID(fParentGwip->sessionid);
 | 
						|
 | 
						|
  // gwi for the sub query
 | 
						|
  gp_walk_info gwi(fParentGwip->timeZone, fParentGwip->subQueriesChain);
 | 
						|
  gwi.thd = fParentGwip->thd;
 | 
						|
 | 
						|
  uint32_t sessionID = csep->sessionID();
 | 
						|
  gwi.sessionid = sessionID;
 | 
						|
  boost::shared_ptr<CalpontSystemCatalog> csc = CalpontSystemCatalog::makeCalpontSystemCatalog(sessionID);
 | 
						|
  csc->identity(CalpontSystemCatalog::FE);
 | 
						|
  gwi.csc = csc;
 | 
						|
 | 
						|
  // traverse the table list of the view
 | 
						|
  TABLE_LIST* table_ptr = fSelect.get_table_list();
 | 
						|
  CalpontSelectExecutionPlan::SelectList derivedTbList;
 | 
						|
 | 
						|
  // @bug 1796. Remember table order on the FROM list.
 | 
						|
  gwi.clauseType = FROM;
 | 
						|
 | 
						|
  try
 | 
						|
  {
 | 
						|
    for (; table_ptr; table_ptr = table_ptr->next_local)
 | 
						|
    {
 | 
						|
      // mysql put vtable here for from sub. we ignore it
 | 
						|
      if (string(table_ptr->table_name.str).find("$vtable") != string::npos)
 | 
						|
        continue;
 | 
						|
 | 
						|
      string viewName = getViewName(table_ptr);
 | 
						|
      if (lower_case_table_names)
 | 
						|
      {
 | 
						|
        boost::algorithm::to_lower(viewName);
 | 
						|
      }
 | 
						|
 | 
						|
      if (table_ptr->derived)
 | 
						|
      {
 | 
						|
        SELECT_LEX* select_cursor = table_ptr->derived->first_select();
 | 
						|
        FromSubQuery* fromSub = new FromSubQuery(gwi, select_cursor);
 | 
						|
        string alias(table_ptr->alias.str);
 | 
						|
        if (lower_case_table_names)
 | 
						|
        {
 | 
						|
          boost::algorithm::to_lower(alias);
 | 
						|
        }
 | 
						|
        gwi.viewName =
 | 
						|
            make_aliasview("", alias, table_ptr->belong_to_view->alias.str, "", true, lower_case_table_names);
 | 
						|
        fromSub->alias(alias);
 | 
						|
        gwi.derivedTbList.push_back(SCSEP(fromSub->transform()));
 | 
						|
        // set alias to both table name and alias name of the derived table
 | 
						|
        CalpontSystemCatalog::TableAliasName tn = make_aliasview("", "", alias, viewName);
 | 
						|
        gwi.tbList.push_back(tn);
 | 
						|
        gwi.tableMap[tn] = make_pair(0, table_ptr);
 | 
						|
        // TODO MCOL-2178 isUnion member only assigned, never used
 | 
						|
        // MIGR::infinidb_vtable.isUnion = true; //by-pass the 2nd pass of rnd_init
 | 
						|
      }
 | 
						|
      else if (table_ptr->view)
 | 
						|
      {
 | 
						|
        // for nested view, the view name is vout.vin... format
 | 
						|
        CalpontSystemCatalog::TableAliasName tn =
 | 
						|
            make_aliasview(table_ptr->db.str, table_ptr->table_name.str, table_ptr->alias.str, viewName, true,
 | 
						|
                           lower_case_table_names);
 | 
						|
        gwi.viewName = make_aliastable(table_ptr->db.str, table_ptr->table_name.str, viewName, true,
 | 
						|
                                       lower_case_table_names);
 | 
						|
        View* view = new View(*table_ptr->view->first_select_lex(), &gwi);
 | 
						|
        view->viewName(gwi.viewName);
 | 
						|
        gwi.viewList.push_back(view);
 | 
						|
        view->transform();
 | 
						|
      }
 | 
						|
      else
 | 
						|
      {
 | 
						|
        // check foreign engine tables
 | 
						|
        bool columnStore = (table_ptr->table ? ha_mcs_common::isMCSTable(table_ptr->table) : true);
 | 
						|
 | 
						|
        // trigger system catalog cache
 | 
						|
        if (columnStore)
 | 
						|
          csc->columnRIDs(make_table(table_ptr->db.str, table_ptr->table_name.str, lower_case_table_names),
 | 
						|
                          true);
 | 
						|
 | 
						|
        CalpontSystemCatalog::TableAliasName tn =
 | 
						|
            make_aliasview(table_ptr->db.str, table_ptr->table_name.str, table_ptr->alias.str, viewName,
 | 
						|
                           columnStore, lower_case_table_names);
 | 
						|
        gwi.tbList.push_back(tn);
 | 
						|
        gwi.tableMap[tn] = make_pair(0, table_ptr);
 | 
						|
        fParentGwip->tableMap[tn] = make_pair(0, table_ptr);
 | 
						|
      }
 | 
						|
    }
 | 
						|
 | 
						|
    if (gwi.fatalParseError)
 | 
						|
    {
 | 
						|
      setError(gwi.thd, ER_INTERNAL_ERROR, gwi.parseErrorText);
 | 
						|
      delete csep;
 | 
						|
      return;
 | 
						|
    }
 | 
						|
  }
 | 
						|
  catch (IDBExcept& ie)
 | 
						|
  {
 | 
						|
    setError(gwi.thd, ER_INTERNAL_ERROR, ie.what());
 | 
						|
    CalpontSystemCatalog::removeCalpontSystemCatalog(sessionID);
 | 
						|
    delete csep;
 | 
						|
    return;
 | 
						|
  }
 | 
						|
  catch (...)
 | 
						|
  {
 | 
						|
    string emsg = IDBErrorInfo::instance()->errorMsg(ERR_LOST_CONN_EXEMGR);
 | 
						|
    setError(gwi.thd, ER_INTERNAL_ERROR, emsg);
 | 
						|
    CalpontSystemCatalog::removeCalpontSystemCatalog(sessionID);
 | 
						|
    delete csep;
 | 
						|
    return;
 | 
						|
  }
 | 
						|
 | 
						|
  // merge table list to parent select
 | 
						|
  fParentGwip->tbList.insert(fParentGwip->tbList.end(), gwi.tbList.begin(), gwi.tbList.end());
 | 
						|
  fParentGwip->derivedTbList.insert(fParentGwip->derivedTbList.end(), gwi.derivedTbList.begin(),
 | 
						|
                                    gwi.derivedTbList.end());
 | 
						|
  fParentGwip->correlatedTbNameVec.insert(fParentGwip->correlatedTbNameVec.end(),
 | 
						|
                                          gwi.correlatedTbNameVec.begin(), gwi.correlatedTbNameVec.end());
 | 
						|
 | 
						|
  // merge view list to parent
 | 
						|
  fParentGwip->viewList.insert(fParentGwip->viewList.end(), gwi.viewList.begin(), gwi.viewList.end());
 | 
						|
  gwi.viewList.clear();
 | 
						|
 | 
						|
  // merge non-collapsed outer join to parent select
 | 
						|
  stack<ParseTree*> tmpstack;
 | 
						|
 | 
						|
  while (!gwi.ptWorkStack.empty())
 | 
						|
  {
 | 
						|
    tmpstack.push(gwi.ptWorkStack.top());
 | 
						|
    gwi.ptWorkStack.pop();
 | 
						|
  }
 | 
						|
 | 
						|
  while (!tmpstack.empty())
 | 
						|
  {
 | 
						|
    fParentGwip->ptWorkStack.push(tmpstack.top());
 | 
						|
    tmpstack.pop();
 | 
						|
  }
 | 
						|
  while (!gwi.rcWorkStack.empty()) {
 | 
						|
    delete gwi.rcWorkStack.top();
 | 
						|
    gwi.rcWorkStack.pop();
 | 
						|
  }
 | 
						|
 | 
						|
  delete csep;
 | 
						|
}
 | 
						|
 | 
						|
uint32_t View::processJoin(gp_walk_info& gwi, std::stack<execplan::ParseTree*>& outerJoinStack)
 | 
						|
{
 | 
						|
  return buildJoin(gwi, fSelect.top_join_list, outerJoinStack);
 | 
						|
}
 | 
						|
 | 
						|
}  // namespace cal_impl_if
 |