← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/message/table_writer.cc

  • Committer: Brian Aker
  • Date: 2010-07-09 21:43:25 UTC
  • mfrom: (1643.5.1 dr-bug-600624)
  • Revision ID: brian@gaz-20100709214325-4rllc5yyo6bku5sh
Merge Prafulla Tekawade

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/message/table_writer.cc
1
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
3
 *
4
 
 *  Copyright (C) 2009 Sun Microsystems, Inc.
 
4
 *  Copyright (C) 2009 Sun Microsystems
5
5
 *
6
6
 *  This program is free software; you can redistribute it and/or modify
7
7
 *  it under the terms of the GNU General Public License as published by
22
22
#include <iostream>
23
23
#include <fstream>
24
24
#include <string>
 
25
#include <getopt.h>
25
26
#include <drizzled/message/table.pb.h>
26
27
 
27
 
#include <boost/program_options.hpp>
28
 
 
29
28
using namespace std;
30
29
using namespace drizzled;
31
30
 
32
 
namespace po=boost::program_options;
33
 
 
34
31
/*
35
32
  Written from Google proto example
36
33
*/
125
122
    field->set_name(buffer);
126
123
    field->set_type(message::Table::Field::VARCHAR);
127
124
 
128
 
    field_constraints->set_is_notnull((x % 2));
 
125
    field_constraints->set_is_nullable((x % 2));
129
126
 
130
127
    string_field_options->set_length(rand() % 100);
131
128
 
166
163
    field->set_type(message::Table::Field::DECIMAL);
167
164
 
168
165
    field_constraints= field->mutable_constraints();
169
 
    field_constraints->set_is_notnull(false);
 
166
    field_constraints->set_is_nullable(true);
170
167
 
171
168
    numeric_field_options= field->mutable_numeric_options();
172
169
    numeric_field_options->set_precision(8);
203
200
  message::Table::TableOptions *tableopts;
204
201
 
205
202
  table->set_name("t1");
206
 
  table->set_catalog("LOCAL");
207
203
  table->set_type(message::Table::INTERNAL);
208
204
 
209
205
  tableopts= table->mutable_options();
217
213
 
218
214
}
219
215
 
 
216
static void usage(char *argv0)
 
217
{
 
218
  cerr << "Usage:  " << argv0 << " [-t N] TABLE_NAME.dfe" << endl;
 
219
  cerr << endl;
 
220
  cerr << "-t N\tTable Number" << endl;
 
221
  cerr << "\t0 - default" << endl;
 
222
  cerr << endl;
 
223
}
220
224
 
221
225
int main(int argc, char* argv[])
222
226
{
 
227
  int opt;
223
228
  int table_number= 0;
224
229
 
225
230
  GOOGLE_PROTOBUF_VERIFY_VERSION;
226
231
 
227
 
  po::options_description desc("Allowed options");
228
 
  desc.add_options()
229
 
    ("help", "produce help message")
230
 
    ("table-number,t", po::value<int>(&table_number)->default_value(0), "Table Number");
231
 
 
232
 
  po::variables_map vm;
233
 
  po::positional_options_description p;
234
 
  p.add("table-name", 1);
235
 
 
236
 
  // Disable allow_guessing
237
 
  int style = po::command_line_style::default_style & ~po::command_line_style::allow_guessing;
238
 
 
239
 
  po::store(po::command_line_parser(argc, argv).options(desc).style(style).
240
 
            positional(p).run(), vm);
241
 
 
242
 
  if (not vm.count("table-name"))
 
232
  while ((opt= getopt(argc, argv, "t:")) != -1)
243
233
  {
 
234
    switch (opt)
 
235
    {
 
236
    case 't':
 
237
      table_number= atoi(optarg);
 
238
      break;
 
239
    default:
 
240
      usage(argv[0]);
 
241
      exit(EXIT_FAILURE);
 
242
    }
 
243
  }
 
244
 
 
245
  if (optind >= argc) {
244
246
    fprintf(stderr, "Expected Table name argument\n\n");
245
 
    cerr << desc << endl;
 
247
    usage(argv[0]);
246
248
    exit(EXIT_FAILURE);
247
249
  }
248
250
 
258
260
    break;
259
261
  default:
260
262
    fprintf(stderr, "Invalid table number.\n\n");
261
 
    cerr << desc << endl;
 
263
    usage(argv[0]);
262
264
    exit(EXIT_FAILURE);
263
265
  }
264
266
 
265
 
  fstream output(vm["table-name"].as<string>().c_str(),
266
 
                 ios::out | ios::trunc | ios::binary);
267
 
  if (not table.SerializeToOstream(&output))
 
267
  fstream output(argv[optind], ios::out | ios::trunc | ios::binary);
 
268
  if (!table.SerializeToOstream(&output))
268
269
  {
269
270
    cerr << "Failed to write schema." << endl;
270
271
    return -1;