Shell脚本调用C语言之字符串切分函数——strtok

今天上午在写一个需求,要求的比较急,要求当天完成,我大致分析了一下,可以采用从shell脚本中插入一连串的日期,通过调用proc生成的可执行文件,将日期传入后台数据库,在数据库中进行计算。需要切分日期的字符串,转化成整数,插入int 数组中,手工实现太慢,就直接借用系统的strtok函数来用了。

场景模拟:

1. shell脚本:

#diao.sh 
#
!/bin/bash
date1
="20170622,20170623,20170626,20170627,20170628,20170629,20170627"
date2
="20170628,20170629,20170630"

if [ $1 -eq 0 ]
       
then
                compute $date1
else
                compute $date2
~                             

2. 后台proc代码,这里用C代码来模拟,重点讲述用strtok函数实现字符串的切分。

#include<string.h>
#include
<stdlib.h>
#include
<stdio.h> int main(int argv ,char * argc[])
{
 
char buf[100];
 
char * p = NULL;
 
char buf2[100][9];
 
int  data[100];
 
int len  = 0;
 
int i = 0;
  memset(buf,
0x00,sizeof(buf));
  memset(buf2,
0x00,sizeof(buf2)); memset(data,0x00,sizeof(data));
  memcpy(buf,argc[
1],strlen(argc[1]));

  printf(
"buf=%s\n",buf);
/* 下面代码按照","切分字符串,然后转化成整数,存入整数数组中*/
  p
= strtok(buf, ",");             
 
while( p!= NULL){ 
        strcpy(buf2[len],p);
        data[len]
= atoi(buf2[len]);       
        printf(
"buf2[%d]=%s\n",len,buf2[len]);
        len
++;
        p
= strtok(NULL, ",");          // 再次调用strtok函数 
    }
 
/* 上面的代码按照","切分字符串,然后转化成整数,存入整数数组中*/



for ( i = 0 ; i < len ; ++i){
      printf (
"data[%d]=%d\n",i,data[i]);
  }

}

编译运行情况:

Shell脚本调用C语言之字符串切分函数——strtok

 思考:将上述代码中字符串切割,并转化为整数,存入整数数组部分做成一个独立的函数,进行调用,通用性一下子就上来了。

3. 将切分过程做成一个独立的函数

函数名称为:mystrtok,里面还是调用系统的strtok,如果直接用系统的strtok不做任何处理,是试用不了的,因为strtok出来的都是char*类型的。

#include<string.h>
#include<stdlib.h>
#include<stdio.h>
int mystrtok(char * str,const char * delim, char  buf[][9],int * len, int data[])
{
  char * p = NULL;
  int i = 0;
   p = strtok(str, delim);              
  while( p!= NULL){  
        strcpy(buf[i],p);
        data[i] = atoi(buf[i]);         
        i++;
        p = strtok(NULL, delim);          // 再次调用strtok函数  
    } 
    *len = i;
   return 0;
}
int main(int argv ,char * argc[])
{
  char buf[100];
  char * p = NULL;
  char buf2[100][9];
  int   data[100];
  int len  = 0;
  int i = 0;
  memset(buf,0x00,sizeof(buf));
  memset(buf2,0x00,sizeof(buf2));
  memset(data,0x00,sizeof(data));
  memcpy(buf,argc[1],strlen(argc[1]));
   
  printf("buf=%s\n",buf);  
 /* 下面代码按照","切分字符串,然后转化成整数,存入整数数组中*/
/*   p = strtok(buf, ",");              
  while( p!= NULL){  
        strcpy(buf2[len],p);
        data[len] = atoi(buf2[len]);         
        printf("buf2[%d]=%s\n",len,buf2[len]);
        len++;
        p = strtok(NULL, ",");          // 再次调用strtok函数  
    } */ 
  /* 上面的代码按照","切分字符串,然后转化成整数,存入整数数组中*/ 
  /* 思考,将上述代码写成一个独立的函数,进行调用*/
  mystrtok(buf,",",buf2,&len,data); 
  for ( i = 0 ; i < len ; ++i){
      printf ("data[%d]=%d\n",i,data[i]);
   }
  
   
}

运行新的代码:

Shell脚本调用C语言之字符串切分函数——strtok

上述函数可以在任何字符串切割的场景中用到,尤其是数字字符串按照某种方式切割时。

另外一个值得注意的地方就是:shell脚本调用C程��时,main函数的参数中接受到shell脚本的参数,然后进行处理。

特别是字符串类型 char * ,字符数组 char buf[][],字符数组指针 char *p[], const char * 这些类型一定要搞清楚,之间是否可以转,怎么转,互相之间如何赋值的,都要非常清楚。

相关推荐