CorrectedETS.cxx

Go to the documentation of this file.

#include <iostream>
#include <string>
#include <cstdlib>
#include <time.h>
#include <stdio.h>
#include <string.h>
#include <mysql/mysql.h>
#include "TFile.h"
#include "TTree.h"
#include "TROOT.h"
#include "TH1F.h"
#include "TAxis.h"
#include "TH1.h"
#include "TBufferFile.h"
#include <fstream>
#define CALCONSTSQLHELPER
 
using namespace std;
 
int main(int argc, char** argv) {
  if(argc!=3) { 
    printf("error, wrong argument number!\n");
    return 0; 
  }
  char *opt_id=argv[1];
  TFile *f= new TFile(argv[2]);
  cout<<"start "<<endl;
  static char query[80240000];
  char *p;
  int status=-1;
  std::cout << "-3-3-3" << std::endl;
  if(f->IsZombie()) return 0;
  std::cout << "-2-3-2" << std::endl;
  MYSQL conn;
  const char host[] = "bes3db1.ihep.ac.cn";
  const char user[] = "ets";
  const char passwd[] = "ets3400";
  const char db[] = "offlinedb";
  char opt_table_name[16] = "CorrectedETS";
  unsigned int port = 3306;
  const char *unix_socket = NULL;
  unsigned long client_flag = 0;
  std::cout << "-1-1-1" << std::endl;
  if (mysql_init(&conn) == NULL) {
      std::cout << "init db error" << std::endl;
    }
    if (!mysql_real_connect(&conn, host, user, passwd, db, port, unix_socket, client_flag))
    {
      std::cout << "db link error" << std::endl;
    }
  std::cout << "0000" << std::endl;
  gROOT->cd();
  std::cout << "1111" << std::endl;
  TTree *event =  (TTree*)f-> Get("event");
  TTree *event_copy =  event->CloneTree();
  TBufferFile event_buf(TBufferFile::kWrite);
  event_copy->Streamer(event_buf);
  cout<<"hahaha"<<endl;
  //p = strmov(query,"UPDATE %s SET CorrectedETS='",opt_table_name);
  sprintf(query,"UPDATE %s SET FinalETS='",opt_table_name);
  cout<<query<<endl;
  p= query + strlen(query);
  cout<<"111"<<p<<endl;
  p +=mysql_escape_string(p,event_buf.Buffer(),event_buf.BufferSize());
  cout<<"222"<<p<<endl;
  strcat(p,"' WHERE SerNo = ");
  strcat(p,opt_id);
  cout<<p<<endl;
  if (mysql_query(&conn,query))
  {
    cout<< "Failed to update SerNo:"<<opt_id<<endl;
   }
  //status = mysql_query(conn,p);
  printf("status=%d\n",status);
  int kkkk=mysql_errno(&conn);
  printf("err=%d\n",kkkk);
   
  mysql_close(&conn);
  f->Close();
  return 0;
}