/**
 * test ip2region searcher program
 *
<<<<<<< HEAD
 * @author	chenxin<chenxin619315@gmail.com>
 * @date	2015-10-30
=======
 * @author    chenxin<chenxin619315@gmail.com>
 * @date    2015-10-30
>>>>>>> 7e51a4909fdb01014f948c2dbc5cfb5fbed9ce56
*/

#include "ip2region.h"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/time.h>

#define __PRINT_ABOUT__ \
<<<<<<< HEAD
    println("+-------------------------------------+");	\
    println("| ip2region test program              |");	\
    println("| Author: chenxin619315@gmail.com.    |");	\
    println("| Type 'quit' to exit the program.    |");	\
=======
    println("+-------------------------------------+");    \
    println("| ip2region test program              |");    \
    println("| Author: chenxin619315@gmail.com.    |");    \
    println("| Type 'quit' to exit the program.    |");    \
>>>>>>> 7e51a4909fdb01014f948c2dbc5cfb5fbed9ce56
    println("+-------------------------------------+");

//read a line from a command line.
static char *getLine( FILE *fp, char *__dst ) 
{
<<<<<<< HEAD
	register int c;
	register char *cs;

	cs = __dst;
	while ( ( c = getc( fp ) ) != EOF ) {
		if ( c == '\n' ) break;
		*cs++ = c; 
	}
	*cs = '\0';

	return ( c == EOF && cs == __dst ) ? NULL : __dst;
=======
    register int c;
    register char *cs;

    cs = __dst;
    while ( ( c = getc( fp ) ) != EOF ) {
        if ( c == '\n' ) break;
        *cs++ = c; 
    }
    *cs = '\0';

    return ( c == EOF && cs == __dst ) ? NULL : __dst;
>>>>>>> 7e51a4909fdb01014f948c2dbc5cfb5fbed9ce56
}

static double getTime()
{
<<<<<<< HEAD
	struct timeval tv;
	struct timezone tz;
	gettimeofday(&tv, &tz);

	return (tv.tv_sec * 1000 + ((double)tv.tv_usec)/1000);
=======
    struct timeval tv;
    struct timezone tz;
    gettimeofday(&tv, &tz);

    return (tv.tv_sec * 1000 + ((double)tv.tv_usec)/1000);
>>>>>>> 7e51a4909fdb01014f948c2dbc5cfb5fbed9ce56
}

int main( int argc, char **argv )
{
<<<<<<< HEAD
	ip2region_entry ip2rEntry;
	datablock_entry datablock;
	char *dbFile = NULL, *algorithm = NULL;
	char line[256];
	uint_t (*func_ptr)(ip2region_t, char *, datablock_t);
	double s_time, c_time;
	memset(&datablock, 0x00, sizeof(datablock_entry));

	if ( argc < 2  ) {
		printf("Usage: a.out [ip2region db file path] [algorithm]");
		return 0;
	}

	dbFile	  = argv[1];
	algorithm = "B-tree";
	func_ptr  = ip2region_btree_search_string;
	if ( argc >= 3 && strcmp(argv[2], "binary") == 0 ) {
		algorithm = "Binary";
		func_ptr = ip2region_binary_search_string;
	}

	//create a new ip2rObj
	printf("+--initializing %s ... \n", algorithm);
	if ( ip2region_create(&ip2rEntry, dbFile) == 0 )
	{
		println("Error: Fail to create the ip2region object");
		return 0;
	}

	__PRINT_ABOUT__;

	while ( 1 )
	{
		print("ip2region>> ");
		getLine( stdin, line );
		if ( strlen(line) < 2 ) continue;
		if ( strcasecmp( line, "quit" ) == 0 ) {
			println("+--Bye!");
			break;
		}

		s_time = getTime();
		func_ptr(&ip2rEntry, line, &datablock);
		c_time = getTime() - s_time;
		printf("%d|%s in %.5f millseconds\n", datablock.city_id, datablock.region, c_time);
	}

	//destory the ip2rObj
	ip2region_destroy(&ip2rEntry);

	return 0;
=======
    ip2region_entry ip2rEntry;
    datablock_entry datablock;
    char *dbFile = NULL, *algorithm = NULL;
    char line[256];
    uint_t (*func_ptr)(ip2region_t, char *, datablock_t);
    double s_time, c_time;
    memset(&datablock, 0x00, sizeof(datablock_entry));

    if ( argc < 2  ) {
        printf("Usage: a.out [ip2region db file path] [algorithm]");
        return 0;
    }

    dbFile      = argv[1];
    algorithm = "B-tree";
    func_ptr  = ip2region_btree_search_string;
    if ( argc >= 3 && strcmp(argv[2], "binary") == 0 ) {
        algorithm = "Binary";
        func_ptr = ip2region_binary_search_string;
    }

    //create a new ip2rObj
    printf("+--initializing %s ... \n", algorithm);
    if ( ip2region_create(&ip2rEntry, dbFile) == 0 )
    {
        println("Error: Fail to create the ip2region object");
        return 0;
    }

    __PRINT_ABOUT__;

    while ( 1 )
    {
        print("ip2region>> ");
        getLine( stdin, line );
        if ( strlen(line) < 2 ) continue;
        if ( strcasecmp( line, "quit" ) == 0 ) {
            println("+--Bye!");
            break;
        }

        s_time = getTime();
        func_ptr(&ip2rEntry, line, &datablock);
        c_time = getTime() - s_time;
        printf("%d|%s in %.5f millseconds\n", datablock.city_id, datablock.region, c_time);
    }

    //destory the ip2rObj
    ip2region_destroy(&ip2rEntry);

    return 0;
>>>>>>> 7e51a4909fdb01014f948c2dbc5cfb5fbed9ce56
}