Linux-（C）利用Mysql相关API实现类似mysql的操作

程序比较简单，不过也有趣，练习C mysql相关API。

程序大概实现流程：

1、通过主函数传递参数登陆数据库： 程序名  -u  数据库用户名  -p;

./mysql1 -u dbuser1 -p

2、初始化连接数据库。

3、处理输入sql语句。程序分显示跟不显示两路处理。（显示的意思是执行了sql命令之后时候会输出控制台，eg，show、desc、select等命令执行完后会打印信息到控制台  

insert、delete、update则不会）

4、循环输入执行sql语句，直到程序退出。

缺点：

1、 必须事先指定一个要登录的用户登录的数据库。正常的mysql是可以登录用户之后再选择相应的数据库。

2、退格键、左右上下移动不自由，都需要手动设置。（这个程序只实现了退格键）

mysql.c

/*
*/
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<errno.h>
#include<termios.h>
#include<mysql/mysql.h>

MYSQL *connection = NULL;
MYSQL mysql;

char host[20]="localhost";
char dbtable[20]="db1";


void init_db()
{
	mysql_init(&mysql);
}
void disconnect()
{
	if(connection!=NULL)
	{
		mysql_close(&mysql);
		connection=NULL;
	}
}
void showsql(char *sql)
{
	if(mysql_query(connection,sql)!=0)
	{
		printf("mysql query error: %s\n",mysql_error(connection));
		return ;
	}

	MYSQL_RES *res=mysql_store_result(connection);
	if(res==NULL)
	{
		printf("mysql store result error: %s",mysql_error(connection));
		return ;
	}
	/*遍历一张表信息，先统计列数后遍历每一行*/
	int fieldcnt=0;
	MYSQL_FIELD *sqlField=NULL;
	while(1) //统计field(段)
	{
		//mysql_fetch_field() is reset to return information
		//每返回一个语句，下一条语句将会重置，换行吧
		sqlField=mysql_fetch_field(res);
		if(sqlField==NULL)
			break;
		printf("%s\t",sqlField->name);
		fieldcnt++;
	}
	printf("\n");
	MYSQL_ROW sqlRow=NULL;
	while(1)
	{
		sqlRow=mysql_fetch_row(res);
		if(sqlRow==NULL)
			break;
		int i=0;
		for(i=0;i<fieldcnt;++i)
		{
			if(sqlRow[i] == NULL)
				printf("NULL\t");
			else
			    printf("%s\t", (const char *) sqlRow[i]);		//屏幕打印为字段内容
		}
		printf("\n");
	}

	printf("query is ok, %u rows affected\n", (unsigned int)mysql_affected_rows(connection));
    mysql_free_result(res);
	printf("show success\n");
}
void execsql(char *sql)
{
	printf("%s\n",sql);
	if(mysql_query(connection,sql)!=0)
	{
		printf("mysql query error: %s\n",mysql_error(connection));
		return ;
	}
	printf("query is ok, %u rows affected\n", (unsigned int)mysql_affected_rows(connection));

}

void dealsql(char *sql)
{
	if( strncmp(sql,"select",6)==0||
		strncmp(sql,"SELECT",6)==0||
		strncmp(sql,"show",4)==0||
		strncmp(sql,"SHOW",4)==0||
		strncmp(sql,"desc",4)==0||
		strncmp(sql,"DESC",4)==0
			)
	{
		showsql(sql); // show command
	}
	else
	{
		execsql(sql); //execute command
	}
}
void work(char *dbuser,const char * dbuserpawd)
{
	init_db();
	//需要 先绑定一个数据表名 dbtable
	connection=mysql_real_connect(&mysql,
			host,dbuser,dbuserpawd,dbtable,0,NULL,0);
	if(connection==NULL)
	{
		fprintf(stderr, "Failed to connect to database: Error: %s\n",
		          mysql_error(&mysql));
	}
	printf("connect success db1(table)\n");
	while(1)
	{
		char sql[1024];
		memset(sql,0,sizeof(sql));
		write(STDOUT_FILENO,"mysql1>",strlen("mysql1>"));
		read(STDIN_FILENO,sql,sizeof(sql));
		if (strncmp(sql, "quit", 4) == 0)
			break;
		dealsql(sql);
	}
	disconnect();
}

struct termios tmp, old;
//设置退格键“backspace”为退格键。
void setstty()
{
    //得到系统termion的状态
    if(tcgetattr(STDIN_FILENO, &tmp) == -1)
    {
        printf("tcgetattr error： %s\n", strerror(errno));
        return ;
    }
    old = tmp;
    //'\b'为退格键的ASCII码 因此下面这个语句表示退格键被修改为erase功能,即能擦除
    tmp.c_cc[VERASE] = '\b';

    //设置系统termion的状态，TCSANOW表示设置完成效果立即生效
    if(tcsetattr(STDIN_FILENO, TCSANOW, &tmp) == -1)
    {
        printf("tcsetattr error： %s\n", strerror(errno));
        return ;
    }
}
//恢复之前系统默认的状态，回显^H
void renew()
{
    if(tcsetattr(STDIN_FILENO, TCSANOW, &old) == -1)
    {
        printf("tcsetattr error： %s\n", strerror(errno));
        return ;
    }
}
int main(int argc,char * argv[])
{
	if(argc<4)
	{
		printf("argument: Program -u dbuername -p");
		return EXIT_SUCCESS;
	}
	if(strcmp(argv[1],"-u")!=0)
	{
		printf("the second argument is wrong,should be assigned -u  \n");
		return EXIT_SUCCESS;
	}
	if(strcmp(argv[3],"-p")!=0)
	{
		printf("the fouth argument is wrong,should be assign -p  \n");
		return EXIT_SUCCESS;
	}
	//outputs the string prompt,
	//turns off echoing, reads one line (the "password"),
	//restores the terminal state
	//The function getpass() returns a pointer to a static buffer
	const char *dbpassword = getpass("Enter password:");

	//write(STDOUT_FILENO,dbpassword,strlen(dbpassword));
	setstty();
	work(argv[2],dbpassword);
	renew();
	return 0;
}

执行效果：

stu@ubuntu:~/test1$ make
gcc -Wall -g -o mysql1.o -c mysql1.c
gcc -o mysql1 mysql1.o -lmysqlclient 
----------------ok------------
stu@ubuntu:~/test1$ ./mysql1 -u dbuser1 -p
Enter password:
connect success db1(table)
mysql1>show tables;
Tables_in_db1
table1
table3
query is ok, 2 rows affected
show success
mysql1>select * from table1;
name    sex     age     class
name2   nan     20      b
name1   boy     10      java
name1   boy     10      java
name3   boy     10      java
name4   boy     10      java
query is ok, 5 rows affected
show success
mysql1>delete from table1 where name = 'name4';
delete from table1 where name = 'name4';

query is ok, 1 rows affected
mysql1>select * from table1;
name    sex     age     class
name2   nan     20      b
name1   boy     10      java
name1   boy     10      java
name3   boy     10      java
query is ok, 4 rows affected
show success
mysql1>

退格键回显具体可以参考：http://blog.youkuaiyun.com/qq_33850438/article/details/52184786

另外一个mysql 小程序：http://blog.youkuaiyun.com/qq_33850438/article/details/52156224