libdb/test/xa/src2/bdb2.c

158 lines
3.4 KiB
C

/*
* Copyright (c) 1997 BEA Systems, Inc.
* All Rights Reserved
*
* THIS IS UNPUBLISHED PROPRIETARY SOURCE CODE OF
* BEA Systems, Inc.
* The copyright notice above does not evidence any
* actual or intended publication of such source code.
*
* This server is only called by bdb1. It takes the data sent by bdb1 and
* inserts it into db2.
*/
#include <db.h>
#include <xa.h>
#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <atmi.h> /* TUXEDO Header File */
#include <userlog.h> /* TUXEDO Header File */
#include <string.h>
#include <time.h>
#include <tpadm.h>
#define DATABASE1 "data1.db"
#define DATABASE2 "data2.db"
static DB *dbp1;
static DB *dbp2;
static int opendb(){
int ret;
u_int32_t open_flags = DB_CREATE | DB_AUTO_COMMIT;
/* Create and initialize database object, open the database. */
if ((ret = db_create(&dbp1, NULL, DB_XA_CREATE)) != 0) {
userlog("db_create: %s", db_strerror(ret));
return (EXIT_FAILURE);
}
if ((ret = db_create(&dbp2, NULL, DB_XA_CREATE)) != 0) {
userlog("db_create: %s", db_strerror(ret));
return (EXIT_FAILURE);
}
if ((ret = dbp1->open(dbp1, NULL, DATABASE1, NULL, DB_BTREE, open_flags,
0664)) != 0) {
userlog("open: %s", db_strerror(ret));
return (EXIT_FAILURE);
}
if ((ret = dbp2->open(dbp2, NULL, DATABASE2, NULL, DB_BTREE, open_flags,
0664)) != 0) {
userlog("open: %s", db_strerror(ret));
return (EXIT_FAILURE);
}
return 0;
}
static void closedb(){
(void)dbp1->close(dbp1, 0);
(void)dbp2->close(dbp2, 0);
}
/* Write the given data into the given database. */
static int writedb(DB * dbp, void *buf, u_int32_t size){
DBT key, data;
size_t len;
int ch, ret;
memset(&key, 0, sizeof(DBT));
memset(&data, 0, sizeof(DBT));
key.data = buf;
data.data = buf;
data.size = key.size = size;
ret = dbp->put(dbp, NULL, &key, &data, DB_NOOVERWRITE);
switch (ret) {
case 0:
return (EXIT_SUCCESS);
case DB_KEYEXIST:
return (EXIT_SUCCESS);
case DB_LOCK_DEADLOCK:
return (EXIT_SUCCESS);
default:
userlog("put: %s", db_strerror(ret));
return -1;
}
}
/* Open the databases used by this server when it is started. */
int
tpsvrinit(argc, argv)
int argc;
char **argv;
{
int ret;
/* Some compilers warn if argc and argv aren't used. */
argc = argc;
argv = argv;
tpopen();
if (ret = opendb() != 0){
userlog("put: %s", db_strerror(ret));
}
return(0);
}
/* Close the database when the server is shutdown. */
void
tpsvrdone(void)
{
closedb();
tpclose();
}
/*
* Get the data the calling server just inserted into db1 and insert it into
* db2. Fgets32 is used to get the key passed by the calling server.
*/
int
WRITE2(rqst)
TPSVCINFO *rqst;
{
char buf[100];
DBT key, data;
FBFR32 *reqbuf = (FBFR32*)rqst->data;
int ret;
Fgets32(reqbuf, TA_REPOSPARAM, 0, buf);
userlog("buf:[%s]", buf);
memset(&key, 0, sizeof(DBT));
memset(&data, 0, sizeof(DBT));
key.data = buf;
key.size = (u_int32_t)strlen(buf);
data.flags = DB_DBT_MALLOC;
/* Get the data that the calling server inserted into db1 */
switch (ret = dbp1->get(dbp1, NULL, &key, &data, DB_READ_UNCOMMITTED)){
case 0:
break;
case DB_LOCK_DEADLOCK:
tpreturn(TPSUCCESS, 1L, rqst->data, 0L, 0);
default:
userlog("get: %s", db_strerror(ret));
tpreturn(TPFAIL, 0L, rqst->data, 0L, 0);
}
/* Write the data to db2 */
if(writedb(dbp2, data.data, data.size) != 0){
tpreturn(TPSUCCESS, 1L, rqst->data, 0L, 0);
}
tpreturn(TPSUCCESS, 0, rqst->data, 0L, 0);
}