Skip to content

Commit 7a0614c

Browse files
committed
change
1 parent 5cad800 commit 7a0614c

File tree

1 file changed

+6
-12
lines changed

1 file changed

+6
-12
lines changed

mysql.cc

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,10 @@ int sql_findexist(std::string receiver){
1212
sql::PreparedStatement * readdatament = con->prepareStatement(readdata);
1313
sql::ResultSet *resultSet =readdatament->executeQuery();
1414
std::string usernamelist;
15-
if(resultSet->next()){
16-
usernamelist = resultSet->getString("username");
15+
while(resultSet->next()){
16+
if(receiver == resultSet->getString("username"))return 1;
1717
}
18-
int pos = usernamelist.find(receiver);
19-
if(pos!=std::string::npos)return 1;
20-
else return 0;
18+
return 0;
2119
}
2220
int lockcheck(std::string filename){
2321
sql::mysql::MySQL_Driver *driver = sql::mysql::get_mysql_driver_instance();
@@ -26,19 +24,15 @@ int lockcheck(std::string filename){
2624
std::string readdata ="SELECT filename FROM file";
2725
sql::PreparedStatement *readdatament =con->prepareStatement(readdata);
2826
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;
3230
}
33-
int pos =filenamelist.find(filename);
34-
if(pos!=std::string::npos)return 1;
35-
else {
3631
std::string changestate ="INSERT INTO file(filename) VALUES (?)";
3732
sql::PreparedStatement *changestatement =con->prepareStatement(changestate);
3833
changestatement->setString(1,filename);
3934
changestatement->executeUpdate();
4035
return 0;
41-
}
4236
}
4337
void process(sql::PreparedStatement *readdatament, std::vector<std::string> s, sql::Connection *con)
4438
{

0 commit comments

Comments
 (0)