adding a primitive mariadb client device
This commit is contained in:
		| @ -4,36 +4,55 @@ | |||||||
| #include <stdio.h> | #include <stdio.h> | ||||||
| #include <string.h> | #include <string.h> | ||||||
|  |  | ||||||
|  | #include <mariadb/mysql.h> | ||||||
|  |  | ||||||
| static void maria_on_disconnect (mio_dev_maria_t* dev) | static void maria_on_disconnect (mio_dev_maria_t* dev) | ||||||
| { | { | ||||||
| } | } | ||||||
|  |  | ||||||
| static void maria_on_connect (mio_dev_maria_t* dev) | static void maria_on_connect (mio_dev_maria_t* dev) | ||||||
| { | { | ||||||
| printf ("CONNEcTED...\n"); | printf ("CONNECTED...\n"); | ||||||
| 	if (mio_dev_maria_querywithbchars(dev, "SHOW STATUS", 11) <= -1) | 	if (mio_dev_maria_querywithbchars(dev, "SHOW STATUS", 11) <= -1) | ||||||
| 	{ | 	{ | ||||||
| 		mio_dev_maria_halt (dev); | 		mio_dev_maria_halt (dev); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| static void maria_on_query_started (mio_dev_maria_t* dev) | static void maria_on_query_started (mio_dev_maria_t* dev, int maria_ret) | ||||||
| { | { | ||||||
| printf ("QUERY SENT...\n"); | 	if (maria_ret != 0) | ||||||
| 	if (mio_dev_maria_fetchrow(dev) <= -1) |  | ||||||
| 	{ | 	{ | ||||||
| printf ("FETCH ROW FAILURE\n"); | printf ("QUERY NOT SENT PROPERLY..%s\n", mysql_error(dev->hnd)); | ||||||
| 		mio_dev_maria_halt (dev); | 	} | ||||||
|  | 	else | ||||||
|  | 	{ | ||||||
|  | printf ("QUERY SENT..\n"); | ||||||
|  | 		if (mio_dev_maria_fetchrows(dev) <= -1) | ||||||
|  | 		{ | ||||||
|  | printf ("FETCH ROW FAILURE - %s\n", mysql_error(dev->hnd)); | ||||||
|  | 			mio_dev_maria_halt (dev); | ||||||
|  | 		} | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| static void maria_on_row_fetched (mio_dev_maria_t* dev, void* row) | static void maria_on_row_fetched (mio_dev_maria_t* dev, void* data) | ||||||
| { | { | ||||||
| 	if (!row) printf ("NO MORE ROW..\n"); | 	MYSQL_ROW row = (MYSQL_ROW)data; | ||||||
|  | 	static int x = 0; | ||||||
|  | 	if (!row)  | ||||||
|  | 	{ | ||||||
|  | 		printf ("NO MORE ROW..\n"); | ||||||
|  | 		if (x == 0 && mio_dev_maria_querywithbchars(dev, "SELECT * FROM pdns.records", 26) <= -1) mio_dev_maria_halt (dev); | ||||||
|  | 		x++; | ||||||
|  | 	} | ||||||
| 	else | 	else | ||||||
| 	{	 | 	{ | ||||||
| 		printf ("GOT ROW\n"); | 		if (x == 0) | ||||||
| 		mio_dev_maria_fetchrow (dev); | 			printf ("%s %s\n", row[0], row[1]); | ||||||
|  | 		else if (x == 1) | ||||||
|  | 			printf ("%s %s %s %s %s\n", row[0], row[1], row[2], row[3], row[4]); | ||||||
|  | 		//printf ("GOT ROW\n"); | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | |||||||
							
								
								
									
										443
									
								
								mio/lib/maria.c
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										443
									
								
								mio/lib/maria.c
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,443 @@ | |||||||
|  | /* | ||||||
|  |  * $Id$ | ||||||
|  |  * | ||||||
|  |     Copyright (c) 2016-2020 Chung, Hyung-Hwan. All rights reserved. | ||||||
|  |  | ||||||
|  |     Redistribution and use in source and binary forms, with or without | ||||||
|  |     modification, are permitted provided that the following conditions | ||||||
|  |     are met: | ||||||
|  |     1. Redistributions of source code must retain the above copyright | ||||||
|  |        notice, this list of conditions and the following disclaimer. | ||||||
|  |     2. Redistributions in binary form must reproduce the above copyright | ||||||
|  |        notice, this list of conditions and the following disclaimer in the | ||||||
|  |        documentation and/or other materials provided with the distribution. | ||||||
|  |  | ||||||
|  |     THIS SOFTWARE IS PROVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR | ||||||
|  |     IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WAfRRANTIES | ||||||
|  |     OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. | ||||||
|  |     IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, | ||||||
|  |     INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT | ||||||
|  |     NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | ||||||
|  |     DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | ||||||
|  |     THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | ||||||
|  |     (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF | ||||||
|  |     THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||||||
|  |  */ | ||||||
|  |  | ||||||
|  | #include <mio-maria.h> | ||||||
|  | #include "mio-prv.h" | ||||||
|  |  | ||||||
|  | #include <mariadb/mysql.h> | ||||||
|  |  | ||||||
|  |  | ||||||
|  | /* ========================================================================= */ | ||||||
|  |  | ||||||
|  | static int dev_maria_make (mio_dev_t* dev, void* ctx) | ||||||
|  | { | ||||||
|  | 	mio_t* mio = dev->mio; | ||||||
|  | 	mio_dev_maria_t* rdev = (mio_dev_maria_t*)dev; | ||||||
|  | 	mio_dev_maria_make_t* mi = (mio_dev_maria_make_t*)ctx; | ||||||
|  |  | ||||||
|  | 	rdev->hnd = mysql_init(MIO_NULL); | ||||||
|  | 	if (MIO_UNLIKELY(!rdev->hnd))  | ||||||
|  | 	{ | ||||||
|  | 		mio_seterrnum (mio, MIO_ESYSMEM); | ||||||
|  | 		return -1; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	if (mysql_options(rdev->hnd, MYSQL_OPT_NONBLOCK, 0) != 0) | ||||||
|  | 	{ | ||||||
|  | 		mio_seterrbfmt (mio, MIO_ESYSERR, "%s", mysql_error(rdev->hnd)); | ||||||
|  | 		mysql_close (rdev->hnd); | ||||||
|  | 		rdev->hnd = MIO_NULL; | ||||||
|  | 		return -1; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	{ | ||||||
|  | 		my_bool x = 0; | ||||||
|  | 		mysql_options(rdev->hnd, MYSQL_OPT_RECONNECT, &x); | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	rdev->dev_cap = MIO_DEV_CAP_IN | MIO_DEV_CAP_OUT | MIO_DEV_CAP_VIRTUAL; /* mysql_init() doesn't create a socket. so no IO is possible at this point */ | ||||||
|  | 	rdev->on_read = mi->on_read; | ||||||
|  | 	rdev->on_write = mi->on_write; | ||||||
|  | 	rdev->on_connect = mi->on_connect; | ||||||
|  | 	rdev->on_disconnect = mi->on_disconnect; | ||||||
|  | 	rdev->on_query_started = mi->on_query_started; | ||||||
|  | 	rdev->on_row_fetched = mi->on_row_fetched; | ||||||
|  | 	return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int dev_maria_kill (mio_dev_t* dev, int force) | ||||||
|  | { | ||||||
|  | 	/*mio_t* mio = dev->mio;*/ | ||||||
|  | 	mio_dev_maria_t* rdev = (mio_dev_maria_t*)dev; | ||||||
|  |  | ||||||
|  | 	if (rdev->on_disconnect) rdev->on_disconnect (rdev); | ||||||
|  |  | ||||||
|  | 	if (rdev->hnd) | ||||||
|  | 	{ | ||||||
|  | 		mysql_close (rdev->hnd); | ||||||
|  | 		rdev->hnd = MIO_NULL; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return 0; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static mio_syshnd_t dev_maria_getsyshnd (mio_dev_t* dev) | ||||||
|  | { | ||||||
|  | 	mio_dev_maria_t* rdev = (mio_dev_maria_t*)dev; | ||||||
|  | 	return (mio_syshnd_t)mysql_get_socket(rdev->hnd); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int events_to_mysql_wstatus (int events) | ||||||
|  | { | ||||||
|  | 	int wstatus = 0; | ||||||
|  | 	if (events & MIO_DEV_EVENT_IN) wstatus |= MYSQL_WAIT_READ; | ||||||
|  | 	if (events & MIO_DEV_EVENT_OUT) wstatus |= MYSQL_WAIT_WRITE; | ||||||
|  | 	if (events & MIO_DEV_EVENT_PRI) wstatus |= MYSQL_WAIT_EXCEPT; | ||||||
|  | 	return wstatus; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static int mysql_wstatus_to_events (int wstatus) | ||||||
|  | { | ||||||
|  | 	int events = 0; | ||||||
|  | 	if (wstatus & MYSQL_WAIT_READ) events |= MIO_DEV_EVENT_IN; | ||||||
|  | 	if (wstatus & MYSQL_WAIT_WRITE) events |= MIO_DEV_EVENT_OUT; | ||||||
|  | 	if (wstatus & MYSQL_WAIT_EXCEPT) events |= MIO_DEV_EVENT_PRI; | ||||||
|  | /* TODO: wstatus& MYSQL_WAIT_TIMEOUT? */ | ||||||
|  | 	return events; | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static MIO_INLINE void watch_mysql (mio_dev_maria_t* rdev, int wstatus) | ||||||
|  | { | ||||||
|  | 	if (mio_dev_watch((mio_dev_t*)rdev, MIO_DEV_WATCH_UPDATE, mysql_wstatus_to_events(wstatus)) <= -1) | ||||||
|  | 	{ | ||||||
|  | 		/* watcher update failure. it's critical */ | ||||||
|  | 		mio_stop (rdev->mio, MIO_STOPREQ_WATCHER_ERROR); | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  |  | ||||||
|  | static void start_fetch_row (mio_dev_maria_t* rdev) | ||||||
|  | { | ||||||
|  | 	MYSQL_ROW row; | ||||||
|  | 	int status; | ||||||
|  |  | ||||||
|  | 	status = mysql_fetch_row_start(&row, rdev->res); | ||||||
|  | 	MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_ROW_FETCHING); | ||||||
|  | 	if (status) | ||||||
|  | 	{ | ||||||
|  | 		/* row fetched */ | ||||||
|  | 		rdev->row_fetched = 0; | ||||||
|  | 		watch_mysql (rdev, status); | ||||||
|  | 	} | ||||||
|  | 	else | ||||||
|  | 	{ | ||||||
|  | 		/* row fetched - don't handle it immediately here */ | ||||||
|  | 		rdev->row_fetched = 1; | ||||||
|  | 		rdev->row_wstatus = status; | ||||||
|  | 		rdev->row = row; | ||||||
|  | 		watch_mysql (rdev, MYSQL_WAIT_READ | MYSQL_WAIT_WRITE); | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  |  | ||||||
|  | static int dev_maria_ioctl (mio_dev_t* dev, int cmd, void* arg) | ||||||
|  | { | ||||||
|  | 	mio_t* mio = dev->mio; | ||||||
|  | 	mio_dev_maria_t* rdev = (mio_dev_maria_t*)dev; | ||||||
|  |  | ||||||
|  | 	switch (cmd) | ||||||
|  | 	{ | ||||||
|  | 		case MIO_DEV_MARIA_CONNECT: | ||||||
|  | 		{ | ||||||
|  | 			mio_dev_maria_connect_t* ci = (mio_dev_maria_connect_t*)arg; | ||||||
|  | 			MYSQL* ret; | ||||||
|  | 			int status; | ||||||
|  |  | ||||||
|  |  | ||||||
|  | 			if (MIO_DEV_MARIA_GET_PROGRESS(rdev)) | ||||||
|  | 			{ | ||||||
|  | 				/* can't connect again */ | ||||||
|  | 				mio_seterrbfmt (mio, MIO_EPERM, "operation in progress. disallowed to connect again"); | ||||||
|  | 				return -1; | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			status = mysql_real_connect_start(&ret, rdev->hnd, ci->host, ci->username, ci->password, ci->dbname, ci->port, MIO_NULL, 0); | ||||||
|  | 			rdev->dev_cap &= ~MIO_DEV_CAP_VIRTUAL; /* a socket is created in mysql_real_connect_start() */ | ||||||
|  | 			if (status) | ||||||
|  | 			{ | ||||||
|  | 				/* not connected */ | ||||||
|  | 				MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_CONNECTING); | ||||||
|  | 				rdev->connected = 0; | ||||||
|  | 				watch_mysql (rdev, status); | ||||||
|  | 			} | ||||||
|  | 			else | ||||||
|  | 			{ | ||||||
|  | 				/* connected immediately */ | ||||||
|  | 				if (MIO_UNLIKELY(!ret)) | ||||||
|  | 				{ | ||||||
|  | 					mio_seterrbfmt (mio, MIO_ESYSERR, "%s", mysql_error(rdev->hnd)); | ||||||
|  | 					return -1; | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				/* regiter it in the multiplexer so that the ready() handler is | ||||||
|  | 				 * invoked to call the on_connect() callback */ | ||||||
|  | 				rdev->connected = 1; | ||||||
|  | 				watch_mysql (rdev, MYSQL_WAIT_READ | MYSQL_WAIT_WRITE); /* TODO: verify this */ | ||||||
|  | 			} | ||||||
|  | 			return 0; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		case MIO_DEV_MARIA_QUERY_WITH_BCS: | ||||||
|  | 		{ | ||||||
|  | 			const mio_bcs_t* qstr = (const mio_bcs_t*)arg; | ||||||
|  | 			int err, status; | ||||||
|  |  | ||||||
|  | 			if (rdev->res) /* TODO: more accurate check */ | ||||||
|  | 			{ | ||||||
|  | 				mio_seterrbfmt (mio, MIO_EPERM, "operation in progress. disallowed to query again"); | ||||||
|  | 				return -1; | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			status = mysql_real_query_start(&err, rdev->hnd, qstr->ptr, qstr->len); | ||||||
|  | 			MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_QUERY_STARTING); | ||||||
|  | 			if (status) | ||||||
|  | 			{ | ||||||
|  | 				/* not done */ | ||||||
|  | 				rdev->query_started = 0; | ||||||
|  | 				watch_mysql (rdev, status); | ||||||
|  | 			} | ||||||
|  | 			else | ||||||
|  | 			{ | ||||||
|  | 				/* query sent immediately */ | ||||||
|  | 				if (MIO_UNLIKELY(err)) | ||||||
|  | 				{ | ||||||
|  | 					mio_seterrbfmt (mio, MIO_ESYSERR, "%s", mysql_error(rdev->hnd)); | ||||||
|  | 					return -1; | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				rdev->query_started = 1; | ||||||
|  | 				rdev->query_ret = err; | ||||||
|  | 				watch_mysql (rdev, MYSQL_WAIT_READ | MYSQL_WAIT_WRITE); | ||||||
|  | 			} | ||||||
|  | 			return 0; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		case MIO_DEV_MARIA_FETCH_ROW: | ||||||
|  | 		{ | ||||||
|  | 			int status; | ||||||
|  |  | ||||||
|  | 			if (!rdev->res) | ||||||
|  | 			{ | ||||||
|  | 				rdev->res = mysql_use_result(rdev->hnd); | ||||||
|  | 				if (MIO_UNLIKELY(!rdev->res)) | ||||||
|  | 				{ | ||||||
|  | 					mio_seterrbfmt (mio, MIO_ESYSERR, "%s", mysql_error(rdev->hnd)); | ||||||
|  | 					return -1; | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			start_fetch_row (rdev); | ||||||
|  | 			return 0; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		default: | ||||||
|  | 			mio_seterrnum (mio, MIO_EINVAL); | ||||||
|  | 			return -1; | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static mio_dev_mth_t dev_maria_methods =  | ||||||
|  | { | ||||||
|  | 	dev_maria_make, | ||||||
|  | 	dev_maria_kill, | ||||||
|  | 	dev_maria_getsyshnd, | ||||||
|  |  | ||||||
|  | 	MIO_NULL, | ||||||
|  | 	MIO_NULL, | ||||||
|  | 	MIO_NULL, | ||||||
|  | 	dev_maria_ioctl | ||||||
|  | }; | ||||||
|  |  | ||||||
|  | /* ========================================================================= */ | ||||||
|  |  | ||||||
|  |  | ||||||
|  | static int dev_evcb_maria_ready (mio_dev_t* dev, int events) | ||||||
|  | { | ||||||
|  | 	mio_t* mio = dev->mio; | ||||||
|  | 	mio_dev_maria_t* rdev = (mio_dev_maria_t*)dev; | ||||||
|  |  | ||||||
|  | #if 0 | ||||||
|  | 	if (events & MIO_DEV_EVENT_ERR) | ||||||
|  | 	{ | ||||||
|  | 		int errcode; | ||||||
|  | 		mio_scklen_t len; | ||||||
|  |  | ||||||
|  | 		len = MIO_SIZEOF(errcode); | ||||||
|  | 		if (getsockopt(mysql_get_socket(rdev->hnd), SOL_SOCKET, SO_ERROR, (char*)&errcode, &len) == -1) | ||||||
|  | 		{ | ||||||
|  | 			/* the error number is set to the socket error code. | ||||||
|  | 			 * errno resulting from getsockopt() doesn't reflect the actual | ||||||
|  | 			 * socket error. so errno is not used to set the error number. | ||||||
|  | 			 * instead, the generic device error MIO_EDEVERRR is used */ | ||||||
|  | 			mio_seterrbfmt (mio, MIO_EDEVERR, "device error - unable to get SO_ERROR"); | ||||||
|  | 		} | ||||||
|  | 		else | ||||||
|  | 		{ | ||||||
|  | 			mio_seterrwithsyserr (mio, 0, errcode); | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		return -1; | ||||||
|  | 	} | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | 	switch (MIO_DEV_MARIA_GET_PROGRESS(rdev)) | ||||||
|  | 	{ | ||||||
|  | 		case MIO_DEV_MARIA_CONNECTING: | ||||||
|  | 		 | ||||||
|  | 			if (rdev->connected) | ||||||
|  | 			{ | ||||||
|  | 				rdev->connected = 0; | ||||||
|  | 				MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_CONNECTED); | ||||||
|  | 				if (rdev->on_connect) rdev->on_connect (rdev); | ||||||
|  | 			} | ||||||
|  | 			else | ||||||
|  | 			{ | ||||||
|  | 				int status; | ||||||
|  | 				MYSQL* tmp; | ||||||
|  |  | ||||||
|  | 				status = mysql_real_connect_cont(&tmp, rdev->hnd, events_to_mysql_wstatus(events)); | ||||||
|  | 				watch_mysql (rdev, status); | ||||||
|  |  | ||||||
|  | 				if (!status) | ||||||
|  | 				{ | ||||||
|  | 					/* connected */ | ||||||
|  | 					MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_CONNECTED); | ||||||
|  | 					if (rdev->on_connect) rdev->on_connect (rdev); | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  | 			break; | ||||||
|  |  | ||||||
|  | 		case MIO_DEV_MARIA_QUERY_STARTING: | ||||||
|  | 			if (rdev->query_started) | ||||||
|  | 			{ | ||||||
|  | 				rdev->query_started = 0; | ||||||
|  | 				MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_QUERY_STARTED); | ||||||
|  | 				if (rdev->on_query_started) rdev->on_query_started (rdev, rdev->query_ret); | ||||||
|  | 			} | ||||||
|  | 			else | ||||||
|  | 			{ | ||||||
|  | 				int status; | ||||||
|  | 				int tmp; | ||||||
|  |  | ||||||
|  | 				status = mysql_real_query_cont(&tmp, rdev->hnd, events_to_mysql_wstatus(events)); | ||||||
|  | 				watch_mysql (rdev, status); | ||||||
|  |  | ||||||
|  | 				if (!status) | ||||||
|  | 				{ | ||||||
|  | 					/* query sent */ | ||||||
|  | 					MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_QUERY_STARTED); | ||||||
|  | 					if (rdev->on_query_started) rdev->on_query_started (rdev, tmp); | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			break; | ||||||
|  |  | ||||||
|  | 		case MIO_DEV_MARIA_ROW_FETCHING: | ||||||
|  | 		{ | ||||||
|  | 			int status; | ||||||
|  | 			MYSQL_ROW row; | ||||||
|  |  | ||||||
|  | 			if (rdev->row_fetched) | ||||||
|  | 			{ | ||||||
|  | 				row = (MYSQL_ROW)rdev->row; | ||||||
|  | 				rdev->row_fetched = 0; | ||||||
|  |  | ||||||
|  | 				if (!row) | ||||||
|  | 				{ | ||||||
|  | 					MIO_ASSERT (mio, rdev->res != MIO_NULL); | ||||||
|  | 					mysql_free_result (rdev->res); /* this doesn't block after the last row */ | ||||||
|  | 					rdev->res = MIO_NULL; | ||||||
|  |  | ||||||
|  | 					watch_mysql (rdev, rdev->row_wstatus); | ||||||
|  | 				} | ||||||
|  |  | ||||||
|  | 				MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_ROW_FETCHED); | ||||||
|  | 				if (rdev->on_row_fetched) rdev->on_row_fetched (rdev, row); | ||||||
|  |  | ||||||
|  | 				if (row) start_fetch_row (rdev); | ||||||
|  | 			} | ||||||
|  | 			else | ||||||
|  | 			{ | ||||||
|  | 				/* TODO: if rdev->res is MIO_NULL, error.. */ | ||||||
|  | 				status = mysql_fetch_row_cont(&row, rdev->res, events_to_mysql_wstatus(events)); | ||||||
|  |  | ||||||
|  | 				if (!status) | ||||||
|  | 				{ | ||||||
|  | 					if (!row)  | ||||||
|  | 					{ | ||||||
|  | 						/* the last row has been received - cleanup before invoking the callback */ | ||||||
|  | 						watch_mysql (rdev, status); | ||||||
|  |  | ||||||
|  | 						MIO_ASSERT (mio, rdev->res != MIO_NULL); | ||||||
|  | 						mysql_free_result (rdev->res); /* this doesn't block after the last row */ | ||||||
|  | 						rdev->res = MIO_NULL; | ||||||
|  | 					} | ||||||
|  |  | ||||||
|  | 					MIO_DEV_MARIA_SET_PROGRESS (rdev, MIO_DEV_MARIA_ROW_FETCHED); | ||||||
|  | 					if (rdev->on_row_fetched) rdev->on_row_fetched (rdev, row); | ||||||
|  |  | ||||||
|  | 					if (row) start_fetch_row (rdev); /* arrange to fetch the next row */ | ||||||
|  | 				} | ||||||
|  | 				else | ||||||
|  | 				{ | ||||||
|  | 					watch_mysql (rdev, status); | ||||||
|  | 				} | ||||||
|  | 			} | ||||||
|  |  | ||||||
|  | 			break; | ||||||
|  | 		} | ||||||
|  |  | ||||||
|  | 		default: | ||||||
|  | 			mio_seterrbfmt (mio, MIO_EINTERN, "invalid progress state in maria"); | ||||||
|  | 			return -1; | ||||||
|  | 	} | ||||||
|  |  | ||||||
|  | 	return 0; /* success. but skip core event handling */ | ||||||
|  | } | ||||||
|  |  | ||||||
|  | static mio_dev_evcb_t dev_maria_event_callbacks = | ||||||
|  | { | ||||||
|  | 	dev_evcb_maria_ready, | ||||||
|  | 	MIO_NULL, /* no read callback */ | ||||||
|  | 	MIO_NULL  /* no write callback */ | ||||||
|  | }; | ||||||
|  |  | ||||||
|  |  | ||||||
|  | /* ========================================================================= */ | ||||||
|  |  | ||||||
|  |  | ||||||
|  | mio_dev_maria_t* mio_dev_maria_make (mio_t* mio, mio_oow_t xtnsize, const mio_dev_maria_make_t* mi) | ||||||
|  | { | ||||||
|  | 	return (mio_dev_maria_t*)mio_dev_make( | ||||||
|  | 		mio, MIO_SIZEOF(mio_dev_maria_t) + xtnsize, | ||||||
|  | 		&dev_maria_methods, &dev_maria_event_callbacks, (void*)mi); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int mio_dev_maria_connect (mio_dev_maria_t* dev, mio_dev_maria_connect_t* ci) | ||||||
|  | { | ||||||
|  | 	return mio_dev_ioctl((mio_dev_t*)dev, MIO_DEV_MARIA_CONNECT, ci); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int mio_dev_maria_querywithbchars (mio_dev_maria_t* dev, const mio_bch_t* qstr, mio_oow_t qlen) | ||||||
|  | { | ||||||
|  | 	mio_bcs_t bcs = { (mio_bch_t*)qstr, qlen}; | ||||||
|  | 	return mio_dev_ioctl((mio_dev_t*)dev, MIO_DEV_MARIA_QUERY_WITH_BCS, &bcs); | ||||||
|  | } | ||||||
|  |  | ||||||
|  | int mio_dev_maria_fetchrows (mio_dev_maria_t* dev) | ||||||
|  | { | ||||||
|  | 	return mio_dev_ioctl((mio_dev_t*)dev, MIO_DEV_MARIA_FETCH_ROW, MIO_NULL); | ||||||
|  | } | ||||||
							
								
								
									
										203
									
								
								mio/lib/mio-maria.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										203
									
								
								mio/lib/mio-maria.h
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,203 @@ | |||||||
|  | /* | ||||||
|  |  * $Id$ | ||||||
|  |  * | ||||||
|  |     Copyright (c) 2016-2020 Chung, Hyung-Hwan. All rights reserved. | ||||||
|  |  | ||||||
|  |     Redistribution and use in source and binary forms, with or without | ||||||
|  |     modification, are permitted mariavided that the following conditions | ||||||
|  |     are met: | ||||||
|  |     1. Redistributions of source code must retain the above copyright | ||||||
|  |        notice, this list of conditions and the following disclaimer. | ||||||
|  |     2. Redistributions in binary form must remariaduce the above copyright | ||||||
|  |        notice, this list of conditions and the following disclaimer in the | ||||||
|  |        documentation and/or other materials mariavided with the distribution. | ||||||
|  |  | ||||||
|  |     THIS SOFTWARE IS PIPEVIDED BY THE AUTHOR "AS IS" AND ANY EXPRESS OR | ||||||
|  |     IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WAfRRANTIES | ||||||
|  |     OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. | ||||||
|  |     IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, | ||||||
|  |     INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT | ||||||
|  |     NOT LIMITED TO, PIPECUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | ||||||
|  |     DATA, OR PIPEFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | ||||||
|  |     THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | ||||||
|  |     (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF | ||||||
|  |     THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||||||
|  |  */ | ||||||
|  |  | ||||||
|  | #ifndef _MIO_PIPE_H_ | ||||||
|  | #define _MIO_PIPE_H_ | ||||||
|  |  | ||||||
|  | #include <mio.h> | ||||||
|  |  | ||||||
|  | typedef struct mio_dev_maria_t mio_dev_maria_t; | ||||||
|  |  | ||||||
|  | enum mio_dev_maria_state_t | ||||||
|  | { | ||||||
|  | 	/* the following items(progress bits) are mutually exclusive */ | ||||||
|  | 	MIO_DEV_MARIA_CONNECTING      = (1 << 0), | ||||||
|  | 	MIO_DEV_MARIA_CONNECTED       = (1 << 1), | ||||||
|  | 	MIO_DEV_MARIA_QUERY_STARTING  = (1 << 2), | ||||||
|  | 	MIO_DEV_MARIA_QUERY_STARTED   = (1 << 3), | ||||||
|  | 	MIO_DEV_MARIA_ROW_FETCHING    = (1 << 4), | ||||||
|  | 	MIO_DEV_MARIA_ROW_FETCHED     = (1 << 5), | ||||||
|  |  | ||||||
|  |  | ||||||
|  | #if 0 | ||||||
|  | 	/* the following items can be bitwise-ORed with an exclusive item above */ | ||||||
|  | 	MIO_DEV_MARIA_LENIENT        = (1 << 14), | ||||||
|  | 	MIO_DEV_MARIA_INTERCEPTED    = (1 << 15), | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | 	/* convenience bit masks */ | ||||||
|  | 	MIO_DEV_MARIA_ALL_PROGRESS_BITS = (MIO_DEV_MARIA_CONNECTING | | ||||||
|  | 	                                   MIO_DEV_MARIA_CONNECTED | | ||||||
|  | 	                                   MIO_DEV_MARIA_QUERY_STARTING | | ||||||
|  | 	                                   MIO_DEV_MARIA_QUERY_STARTED | | ||||||
|  | 	                                   MIO_DEV_MARIA_ROW_FETCHING | | ||||||
|  | 	                                   MIO_DEV_MARIA_ROW_FETCHED) | ||||||
|  | }; | ||||||
|  | typedef enum mio_dev_maria_state_t mio_dev_maria_state_t; | ||||||
|  |  | ||||||
|  | #define MIO_DEV_MARIA_SET_PROGRESS(dev,bit) do { \ | ||||||
|  | 	(dev)->state &= ~MIO_DEV_MARIA_ALL_PROGRESS_BITS; \ | ||||||
|  | 	(dev)->state |= (bit); \ | ||||||
|  | } while(0) | ||||||
|  |  | ||||||
|  | #define MIO_DEV_MARIA_GET_PROGRESS(dev) ((dev)->state & MIO_DEV_MARIA_ALL_PROGRESS_BITS) | ||||||
|  |  | ||||||
|  |  | ||||||
|  | typedef int (*mio_dev_maria_on_read_t) ( | ||||||
|  | 	mio_dev_maria_t*    dev, | ||||||
|  | 	const void*         data, | ||||||
|  | 	mio_iolen_t         len | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | typedef int (*mio_dev_maria_on_write_t) ( | ||||||
|  | 	mio_dev_maria_t*    dev, | ||||||
|  | 	mio_iolen_t         wrlen, | ||||||
|  | 	void*               wrctx | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | typedef void (*mio_dev_maria_on_connect_t) ( | ||||||
|  | 	mio_dev_maria_t*    dev | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | typedef void (*mio_dev_maria_on_disconnect_t) ( | ||||||
|  | 	mio_dev_maria_t*    dev | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | typedef void (*mio_dev_maria_on_query_started_t) ( | ||||||
|  | 	mio_dev_maria_t*    dev, | ||||||
|  | 	int                 maria_ret | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | typedef void (*mio_dev_maria_on_row_fetched_t) ( | ||||||
|  | 	mio_dev_maria_t*    dev, | ||||||
|  | 	void*               row_data | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | struct mio_dev_maria_t | ||||||
|  | { | ||||||
|  | 	MIO_DEV_HEADER; | ||||||
|  |  | ||||||
|  | 	void* hnd; | ||||||
|  | 	void* res; | ||||||
|  | 	int state; | ||||||
|  |  | ||||||
|  | 	unsigned int connected; | ||||||
|  | 	unsigned int query_started; | ||||||
|  | 	unsigned int row_fetched; | ||||||
|  |  | ||||||
|  | 	int query_ret; | ||||||
|  | 	int row_wstatus; | ||||||
|  | 	void* row; | ||||||
|  |  | ||||||
|  | 	mio_dev_maria_on_read_t on_read; | ||||||
|  | 	mio_dev_maria_on_write_t on_write; | ||||||
|  | 	mio_dev_maria_on_connect_t on_connect; | ||||||
|  | 	mio_dev_maria_on_disconnect_t on_disconnect; | ||||||
|  | 	mio_dev_maria_on_query_started_t on_query_started; | ||||||
|  | 	mio_dev_maria_on_row_fetched_t on_row_fetched; | ||||||
|  | }; | ||||||
|  |  | ||||||
|  | typedef struct mio_dev_maria_make_t mio_dev_maria_make_t; | ||||||
|  | struct mio_dev_maria_make_t | ||||||
|  | { | ||||||
|  | 	mio_dev_maria_on_write_t on_write; /* mandatory */ | ||||||
|  | 	mio_dev_maria_on_read_t on_read; /* mandatory */ | ||||||
|  | 	mio_dev_maria_on_connect_t on_connect; /* optional */ | ||||||
|  | 	mio_dev_maria_on_disconnect_t on_disconnect; /* optional */ | ||||||
|  | 	mio_dev_maria_on_query_started_t on_query_started; | ||||||
|  | 	mio_dev_maria_on_row_fetched_t on_row_fetched; | ||||||
|  | }; | ||||||
|  |  | ||||||
|  | typedef struct mio_dev_maria_connect_t mio_dev_maria_connect_t; | ||||||
|  | struct mio_dev_maria_connect_t | ||||||
|  | { | ||||||
|  | 	const mio_bch_t* host; | ||||||
|  | 	const mio_bch_t* username; | ||||||
|  | 	const mio_bch_t* password; | ||||||
|  | 	const mio_bch_t* dbname; | ||||||
|  | 	mio_uint16_t port; | ||||||
|  | }; | ||||||
|  |  | ||||||
|  | enum mio_dev_maria_ioctl_cmd_t | ||||||
|  | { | ||||||
|  | 	MIO_DEV_MARIA_CONNECT, | ||||||
|  | 	MIO_DEV_MARIA_QUERY_WITH_BCS, | ||||||
|  | 	MIO_DEV_MARIA_FETCH_ROW | ||||||
|  | }; | ||||||
|  | typedef enum mio_dev_maria_ioctl_cmd_t mio_dev_maria_ioctl_cmd_t; | ||||||
|  |  | ||||||
|  | #ifdef __cplusplus | ||||||
|  | extern "C" { | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | MIO_EXPORT  mio_dev_maria_t* mio_dev_maria_make ( | ||||||
|  | 	mio_t*                      mio, | ||||||
|  | 	mio_oow_t                   xtnsize, | ||||||
|  | 	const mio_dev_maria_make_t* data | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | #if defined(MIO_HAVE_INLINE) | ||||||
|  | static MIO_INLINE mio_t* mio_dev_maria_getmio (mio_dev_maria_t* maria) { return mio_dev_getmio((mio_dev_t*)maria); } | ||||||
|  | #else | ||||||
|  | #	define mio_dev_maria_getmio(maria) mio_dev_getmio(maria) | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | #if defined(MIO_HAVE_INLINE) | ||||||
|  | static MIO_INLINE void* mio_dev_maria_getxtn (mio_dev_maria_t* maria) { return (void*)(maria + 1); } | ||||||
|  | #else | ||||||
|  | #	define mio_dev_maria_getxtn(maria) ((void*)(((mio_dev_maria_t*)maria) + 1)) | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |  | ||||||
|  | MIO_EXPORT int mio_dev_maria_connect ( | ||||||
|  | 	mio_dev_maria_t*     maria, | ||||||
|  | 	mio_dev_maria_connect_t* ci | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | MIO_EXPORT int mio_dev_maria_querywithbchars ( | ||||||
|  | 	mio_dev_maria_t*     maria, | ||||||
|  | 	const mio_bch_t*     qstr, | ||||||
|  | 	mio_oow_t            qlen | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | MIO_EXPORT int mio_dev_maria_fetchrows ( | ||||||
|  | 	mio_dev_maria_t*     maria | ||||||
|  | ); | ||||||
|  |  | ||||||
|  | #if defined(MIO_HAVE_INLINE) | ||||||
|  | static MIO_INLINE void mio_dev_maria_kill (mio_dev_maria_t* maria) { mio_dev_kill ((mio_dev_t*)maria); } | ||||||
|  | static MIO_INLINE void mio_dev_maria_halt (mio_dev_maria_t* maria) { mio_dev_halt ((mio_dev_t*)maria); } | ||||||
|  | #else | ||||||
|  | #	define mio_dev_maria_kill(maria) mio_dev_kill((mio_dev_t*)maria) | ||||||
|  | #	define mio_dev_maria_halt(maria) mio_dev_halt((mio_dev_t*)maria) | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  |  | ||||||
|  | #ifdef __cplusplus | ||||||
|  | } | ||||||
|  | #endif | ||||||
|  |  | ||||||
|  | #endif | ||||||
		Reference in New Issue
	
	Block a user