[u/mrichter/AliRoot.git] / SHUTTLE / test / TestShuttleLogbook.C


TSQLServer *fServer=0;

//______________________________________________________________________________________________
Bool_t Connect(){
// Connect to MySQL Server of the DAQ logbook

	// check connection: if already connected return
	if(fServer && fServer->IsConnected()) return kTRUE;

//	TString serverHost="mysql://pcald30.cern.ch";
	TString serverHost="mysql://aldaqgw01-gpn.cern.ch:1434";
	fServer = TSQLServer::Connect(serverHost.Data(),"shuttle","alice");
	
	printf("Connecting to mysql server on %s !!!!!!\n", serverHost.Data());

	if (!fServer || !fServer->IsConnected()) {
		printf("Can't establish connection to DAQ log book DB!\n");
		if(fServer) delete fServer;
		return kFALSE;
	}

	// Get table
	TSQLResult* aResult=0;
	aResult = fServer->GetTables("SHUTTLE_TEST");
	delete aResult;
	return kTRUE;
}

//______________________________________________________________________________________________
Bool_t QueryShuttleLogbook(AliShuttleLogbookEntry& entry, Int_t runNumber=-1)
{
// Query DAQ's Shuttle logbook and fills detector status array

	if(runNumber<=0 && entry.GetRun()<=0) {
		printf("Use a valid Run number!\n");
		return kFALSE;
	}
	if(!Connect()) return kFALSE;
	if(runNumber<=0) runNumber= entry.GetRun();
	//entry.SetRun(runNumber); 

	// check connection, in case connect
	if(!Connect()) return kFALSE;

	TString sqlQuery;
	sqlQuery = Form("select * from logbook_shuttle where run = %d", runNumber);

	TSQLResult* aResult = fServer->Query(sqlQuery);
	if (!aResult) {
		printf("Can't execute query <%s>!\n", sqlQuery.Data());
		return kFALSE;
	}

	// TODO Check field count!
//	if (aResult->GetFieldCount() != 24) {
//		printf("Invalid SQL result field number!\n");
//		delete aResult;
//		return kFALSE;
//	}

	TSQLRow* aRow;
	while ((aRow = aResult->Next())) {
		TString runString(aRow->GetField(0), aRow->GetFieldLength(0));
		Int_t run = runString.Atoi();

		// loop on detectors
		for(UInt_t ii = 0; ii < 22; ii++){
			entry.SetDetectorStatus(aResult->GetFieldName(ii), aRow->GetField(ii));
		}

		delete aRow;
	}

	delete aResult; aResult=0;

	// Query run parameters from logbook!

	sqlQuery = Form("select * from logbook where run=%d", runNumber);

	aResult = fServer->Query(sqlQuery);
	if (!aResult) {
		printf("Can't execute query <%s>!", sqlQuery.Data());
		return kFALSE;
	}

	if(aResult->GetRowCount() == 0) {
		printf("QueryRunParameters - No entry in DAQ Logbook for run %d!", runNumber);
		delete aResult;
		return kFALSE;
	}

	if(aResult->GetRowCount() > 1) {
		printf("More than one entry in DAQ Logbook for run %d!", runNumber);
		delete aResult;
		return kFALSE;
	}

	while ((aRow = aResult->Next())) {

		for(UInt_t ii = 0; ii < aResult->GetFieldCount(); ii++)
			entry.SetRunParameter(aResult->GetFieldName(ii), aRow->GetField(ii));

		UInt_t startTime, endTime;
		TString startTimeString = entry.GetRunParameter("time_start");
		UInt_t startTime = startTimeString.Atoi();
		TString endTimeString = entry.GetRunParameter("time_end");
		UInt_t endTime = endTimeString.Atoi();

		if (!startTime || !endTime || startTime > endTime) {
			printf("QueryRunParameters - Invalid parameters for Run %d: startTime = %d, endTime = %d",
					runNumber, startTime, endTime);
			delete aRow;
			delete aResult;
			return kFALSE;
		}

//		entry.SetStartTime(startTime);
//		entry.SetEndTime(endTime);

		delete aRow;
	}

	entry.Print("all");

	delete aResult;
	return kTRUE;
}
//______________________________________________________________________________________________
Bool_t UpdateShuttleLogbook(AliShuttleLogbookEntry& entry, Int_t runNumber=-1)
{
  // Update Shuttle logbook table - TEST ONLY, USE WITH CARE!


	if(runNumber<=0 && entry.GetRun()<=0) {
		printf("Use a valid Run number!\n");
		return kFALSE;
	}
	if(!Connect()) return kFALSE;
	if(runNumber<=0) runNumber= entry.GetRun();
	entry.SetRun(runNumber); 

	TString sqlQuery("update logbook_shuttle set ");

	for(UInt_t i=0; i < AliShuttleInterface::NDetectors(); i++){
		sqlQuery += Form("%s=\"%s\"", AliShuttleInterface::GetDetName(i), entry.GetDetectorStatusName(entry.GetDetectorStatus(i)));
		if(i < AliShuttleInterface::NDetectors()-1) sqlQuery += ", ";
	}

	sqlQuery += Form(" where run=%d;",entry.GetRun());

	printf("sqlQuery: %s\n", sqlQuery.Data());

	TSQLResult* aResult;
	aResult = fServer->Query(sqlQuery);
	if (!aResult) {
		printf("Can't execute query <%s>!\n", sqlQuery.Data());
		return kFALSE;
	}

	delete aResult;

	return kTRUE;
}

