summaryrefslogtreecommitdiff
path: root/cpp/src/IceStorm/Server.cpp
blob: b5ff66fde5bd27816bd981564209c80db3430c1f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
// **********************************************************************
//
// Copyright (c) 2001
// MutableRealms, Inc.
// Huntsville, AL, USA
//
// All Rights Reserved
//
// **********************************************************************

#include <Freeze/Application.h>
#include <IceStorm/TopicManagerI.h>
#include <IceStorm/TraceLevels.h>

using namespace std;
using namespace Ice;
using namespace IceStorm;

namespace IceStorm
{

class Server : public Freeze::Application
{
public:

    Server(const string& dbEnvName) :
	Freeze::Application(dbEnvName)
    {
    }

    void usage();
    virtual int runFreeze(int, char*[], const Freeze::DBEnvironmentPtr&);
};

} // End namespace IceStorm

//
// TODO: get rid of this when the prescanning of the argument list is
// not needed.
//
static void
usage(const char* appName)
{
    cerr << "Usage: " << appName << " [options]\n";
    cerr <<	
	"Options:\n"
	"-h, --help           Show this message.\n"
	"-v, --version        Display the Ice version.\n"
        "--dbdir directory    Set the database location.\n"
	;
}

int
main(int argc, char* argv[])
{
    //
    // TODO: This is a hack.
    //
    // Prescan the argument list for --dbdir. Process & consume if
    // found.
    //
    string dbdir = "db";
    int i = 1;
    while (i < argc)
    {
	if (strcmp(argv[i], "--dbdir") == 0)
	{
	    if (i +1 >= argc)
	    {
		usage(argv[0]);
		return EXIT_FAILURE;
	    }
	    dbdir = argv[i + 1];
	    //
	    // Consume arguments
	    //
	    while (i < argc - 2)
	    {
		argv[i] = argv[i + 2];
		++i;
	    }
	    argc -= 2;
	}
	else
	{
	    ++i;
	}
    }
    
    addArgumentPrefix("IcePack");
    Server app(dbdir);
    return app.main(argc, argv);
}

void
IceStorm::Server::usage()
{
    //
    // TODO: This is only needed because of the forced prescan of the
    // arguments.
    //
    ::usage(appName());
}

int
IceStorm::Server::runFreeze(int argc, char* argv[], const Freeze::DBEnvironmentPtr& dbEnv)
{
    for (int i = 1; i < argc; ++i)
    {
	if (strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--help") == 0)
	{
	    usage();
	    return EXIT_SUCCESS;
	}
	else if (strcmp(argv[i], "-v") == 0 || strcmp(argv[i], "--version") == 0)
	{
	    cout << ICE_STRING_VERSION << endl;
	    return EXIT_SUCCESS;
	}
	else
	{
	    cerr << appName() << ": unknown option `" << argv[i] << "'" << endl;
	    usage();
	    return EXIT_FAILURE;
	}
    }

    Freeze::DBPtr dbTopicManager = dbEnv->openDB("topicmanager");

    //PropertiesPtr properties = communicator()->getProperties();

    TraceLevelsPtr traceLevels = new TraceLevels(communicator()->getProperties());
    ObjectAdapterPtr adapter = communicator()->createObjectAdapterFromProperty("TopicManager",
									       "IceStorm.TopicManager.Endpoints");
    ObjectPtr object = new TopicManagerI(communicator(), adapter, traceLevels, dbTopicManager);
    adapter->add(object, stringToIdentity("TopicManager"));
    adapter->activate();

    shutdownOnInterrupt();
    communicator()->waitForShutdown();
    ignoreInterrupt();

    // TODO: topic manager ::reap()

    return EXIT_SUCCESS;
}