misc/downloadd/dblayer.c

/***************************************************************************
 *   Copyright (C) 2006 by Torben H. Nielsen                               *
 *   torben@t-hoerup.dk                                                    *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#include <mysql/mysql.h>
#include <string.h>
#include <stdio.h>

#include "daemon.h"
#include "dblayer.h"


/* use short-term connections:
 * -connect
 * -do the stuf
 * -disconnect
 * It may introduce some overhead, but it may be safer!
 */


/* return 1 if there was a download in queue */
int nextdownload(dlrecord *dl) {
        MYSQL connection;
        MYSQL_RES *result;
        MYSQL_ROW row;
        int retval,antal;
        char buf[150];
        int success=0;
        char *pos;

        mysql_init( &connection);
        
        if (!mysql_real_connect( &connection, CONFIG->dbhost, CONFIG->dbuser, CONFIG->dbpass, CONFIG->dbname, CONFIG->dbport, NULL, 0)) {
                sprintf(buf,"Could not connect to DB : %s\n", mysql_error(&connection));
                log_message(buf);
                daemon_shutdown(1);
        }
        
        
        retval = mysql_query( &connection, "SELECT id,url,outname FROM downloads WHERE status=0 ORDER BY submitted LIMIT 1");
        if (retval==0) {
                result = mysql_store_result( &connection );
                antal = mysql_num_rows( result );
                if (antal == 1) {
                        row = mysql_fetch_row(result);
                        dl->id = atoi( row[0] );
                        strcpy(dl->URL, row[1]);
                        
                        if ( strlen(row[2]) > 0) { // the user submitted a name for the output
                                strcpy(dl->filename, row[2]);
                        } else {
                                dl->filename[0] = 0;
                                pos = strrchr(row[1], '/');
                                pos++;
                                strcpy(dl->filename, pos);      
                        }
                        success = 1;
                }
                mysql_free_result(result);
        }
        mysql_close( &connection );
        return success;
}

int sql_worker(dlrecord *dl, char *sql) {
        MYSQL connection;
        mysql_init( &connection);
        int retval;
        char buf[120];
        
        if (!mysql_real_connect( &connection, CONFIG->dbhost, CONFIG->dbuser, CONFIG->dbpass, CONFIG->dbname, CONFIG->dbport, NULL, 0)) {
                sprintf(buf,"Could not connect to DB : %s\n", mysql_error(&connection));
                log_message(buf);
        }
        retval = mysql_query(&connection, sql);
        mysql_close(&connection);
        return retval;
}

int sql_update_bytecount(dlrecord *record, long long bytes) {
        char buf[120];
        sprintf(buf, "UPDATE downloads SET bytecount=%lld WHERE id=%d", bytes, record->id);
        return sql_worker(record, buf);
}

int sql_start_download(dlrecord *record) {
        char buf[120];
        sprintf(buf, "UPDATE downloads SET status=1, started=now() WHERE id=%d", record->id);
        return sql_worker(record, buf);
}

int sql_abort_download(dlrecord *record, char *message) {
        char buf[250];
        sprintf(buf, "UPDATE downloads SET status=-1, completed=now(),message='%s' WHERE id=%d",message, record->id);
        return sql_worker(record, buf);
}

int sql_stop_download(dlrecord *record) {
        char buf[120];
        sprintf(buf, "UPDATE downloads SET status=2, completed=now() WHERE id=%d", record->id);
        return sql_worker(record, buf);
}
1	torben	328	/***************************************************************************
2			* Copyright (C) 2006 by Torben H. Nielsen *
3			* torben@t-hoerup.dk *
4			* *
5			* This program is free software; you can redistribute it and/or modify *
6			* it under the terms of the GNU General Public License as published by *
7			* the Free Software Foundation; either version 2 of the License, or *
8			* (at your option) any later version. *
9			* *
10			* This program is distributed in the hope that it will be useful, *
11			* but WITHOUT ANY WARRANTY; without even the implied warranty of *
12			* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
13			* GNU General Public License for more details. *
14			* *
15			* You should have received a copy of the GNU General Public License *
16			* along with this program; if not, write to the *
17			* Free Software Foundation, Inc., *
18			* 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. *
19			***************************************************************************/
20
21			#include <mysql/mysql.h>
22			#include <string.h>
23			#include <stdio.h>
24
25			#include "daemon.h"
26			#include "dblayer.h"
27
28
29			/* use short-term connections:
30			* -connect
31			* -do the stuf
32			* -disconnect
33			* It may introduce some overhead, but it may be safer!
34			*/
35
36
37			/* return 1 if there was a download in queue */
38			int nextdownload(dlrecord *dl) {
39			MYSQL connection;
40			MYSQL_RES *result;
41			MYSQL_ROW row;
42			int retval,antal;
43			char buf[150];
44			int success=0;
45			char *pos;
46
47			mysql_init( &connection);
48
49			if (!mysql_real_connect( &connection, CONFIG->dbhost, CONFIG->dbuser, CONFIG->dbpass, CONFIG->dbname, CONFIG->dbport, NULL, 0)) {
50			sprintf(buf,"Could not connect to DB : %s\n", mysql_error(&connection));
51			log_message(buf);
52			daemon_shutdown(1);
53			}
54
55
56			retval = mysql_query( &connection, "SELECT id,url,outname FROM downloads WHERE status=0 ORDER BY submitted LIMIT 1");
57			if (retval==0) {
58			result = mysql_store_result( &connection );
59			antal = mysql_num_rows( result );
60			if (antal == 1) {
61			row = mysql_fetch_row(result);
62			dl->id = atoi( row[0] );
63			strcpy(dl->URL, row[1]);
64
65			if ( strlen(row[2]) > 0) { // the user submitted a name for the output
66			strcpy(dl->filename, row[2]);
67			} else {
68			dl->filename[0] = 0;
69			pos = strrchr(row[1], '/');
70			pos++;
71			strcpy(dl->filename, pos);
72			}
73			success = 1;
74			}
75			mysql_free_result(result);
76			}
77			mysql_close( &connection );
78			return success;
79			}
80
81			int sql_worker(dlrecord dl, char sql) {
82			MYSQL connection;
83			mysql_init( &connection);
84			int retval;
85			char buf[120];
86
87			if (!mysql_real_connect( &connection, CONFIG->dbhost, CONFIG->dbuser, CONFIG->dbpass, CONFIG->dbname, CONFIG->dbport, NULL, 0)) {
88			sprintf(buf,"Could not connect to DB : %s\n", mysql_error(&connection));
89			log_message(buf);
90			}
91			retval = mysql_query(&connection, sql);
92			mysql_close(&connection);
93			return retval;
94			}
95
96			int sql_update_bytecount(dlrecord *record, long long bytes) {
97			char buf[120];
98			sprintf(buf, "UPDATE downloads SET bytecount=%lld WHERE id=%d", bytes, record->id);
99			return sql_worker(record, buf);
100			}
101
102			int sql_start_download(dlrecord *record) {
103			char buf[120];
104			sprintf(buf, "UPDATE downloads SET status=1, started=now() WHERE id=%d", record->id);
105			return sql_worker(record, buf);
106			}
107
108			int sql_abort_download(dlrecord record, char message) {
109			char buf[250];
110			sprintf(buf, "UPDATE downloads SET status=-1, completed=now(),message='%s' WHERE id=%d",message, record->id);
111			return sql_worker(record, buf);
112			}
113
114			int sql_stop_download(dlrecord *record) {
115			char buf[120];
116			sprintf(buf, "UPDATE downloads SET status=2, completed=now() WHERE id=%d", record->id);
117			return sql_worker(record, buf);
118			}