//______________________________________________________________________________________________
Bool_t UpdateShuttleLogbook(Int_t runNumber, const char* detCode, AliShuttleLogbookEntry::Status status)
{
  // Update Shuttle logbook table - TEST ONLY, USE WITH CARE!


	if(AliShuttleInterface::GetDetPos(detCode) < 0) return kFALSE;
	if(!Connect()) return kFALSE;

	TString sqlQuery("update logbook_shuttle set ");


	sqlQuery += Form("%s=\"%s\" ", detCode, AliShuttleLogbookEntry::GetDetectorStatusName(status));

	sqlQuery += Form("where run=%d;",runNumber);

	printf("sqlQuery: %s\n", sqlQuery.Data());

	TSQLResult* aResult;
	aResult = fServer->Query(sqlQuery);
	if (!aResult) {
		printf("Can't execute query <%s>!\n", sqlQuery.Data());
		return kFALSE;
	}

	delete aResult;

	return kTRUE;
}

//______________________________________________________________________________________________
Bool_t UpdateShuttleLogbook(Int_t runNumber, const char* detCode, const char* status)
{
  // Update Shuttle logbook table - TEST ONLY, USE WITH CARE!


	if(AliShuttleInterface::GetDetPos(detCode) < 0) return kFALSE;
	if(!Connect()) return kFALSE;

	TString sqlQuery("update logbook_shuttle set ");


	sqlQuery += Form("%s=\"%s\" ", detCode, status);

	sqlQuery += Form("where run=%d;",runNumber);

	printf("sqlQuery: %s\n", sqlQuery.Data());

	TSQLResult* aResult;
	aResult = fServer->Query(sqlQuery);
	if (!aResult) {
		printf("Can't execute query <%s>!\n", sqlQuery.Data());
		return kFALSE;
	}

	delete aResult;

	return kTRUE;
}

//______________________________________________________________________________________________
Bool_t InsertNewRun(AliShuttleLogbookEntry& entry, Int_t runNumber=-1)
{
  // Update Shuttle logbook table - TEST ONLY, USE WITH CARE!

	if(runNumber<=0 && entry.GetRun()<=0) {
		printf("Use a valid Run number!\n");
		return kFALSE;
	}
	if(!Connect()) return kFALSE;
	if(runNumber<=0) runNumber= entry.GetRun();
	entry.SetRun(runNumber); 

	TString sqlQuery = Form("insert into logbook_shuttle (run) values (%d);", runNumber);

	printf("sqlQuery: %s\n", sqlQuery.Data());

	TSQLResult* aResult;
	aResult = fServer->Query(sqlQuery);
	if (!aResult) {
		printf("Can't execute query <%s>!\n", sqlQuery.Data());
		return kFALSE;
	}

	delete aResult;

	UpdateShuttleLogbook(entry);

	return kTRUE;
}

//______________________________________________________________________________________________
Bool_t DeleteEntries(Int_t runNumber=-1)
{
  // Update Shuttle logbook table - TEST ONLY, USE WITH CARE!

	if(!Connect()) return kFALSE;
	
	TString runStr;
	if(runNumber>0) runStr=Form("where run=%d",runNumber);
	TString sqlQuery = Form("delete from logbook_shuttle %s;", runStr.Data());

	printf("sqlQuery: %s\n", sqlQuery.Data());

	TSQLResult* aResult;
	aResult = fServer->Query(sqlQuery);
	if (!aResult) {
		printf("Can't execute query <%s>!\n", sqlQuery.Data());
		return kFALSE;
	}

	delete aResult;

	return kTRUE;
}

