atmitest/test059_pq/atmiclt59.c

0001 /**
0002  * @brief Test print queue ops - client
0003  *
0004  * @file atmiclt59.c
0005  */
0006 /* -----------------------------------------------------------------------------
0007  * Enduro/X Middleware Platform for Distributed Transaction Processing
0008  * Copyright (C) 2009-2016, ATR Baltic, Ltd. All Rights Reserved.
0009  * Copyright (C) 2017-2023, Mavimax, Ltd. All Rights Reserved.
0010  * This software is released under one of the following licenses:
0011  * AGPL (with Java and Go exceptions) or Mavimax's license for commercial use.
0012  * See LICENSE file for full text.
0013  * -----------------------------------------------------------------------------
0014  * AGPL license:
0015  *
0016  * This program is free software; you can redistribute it and/or modify it under
0017  * the terms of the GNU Affero General Public License, version 3 as published
0018  * by the Free Software Foundation;
0019  *
0020  * This program is distributed in the hope that it will be useful, but WITHOUT ANY
0021  * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
0022  * PARTICULAR PURPOSE. See the GNU Affero General Public License, version 3
0023  * for more details.
0024  *
0025  * You should have received a copy of the GNU Affero General Public License along
0026  * with this program; if not, write to the Free Software Foundation, Inc.,
0027  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
0028  *
0029  * -----------------------------------------------------------------------------
0030  * A commercial use license is available from Mavimax, Ltd
0031  * contact@mavimax.com
0032  * -----------------------------------------------------------------------------
0033  */
0034 #include <string.h>
0035 #include <stdio.h>
0036 #include <stdlib.h>
0037 #include <memory.h>
0038 #include <math.h>
0039
0040 #include <atmi.h>
0041 #include <ubf.h>
0042 #include <ndebug.h>
0043 #include <test.fd.h>
0044 #include <ndrstandard.h>
0045 #include <nstopwatch.h>
0046 #include <fcntl.h>
0047 #include <unistd.h>
0048 #include <nstdutil.h>
0049 #include <sys_mqueue.h>
0050 #include "test59.h"
0051 /*---------------------------Externs------------------------------------*/
0052 /*---------------------------Macros-------------------------------------*/
0053 /*---------------------------Enums--------------------------------------*/
0054 /*---------------------------Typedefs-----------------------------------*/
0055 /*---------------------------Globals------------------------------------*/
0056 /*---------------------------Statics------------------------------------*/
0057 /*---------------------------Prototypes---------------------------------*/
0058
0059 /**
0060  * Do the test call to the server
0061  */
0062 int main(int argc, char** argv)
0063 {
0064
0065     UBFH *p_ub = (UBFH *)tpalloc("UBF", NULL, 56000);
0066     int ret=EXSUCCEED;
0067     mqd_t mq = (mqd_t)EXFAIL;
0068     int i;
0069     int num;
0070
0071     if (argc < 4)
0072     {
0073         NDRX_LOG(log_error, "%s <command 's' - call service | 'q' - make q> <name> "
0074                 "<nummsg>\n");
0075         EXFAIL_OUT(ret);
0076     }
0077
0078     num = atoi(argv[3]);
0079     if ('s' == argv[1][0])
0080     {
0081         if (EXFAIL==CBchg(p_ub, T_STRING_FLD, 0, VALUE_EXPECTED, 0, BFLD_STRING))
0082         {
0083             NDRX_LOG(log_debug, "Failed to set T_STRING_FLD[0]: %s", Bstrerror(Berror));
0084             ret=EXFAIL;
0085             goto out;
0086         }
0087
0088         /* put some message in target service */
0089         for (i=0; i<num;i++)
0090         {
0091             if (EXFAIL == tpacall(argv[2], (char *)p_ub, 0L, 0))
0092             {
0093                 NDRX_LOG(log_error, "%s failed: %s", argv[2], tpstrerror(tperrno));
0094                 ret=EXFAIL;
0095                 goto out;
0096             }
0097         }
0098     }
0099     else if ('q' == argv[1][0])
0100     {
0101         /* create some arbitrary queue and enqueue some data to it */
0102         struct mq_attr attr;
0103         char qstr[PATH_MAX];
0104     char buf[128];
0105         /* initialize the queue attributes */
0106         attr.mq_flags = 0;
0107         attr.mq_maxmsg = 10;
0108         attr.mq_msgsize = 128;
0109         attr.mq_curmsgs = 0;
0110
0111         snprintf(qstr, sizeof(qstr), "%s", argv[2]);
0112
0113         /* create the message queue */
0114         if ((mqd_t)EXFAIL==(mq = ndrx_mq_open(qstr, O_CREAT | O_RDWR, 0644, &attr)))
0115         {
0116             NDRX_LOG(log_error, "Failed to open queue: [%s]: %s",
0117                     qstr, strerror(errno));
0118             EXFAIL_OUT(ret);
0119         }
0120
0121         for (i=0; i<num;i++)
0122         {
0123             if (EXSUCCEED!=ndrx_mq_send(mq, buf, sizeof(buf), 0))
0124             {
0125                 NDRX_LOG(log_error, "Failed to send msg to [%s]: %s",
0126                         qstr, strerror(errno));
0127                 EXFAIL_OUT(ret);
0128             }
0129         }
0130     }
0131     else
0132     {
0133         NDRX_LOG(log_error, "Unknown command [%s]", argv[1]);
0134         EXFAIL_OUT(ret);
0135     }
0136
0137 out:
0138
0139     if ((mqd_t)EXFAIL!=mq)
0140     {
0141         ndrx_mq_close(mq);
0142     }
0143
0144     tpterm();
0145     fprintf(stderr, "Exit with %d\n", ret);
0146
0147     return ret;
0148 }
0149
0150 /* vim: set ts=4 sw=4 et smartindent: */