on master help sign in
better version of executing queries, may need to get rid of the old ones later
commitfaaad7e056aefc591a757b56d50b80925e403c71 authormelissam <mfm6@st-andrews.ac.uk> dateFri 20 Jul 2012, 12:03 (14 years ago) parent5ebf30e
3 files changed+196−22
contribs/lib/SQLite3/sqlite3.idr+71−8 contribs/lib/SQLite3/sqlite3api.c+107−14 contribs/lib/SQLite3/sqlite3api.h+18
contribs/lib/SQLite3/sqlite3.idr+71−8
@@ -63,10 +63,12 @@ close_db (MkDBPointer pointer) =mkForeign (FFun "sqlite3_close_idr" [FPtr] FInt)
6363exec_db : DBPointer -> String -> DB Int
6464exec_db (MkDBPointer pointer) stmt = do
6565 x <- liftIO(mkForeign (FFun "sqlite3_exec_idr" [FPtr, FString] FInt)pointer stmt)
66− if (x /= 0)
66+ if (x == 1)
6767 then fail "Could not execute."
6868 else return x
69−
69+
70+
71+
7072get_error : DBPointer -> IO String
7173get_error (MkDBPointer pointer) =mkForeign (FFun "sqlite3_get_error" [FPtr] FString)pointer
7274
@@ -80,6 +82,12 @@ prepare_db (MkDBPointer pointer) cmd = do
8082 then fail "Error occured"
8183 else return (MkStmtPtr result)
8284
85+exec_db_v2 : StmtPtr -> DB Int
86+exec_db_v2 (MkStmtPtr pointer) = do
87+ x <- liftIO(mkForeign (FFun "exec_db" [FPtr] FInt)pointer)
88+ if (x == 21)
89+ then fail "No data returned."
90+ else return x
8391
8492step_db : StmtPtr -> DB Int
8593step_db (MkStmtPtr pointer) = do
@@ -194,6 +202,53 @@ get_data (MkDBPointer pointer) tbl_name row col = do
194202 helper 3 = do i <- get_data_val_text pointer tbl_name row col; return (DBText i)
195203 helper _ = return (DBNull)
196204
205+num_row_v2 : DBPointer -> IO Int
206+num_row_v2 (MkDBPointer pointer) = do x <- mkForeign (FFun "sqlite3_get_num_row_v2" [FPtr] FInt)pointer
207+ return (x)
208+
209+num_col_v2 : DBPointer -> IO Int
210+num_col_v2 (MkDBPointer pointer) = do x <- mkForeign (FFun "sqlite3_get_num_col_v2" [FPtr] FInt)pointer
211+ return (x)
212+
213+
214+get_data_v2 : DBPointer -> Int -> Int -> IO DBVal
215+get_data_v2 (MkDBPointer pointer) row col = do
216+ ty <- get_data_type_v2 pointer row col
217+ helper ty
218+ where
219+ get_data_type_v2 : Ptr ->Int -> Int -> IO Int
220+ get_data_type_v2 pointer row col =mkForeign (FFun "sqlite3_get_data_type_v2" [FPtr, FInt, FInt] FInt)pointer row col
221+
222+ get_data_val_int : Ptr -> Int -> IO Int
223+ get_data_val_int pointer col =mkForeign (FFun "sqlite3_get_val_int_v2" [FPtr,FInt] FInt)pointer col
224+
225+ get_data_val_text : Ptr -> Int -> IO String
226+ get_data_val_text pointer col =mkForeign (FFun "sqlite3_get_val_text_v2" [FPtr, FInt] FString)pointer col
227+
228+
229+
230+ get_data_val_float : Ptr -> Int -> IO Float
231+ get_data_val_float pointer col =mkForeign (FFun "sqlite3_get_float_v2" [FPtr, FInt] FFloat)pointer col
232+
233+ helper : Int -> IO DBVal
234+ helper 1 = do i <- get_data_val_int pointer col ; return (DBInt i)
235+ helper 2 = do i <- get_data_val_float pointer col ; return (DBFloat i)
236+ helper 3 = do i <- get_data_val_text pointer col; return (DBText i)
237+ helper _ =return (DBNull)
238+
239+
240+toList_v2 : DBPointer -> DB Table
241+toList_v2 db = do
242+ nbR <- liftIO (num_row_v2 db)
243+ nmC <- liftIO (num_col_v2 db)
244+ res <- forM [1..(nbR)] (\ i =>
245+ forM [0..(nmC-1)] (\ j =>
246+ liftIO(get_data_v2 db i j)
247+
248+ )
249+ )
250+ return res
251+
197252
198253strcat : String -> String-> String
199254strcat str1 str2 = (str1 ++ str2)
@@ -215,8 +270,7 @@ toList name stmt x = do
215270 )
216271 )
217272 liftIO (free_table ptr)
218− return res
219−
273+ return res
220274
221275bind_int : StmtPtr -> Int -> Int -> DB Int
222276bind_int (MkStmtPtr pointer) indexval val = do
@@ -272,14 +326,23 @@ test2 : DB ()
272326test2 = do db <- liftIO (open_db "students.db")
273327 tbl <- (get_table db "SELECT Student.Name FROM Student, Module, Enrollment WHERE Module.Credits = 30 AND Student.School_student = Enrollment.School_Student AND Student.School = Enrollment.School AND Enrollment.Code = Module.Code")
274328 res <- (toList "Student" "select *from Student;" db)
275− --mydata <- liftIO ( get_data db "Student" 1 1)
276329 liftIO(print res)
277330 c <- liftIO (close_db db)
278331 return ()
279−
332+
333+test3 : DB()
334+test3 = do db <- liftIO (open_db "somedb.db")
335+ stmt <- (prepare_db db "select *from tbl1 where num >10;")
336+ exec_db_v2 stmt
337+ res <- toList_v2 db
338+ liftIO(print res)
339+ finalize_db stmt
340+ c <- liftIO (close_db db)
341+ return ()
342+
343+
280344main : IO ()
281−main = do x <- runDB (test2)
282− --y <- runDB (test)
345+main = do x <- runDB (test3)
283346 return ()
284347
285348
contribs/lib/SQLite3/sqlite3api.c+107−14
@@ -110,23 +110,59 @@ void* sqlite3_get_table_idr(void* db,
110110 }
111111
112112 tbl -> data_size = array_size;
113+ return tbl;
114+
115+}
116+
117+int exec_db(void*p){
118+
119+ DBinfo* dbi =(DBinfo*) p;
120+ int rc, col, type, row_counter,i;
121+
122+ const char* col_name;
113123
114− int i, j, row, col;
124+ rc = sqlite3_step(dbi->ppStmt);
125+
126+ if(rc != SQLITE_ROW && rc != SQLITE_DONE){
127+ return 21;
128+ }
129+ col = sqlite3_column_count(dbi->ppStmt);
130+
131+ row_counter =0;
132+
133+ while (rc == SQLITE_ROW) {
115134
116− row = tbl->num_row;
117− col = tbl->num_col;
118−
119−// for( i=0; i< row; i++){
120−// for(j=0;j<col; j++){
121−// printf("%s\n", tbl->table_data[(i+1)*col +j]);
122−//
123−// }
124−//
125−// }
126− return tbl;
135+ rc = sqlite3_step(dbi->ppStmt);
136+ row_counter++;
137+ }
138+
139+ dbi->row_count = row_counter;
140+
141+ dbi->col_count = col;
142+
143+
144+ return type;
127145
128146}
129147
148+int sqlite3_get_num_row_v2(void* p){
149+
150+ DBinfo* dbi =(DBinfo*) p;
151+ int row_number =dbi->row_count;
152+ return row_number;
153+
154+
155+}
156+
157+int sqlite3_get_num_col_v2(void* p){
158+
159+ DBinfo* dbi =(DBinfo*) p;
160+ int col_number =dbi-> col_count;
161+ return col_number;
162+
163+}
164+
165+
130166int sqlite3_get_num_row(void* p){
131167
132168 Table* tbl =(Table*) p;
@@ -142,6 +178,38 @@ int sqlite3_get_num_col(void* p){
142178 int col_number =tbl-> num_col;
143179 return col_number;
144180
181+}
182+int sqlite3_get_data_type_v2(void* p, int nRow, int nCol){
183+
184+ DBinfo* dbi =(DBinfo*) p;
185+ int rc, type, row_counter;
186+ const char* char_int;
187+
188+ rc = sqlite3_reset(dbi->ppStmt);
189+ rc = sqlite3_step(dbi->ppStmt);
190+ row_counter =0;
191+
192+ while (rc == SQLITE_ROW && row_counter < nRow) {
193+
194+ rc = sqlite3_step(dbi->ppStmt);
195+ row_counter++;
196+ }
197+ type =sqlite3_column_type(dbi->ppStmt, nCol);
198+ return type;
199+
200+}
201+
202+
203+int sqlite3_get_val_int_v2(void* p, int nCol){
204+
205+
206+ DBinfo* dbi =(DBinfo*) p;
207+ int val, col;
208+
209+ val =sqlite3_column_int(dbi->ppStmt, nCol);
210+ return val;
211+
212+
145213}
146214
147215int sqlite3_get_data_type(void* p,const char* tbl_name, int nRow, int nCol){
@@ -171,7 +239,6 @@ int sqlite3_get_data_type(void* p,const char* tbl_name, int nRow, int nCol){
171239 }
172240
173241 val =sqlite3_column_type(stmt, nCol);
174− //printf("Type : %d\n", val );
175242
176243 rc =sqlite3_finalize(stmt);
177244 if( rc != SQLITE_OK){
@@ -219,6 +286,22 @@ int sqlite3_get_val_int(void* p,const char* tbl_name, int nRow, int nCol){
219286
220287}
221288
289+const unsigned char* sqlite3_get_val_text_v2(void* p,int nCol){
290+
291+
292+ DBinfo* dbi =(DBinfo*) p;
293+ int rc,i, val, counter;
294+ const unsigned char* text_val;
295+
296+ array =(unsigned char *) GC_malloc(1000*sizeof(char));
297+
298+ text_val =sqlite3_column_text(dbi->ppStmt, nCol);
299+ memcpy(array, text_val, sizeof(text_val));
300+ return array;
301+
302+}
303+
304+
222305
223306const unsigned char* sqlite3_get_val_text(void* p,const char* tbl_name,int nRow, int nCol){
224307
@@ -291,7 +374,6 @@ float sqlite3_get_float(void* p,const char* tbl_name, int nRow, int nCol){
291374 }
292375 double_val =sqlite3_column_double(stmt, nCol);
293376 float float_val =(float)double_val;
294− //printf("float_val %f\n", float_val);
295377 rc =sqlite3_finalize(stmt);
296378 if( rc != SQLITE_OK){
297379 printf("Couldn't finalize.\n");
@@ -300,6 +382,17 @@ float sqlite3_get_float(void* p,const char* tbl_name, int nRow, int nCol){
300382
301383}
302384
385+float sqlite3_get_float_v2(void* p, int nCol){
386+
387+ DBinfo* dbi =(DBinfo*) p;
388+ double double_val;
389+
390+ double_val =sqlite3_column_double(dbi->ppStmt, nCol);
391+ float float_val =(float)double_val;
392+ return float_val;
393+
394+}
395+
303396
304397void sqlite3_free_table_idr(void* db){
305398 Table* tbl =(Table*) db;
contribs/lib/SQLite3/sqlite3api.h+18
@@ -20,6 +20,8 @@ typedef struct {
2020 const char *Ptr_tail;
2121 sqlite3_blob *blob_ptr;
2222 sqlite3_value *value;
23+ int row_count;
24+ int col_count;
2325
2426} DBinfo;
2527
@@ -37,8 +39,12 @@ typedef struct {
3739 DBinfo* database;
3840}Table;
3941
42+int* int_list ();
43+
4044void* sqlite3_open_idr(const char *filename);
4145
46+int exec_db(void*p);
47+
4248int sqlite3_close_idr(void* db);
4349
4450int sqlite3_exec_idr(void*, const char *sql);
@@ -47,6 +53,10 @@ char* sqlite3_get_error(void* db);
4753
4854
4955const unsigned char* sqlite3_get_val_text(void* p,const char* tbl_name,int nRow, int nCol);
56+
57+const unsigned char* sqlite3_get_val_text_v2(void* p,int nCol);
58+
59+float sqlite3_get_float_v2(void* p, int nCol);
5060
5161void* sqlite3_get_table_idr(void* db, const char *sql);
5262
@@ -54,10 +64,18 @@ int sqlite3_get_num_col(void* p);
5464
5565int sqlite3_get_num_row(void* p);
5666
67+int sqlite3_get_num_row_v2(void* p);
68+
69+int sqlite3_get_num_col_v2(void* p);
70+
5771void* sqlite3_get_data(void* p, int nRow, int nCol);
5872
5973int sqlite3_get_data_type(void* p,const char* tbl_name,int nRow, int nCol);
6074
75+int sqlite3_get_data_type_v2(void* p, int nRow, int nCol);
76+
77+int sqlite3_get_val_int_v2(void* p,int nCo);
78+
6179int sqlite3_get_val_int(void* p,const char* tbl_name, int nRow, int nCol);
6280
6381float sqlite3_get_float(void* p,const char* tbl_name, int nRow, int nCol);