DB Testing via ODBC

From PeformIQ Upgrade
Jump to navigation Jump to search

Using Windows ODBC with Python

Note this script explicitly references the ActiveState binary so as to utilize its in built ODBC module.

This script draws the required Manifest Ids from the Data Server (AKA dserver) which is described elsewhere.

Getting ODBC running with Python under Cygwin had too many external dependencies to make it a simple job. It is still on the agenda but will have to wait for the moment.


#!/c/PROGRA~1/ActiveState/Python/python.exe
#
#-------------------------------------------------------------------------------

import os
import sys
import dbi
import odbc
import time
import getopt
import logging

from datetime import datetime

#-------------------------------------------------------------------------------

from RTE import dcl

#-------------------------------------------------------------------------------

__version__   = "1.0.0"
__id__        = "@(#)  skel.py  [%s]  05/03/2008"

verbose_flg   = False

debug_level   = 0


table_name    = "Manifests"
PORT          = 9579
MAX_REQUESTS  = 10

log           = None
handle        = 'HCODEVJ86_%02d'
connector_no  = 1

#===============================================================================

def INFO(msg):
   if log: log.info(' ' + msg)
   if verbose_flg: print "[test]  %s" % msg

#-------------------------------------------------------------------------------

def ERROR(msg):
   if log: log.error(msg)
   sys.stderr.write('[test]  %s\n' % msg)

#-------------------------------------------------------------------------------

def WARNING(msg):
   if log: log.warning('*****' + msg + '*****')
   if verbose_flg: print "[test]  %s" % msg

#===============================================================================

def the_time():
   t = float(ref_time(False)) * 0.001

   return t

#------------------------------------------------------------------------------

t_reference = None

def ref_time(flg):
   global t_reference

   t_now  = datetime.now()

   if (flg):
      t_reference   = t_now
      t             = 0
   else:
      t_delta       = t_now - t_reference
      t             = ((t_delta.seconds * 1000000) + t_delta.microseconds)/1000.0

   return t

#===============================================================================

def query(no_queries, connector_no):
   ds = dcl.dcl(port=PORT)

   if (ds == None):
      print("Connection to data server failed - is data server process running?\n")
      return 1

   type_ref  = ds.RegisterType(table_name)

   if debug_level > 0:print "Data type \"%s\" registered as %d" % (table_name,  type_ref)

   s = odbc.odbc(handle % connector_no)

   cur = s.cursor()

   cnt = 0

   while cnt < no_queries:
      cnt += 1

      sp  = ds.GetNext(type_ref)

      if sp != None:
         manifest = sp[0]
      else:
         print "Type %d exhausted" % (pid, type_ref)
         return

      # print manifest

      qry   = "exec dbp_get_Article_history_for_internet %s" % manifest

      # print qry

      ref_time(True)

      cur.execute(qry)

      t_query = the_time()

      # print cur.description

      # for tup in cur.description:
         # print tup[0]

      row_cnt = 0

      while 1:
         rec = cur.fetchmany(10)
         row_cnt +=  len(rec)
         if not rec: break
         #print rec

      msg = "%02d ManifestNo %s  [%.3f] sec  %3d rows returned" % (cnt, manifest, t_query, row_cnt)

      sys.stderr.write("%s\n" % msg)
      sys.stderr.flush()

      INFO(msg)

#-------------------------------------------------------------------------------

def init():
   global lf
   global log

   pid      = os.getpid()

   if debug_level > 0:  print "My PID is %d" % pid

   log  = logging.getLogger('test')
   hdlr = logging.FileHandler('log/hcodevj86_%02d.log' % connector_no)
   fmtr = logging.Formatter('%(asctime)s %(levelname)s %(message)s')

   hdlr.setFormatter(fmtr)
   log.addHandler(hdlr) 
   log.setLevel(logging.INFO)

   INFO("===== Started processing ==================================")

#===============================================================================