//______________________________________________________________________________________________
void TestShuttleLogbook(){

	gSystem->Load("libSHUTTLE.so");
	
	DeleteEntries();

	AliShuttleLogbookEntry::Status y[18]={1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1};
	AliShuttleLogbookEntry lb(21242, y);
	lb.SetDetectorStatus("HMP","Unprocessed"); // HMPID
	lb.SetDetectorStatus("ZDC","Unprocessed"); // ZDC
	lb.SetDetectorStatus("TPC","Unprocessed"); // TPC
	lb.SetDetectorStatus("SPD","Unprocessed"); // SPD
	lb.SetDetectorStatus("GRP","Unprocessed"); // GRP
	lb.Print();

	InsertNewRun(lb);
	InsertNewRun(lb,21243);
	InsertNewRun(lb,21244);
	InsertNewRun(lb,21245);
//	InsertNewRun(lb,21246);
//	InsertNewRun(lb,1000);

}
Commit	Line	Data
eba76848	1
	2	TSQLServer *fServer=0;
	3
	4	//______________________________________________________________________________________________
	5	Bool_t Connect(){
	6	// Connect to MySQL Server of the DAQ logbook
	7
	8	// check connection: if already connected return
	9	if(fServer && fServer->IsConnected()) return kTRUE;
	10
2c15234c	11	// TString serverHost="mysql://pcald30.cern.ch";
ea3b94b8	12	TString serverHost="mysql://aldaqgw01-gpn.cern.ch:1434";
ea3b94b8	13	fServer = TSQLServer::Connect(serverHost.Data(),"shuttle","alice");
2c15234c	14
2c15234c	15	printf("Connecting to mysql server on %s !!!!!!\n", serverHost.Data());
eba76848	16
	17	if (!fServer \|\| !fServer->IsConnected()) {
	18	printf("Can't establish connection to DAQ log book DB!\n");
	19	if(fServer) delete fServer;
	20	return kFALSE;
	21	}
	22
	23	// Get table
	24	TSQLResult* aResult=0;
ea3b94b8	25	aResult = fServer->GetTables("SHUTTLE_TEST");
eba76848	26	delete aResult;
	27	return kTRUE;
	28	}
	29
	30	//______________________________________________________________________________________________
	31	Bool_t QueryShuttleLogbook(AliShuttleLogbookEntry& entry, Int_t runNumber=-1)
	32	{
	33	// Query DAQ's Shuttle logbook and fills detector status array
	34
	35	if(runNumber<=0 && entry.GetRun()<=0) {
	36	printf("Use a valid Run number!\n");
	37	return kFALSE;
	38	}
	39	if(!Connect()) return kFALSE;
	40	if(runNumber<=0) runNumber= entry.GetRun();
ea3b94b8	41	//entry.SetRun(runNumber);
eba76848	42
	43	// check connection, in case connect
	44	if(!Connect()) return kFALSE;
	45
	46	TString sqlQuery;
	47	sqlQuery = Form("select * from logbook_shuttle where run = %d", runNumber);
	48
	49	TSQLResult* aResult = fServer->Query(sqlQuery);
	50	if (!aResult) {
	51	printf("Can't execute query <%s>!\n", sqlQuery.Data());
	52	return kFALSE;
	53	}
	54
	55	// TODO Check field count!
ea3b94b8	56	// if (aResult->GetFieldCount() != 24) {
	57	// printf("Invalid SQL result field number!\n");
	58	// delete aResult;
	59	// return kFALSE;
	60	// }
eba76848	61
	62	TSQLRow* aRow;
	63	while ((aRow = aResult->Next())) {
	64	TString runString(aRow->GetField(0), aRow->GetFieldLength(0));
	65	Int_t run = runString.Atoi();
	66
	67	// loop on detectors
ea3b94b8	68	for(UInt_t ii = 0; ii < 22; ii++){
eba76848	69	entry.SetDetectorStatus(aResult->GetFieldName(ii), aRow->GetField(ii));
	70	}
	71
	72	delete aRow;
	73	}
	74
	75	delete aResult; aResult=0;
	76
	77	// Query run parameters from logbook!
	78
	79	sqlQuery = Form("select * from logbook where run=%d", runNumber);
	80
	81	aResult = fServer->Query(sqlQuery);
	82	if (!aResult) {
	83	printf("Can't execute query <%s>!", sqlQuery.Data());
	84	return kFALSE;
	85	}
	86
	87	if(aResult->GetRowCount() == 0) {
	88	printf("QueryRunParameters - No entry in DAQ Logbook for run %d!", runNumber);
	89	delete aResult;
	90	return kFALSE;
	91	}
	92
	93	if(aResult->GetRowCount() > 1) {
	94	printf("More than one entry in DAQ Logbook for run %d!", runNumber);
	95	delete aResult;
	96	return kFALSE;
	97	}
	98
	99	while ((aRow = aResult->Next())) {
	100
	101	for(UInt_t ii = 0; ii < aResult->GetFieldCount(); ii++)
	102	entry.SetRunParameter(aResult->GetFieldName(ii), aRow->GetField(ii));
	103
	104	UInt_t startTime, endTime;
	105	TString startTimeString = entry.GetRunParameter("time_start");
	106	UInt_t startTime = startTimeString.Atoi();
	107	TString endTimeString = entry.GetRunParameter("time_end");
	108	UInt_t endTime = endTimeString.Atoi();
	109
	110	if (!startTime \|\| !endTime \|\| startTime > endTime) {
	111	printf("QueryRunParameters - Invalid parameters for Run %d: startTime = %d, endTime = %d",
	112	runNumber, startTime, endTime);
	113	delete aRow;
	114	delete aResult;
	115	return kFALSE;
	116	}
	117
ea3b94b8	118	// entry.SetStartTime(startTime);
ea3b94b8	119	// entry.SetEndTime(endTime);
eba76848	120
	121	delete aRow;
	122	}
	123
	124	entry.Print("all");
	125
	126	delete aResult;
	127	return kTRUE;
	128	}
	129	//______________________________________________________________________________________________
	130	Bool_t UpdateShuttleLogbook(AliShuttleLogbookEntry& entry, Int_t runNumber=-1)
	131	{
	132	// Update Shuttle logbook table - TEST ONLY, USE WITH CARE!
	133
	134
	135	if(runNumber<=0 && entry.GetRun()<=0) {
	136	printf("Use a valid Run number!\n");
	137	return kFALSE;
	138	}
	139	if(!Connect()) return kFALSE;
	140	if(runNumber<=0) runNumber= entry.GetRun();
	141	entry.SetRun(runNumber);
	142
	143	TString sqlQuery("update logbook_shuttle set ");
	144
	145	for(UInt_t i=0; i < AliShuttleInterface::NDetectors(); i++){
	146	sqlQuery += Form("%s=\"%s\"", AliShuttleInterface::GetDetName(i), entry.GetDetectorStatusName(entry.GetDetectorStatus(i)));
	147	if(i < AliShuttleInterface::NDetectors()-1) sqlQuery += ", ";
	148	}
	149
	150	sqlQuery += Form(" where run=%d;",entry.GetRun());
	151
	152	printf("sqlQuery: %s\n", sqlQuery.Data());
	153
	154	TSQLResult* aResult;
	155	aResult = fServer->Query(sqlQuery);
	156	if (!aResult) {
	157	printf("Can't execute query <%s>!\n", sqlQuery.Data());
	158	return kFALSE;
	159	}
	160
	161	delete aResult;
	162
	163	return kTRUE;
	164	}
	165
	166	//______________________________________________________________________________________________
	167	Bool_t UpdateShuttleLogbook(Int_t runNumber, const char* detCode, AliShuttleLogbookEntry::Status status)
	168	{
	169	// Update Shuttle logbook table - TEST ONLY, USE WITH CARE!
	170
	171
	172	if(AliShuttleInterface::GetDetPos(detCode) < 0) return kFALSE;
	173	if(!Connect()) return kFALSE;
	174
	175	TString sqlQuery("update logbook_shuttle set ");
	176
	177
	178	sqlQuery += Form("%s=\"%s\" ", detCode, AliShuttleLogbookEntry::GetDetectorStatusName(status));
	179
	180	sqlQuery += Form("where run=%d;",runNumber);
	181
	182	printf("sqlQuery: %s\n", sqlQuery.Data());
	183
184	TSQLResult* aResult;
185	aResult = fServer->Query(sqlQuery);
186	if (!aResult) {
187	printf("Can't execute query <%s>!\n", sqlQuery.Data());
188	return kFALSE;
189	}
190
191	delete aResult;
192
193	return kTRUE;
194	}
195
196	//______________________________________________________________________________________________
197	Bool_t UpdateShuttleLogbook(Int_t runNumber, const char* detCode, const char* status)
198	{
199	// Update Shuttle logbook table - TEST ONLY, USE WITH CARE!
200
201
202	if(AliShuttleInterface::GetDetPos(detCode) < 0) return kFALSE;
203	if(!Connect()) return kFALSE;
204
205	TString sqlQuery("update logbook_shuttle set ");
206
207
208	sqlQuery += Form("%s=\"%s\" ", detCode, status);
209
210	sqlQuery += Form("where run=%d;",runNumber);
211
212	printf("sqlQuery: %s\n", sqlQuery.Data());
213
214	TSQLResult* aResult;
215	aResult = fServer->Query(sqlQuery);
216	if (!aResult) {
217	printf("Can't execute query <%s>!\n", sqlQuery.Data());
218	return kFALSE;
219	}
220
221	delete aResult;
222
223	return kTRUE;
224	}
225
226	//______________________________________________________________________________________________
227	Bool_t InsertNewRun(AliShuttleLogbookEntry& entry, Int_t runNumber=-1)
228	{
229	// Update Shuttle logbook table - TEST ONLY, USE WITH CARE!
230
231	if(runNumber<=0 && entry.GetRun()<=0) {
232	printf("Use a valid Run number!\n");
233	return kFALSE;
234	}
235	if(!Connect()) return kFALSE;
236	if(runNumber<=0) runNumber= entry.GetRun();
237	entry.SetRun(runNumber);
238
239	TString sqlQuery = Form("insert into logbook_shuttle (run) values (%d);", runNumber);
240
241	printf("sqlQuery: %s\n", sqlQuery.Data());
242
243	TSQLResult* aResult;
244	aResult = fServer->Query(sqlQuery);
245	if (!aResult) {
246	printf("Can't execute query <%s>!\n", sqlQuery.Data());
247	return kFALSE;
248	}
249
250	delete aResult;
251
252	UpdateShuttleLogbook(entry);
253
254	return kTRUE;
255	}
256
257	//______________________________________________________________________________________________
258	Bool_t DeleteEntries(Int_t runNumber=-1)
259	{
260	// Update Shuttle logbook table - TEST ONLY, USE WITH CARE!
261
262	if(!Connect()) return kFALSE;
263
264	TString runStr;
265	if(runNumber>0) runStr=Form("where run=%d",runNumber);
266	TString sqlQuery = Form("delete from logbook_shuttle %s;", runStr.Data());
267
268	printf("sqlQuery: %s\n", sqlQuery.Data());
269
270	TSQLResult* aResult;
271	aResult = fServer->Query(sqlQuery);
272	if (!aResult) {
273	printf("Can't execute query <%s>!\n", sqlQuery.Data());
274	return kFALSE;
275	}
276
277	delete aResult;
278
279	return kTRUE;
280	}
281
282	//______________________________________________________________________________________________
283	void TestShuttleLogbook(){
284
285	gSystem->Load("libSHUTTLE.so");
be48e3ea	286
be48e3ea	287	DeleteEntries();
eba76848	288
2c15234c	289	AliShuttleLogbookEntry::Status y[18]={1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1};
be48e3ea	290	AliShuttleLogbookEntry lb(21242, y);
59ab3716	291	lb.SetDetectorStatus("HMP","Unprocessed"); // HMPID
	292	lb.SetDetectorStatus("ZDC","Unprocessed"); // ZDC
	293	lb.SetDetectorStatus("TPC","Unprocessed"); // TPC
	294	lb.SetDetectorStatus("SPD","Unprocessed"); // SPD
2c15234c	295	lb.SetDetectorStatus("GRP","Unprocessed"); // GRP
eba76848	296	lb.Print();
	297
	298	InsertNewRun(lb);
be48e3ea	299	InsertNewRun(lb,21243);
eba76848	300	InsertNewRun(lb,21244);
be48e3ea	301	InsertNewRun(lb,21245);
59ab3716	302	// InsertNewRun(lb,21246);
59ab3716	303	// InsertNewRun(lb,1000);
eba76848	304
eba76848	305	}