@@ -12,12 +12,10 @@ int sql_findexist(std::string receiver){
12
12
sql::PreparedStatement * readdatament = con->prepareStatement (readdata);
13
13
sql::ResultSet *resultSet =readdatament->executeQuery ();
14
14
std::string usernamelist;
15
- if (resultSet->next ()){
16
- usernamelist = resultSet->getString (" username" );
15
+ while (resultSet->next ()){
16
+ if (receiver == resultSet->getString (" username" )) return 1 ;
17
17
}
18
- int pos = usernamelist.find (receiver);
19
- if (pos!=std::string::npos)return 1 ;
20
- else return 0 ;
18
+ return 0 ;
21
19
}
22
20
int lockcheck (std::string filename){
23
21
sql::mysql::MySQL_Driver *driver = sql::mysql::get_mysql_driver_instance ();
@@ -26,19 +24,15 @@ int lockcheck(std::string filename){
26
24
std::string readdata =" SELECT filename FROM file" ;
27
25
sql::PreparedStatement *readdatament =con->prepareStatement (readdata);
28
26
sql::ResultSet *resultSet =readdatament->executeQuery ();
29
- std::string filenamelist ;
30
- if (resultSet->next ()){
31
- filenamelist = resultSet->getString (" filename" );
27
+ std::string name ;
28
+ while (resultSet->next ()){
29
+ if ( resultSet->getString (" filename" )==filename) return 1 ;
32
30
}
33
- int pos =filenamelist.find (filename);
34
- if (pos!=std::string::npos)return 1 ;
35
- else {
36
31
std::string changestate =" INSERT INTO file(filename) VALUES (?)" ;
37
32
sql::PreparedStatement *changestatement =con->prepareStatement (changestate);
38
33
changestatement->setString (1 ,filename);
39
34
changestatement->executeUpdate ();
40
35
return 0 ;
41
- }
42
36
}
43
37
void process (sql::PreparedStatement *readdatament, std::vector<std::string> s, sql::Connection *con)
44
38
{
0 commit comments