def main():
   global verbose_flg
   global debug_level
   global connector_no
   global MAX_REQUESTS

   try:
      opts, args = getopt.getopt(sys.argv[1:], "c:dD:M:vVw?")
   except getopt.error, msg:
      print __doc__
      return 1

   for o, a in opts:
      if o == '-?':
         print __doc__
         return 1
      elif o == '-c':
         connector_no   = int(a)
      elif o == '-d':
         debug_level   += 1
      elif o == '-D':
         debug_level    = int(a)
      elif o == '-M':
         MAX_REQUESTS   = int(a)
      elif o == '-v':
         verbose_flg    = True
      elif o == '-V':
         print "[xxxx]  Version: %s" % __version__
         return 1
      else:
         print __doc__
         return 1

   sys.stderr.write("[test]  Working directory is %s - Using ODBC connector %d\n" % (os.getcwd(), connector_no))
   sys.stderr.flush()

   if (debug_level > 0): print "Debugging level set to %d" % debug_level

   if args:
      for arg in args:
         print arg

   init()

   query(MAX_REQUESTS, connector_no)

   return 0

#-------------------------------------------------------------------------------

if __name__ == '__main__' or __name__ == sys.argv[0]:
   try:
      sys.exit(main())
   except KeyboardInterrupt, e:
      print "[xxxx]  Interrupted!"

#-------------------------------------------------------------------------------

"""
Revision History:

     Date     Who   Description
   --------   ---   ------------------------------------------------------------
   20031014   plh   Initial implementation

Problems to fix:

To Do:

Issues:

"""

LoadRunner Script

Here is the LoadRunner equivalent of the above python script without the 'bells and whistles' we would normally add to complete the script.

vuser_init.c

#include "vdf.h"
#include "print.inl"

vuser_init()
{
    lrd_init(&InitInfo, DBTypeVersion);

    lrd_open_context(&Ctx1, LRD_DBTYPE_ODBC, 0, 0, 0);
    lrd_alloc_connection(&Con1, LRD_DBTYPE_ODBC, Ctx1, 0 /*Unused*/, 0);
    lrd_db_option(Con1, OT_ODBC_LOGIN_TIMEOUT, (void FAR const *)30, 0);

    lrd_open_connection(&Con1, LRD_DBTYPE_ODBC, "", lr_decrypt("xxxxx"), "",
        lr_decrypt("xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
        "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
        "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
        "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
        "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
        "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"), Ctx1, 1, 0);

    return VUSER_OK;
  // vuser_init

Action.c

Action()
{
    lrd_open_cursor(&Csr12, Con1, 0);

    lrd_db_option(Csr12, OT_ODBC_CURSOR_FWD_ONLY, 0, 0);
    lrd_db_option(Csr12, OT_ODBC_CONCUR_READ_ONLY, 0, 0);
    lrd_db_option(Csr12, OT_ODBC_MAX_ROWS, (void FAR const *)100, 0);
    lrd_db_option(Csr12, OT_ODBC_CURSOR_CLOSE, 0, 0);

    lr_start_transaction("GetArticleHistory_stmt");

	lrd_stmt(Csr12, "exec dbp_get_Article_history_for_internet {ManifestId}",
        -1, 1 /*Direct exec*/, 0 /*None*/, 0);

    lr_end_transaction("GetArticleHistory_stmt", LR_AUTO);

    lr_start_transaction("GetArticleHistory_fetch");

    lrd_fetch(Csr12, -11, 1, 0, PrintRow12, 0);
    // GRID(12);

    lr_end_transaction("GetArticleHistory_fetch", LR_AUTO);

    lrd_result_set(Csr12, 0, 0, 0);
    lrd_close_cursor(&Csr12, 0);

    return VUSER_OK;
}

vuser_end.c

vuser_end()
{
    lrd_close_connection(&Con1, 0, 0);

    lrd_free_connection(&Con1, 0 /*Unused*/, 0);

    lrd_close_context(&Ctx1, 0, 0);

	lrd_end(0);

    return VUSER_OK;
}  // vuser_end