awtk/tests/fscript_run.cpp

#include "tkc/fs.h"
#include "tkc/mem.h"
#include "tkc/async.h"
#include "tkc/utils.h"
#include "tkc/platform.h"
#include "tkc/time_now.h"
#include "tkc/fscript.h"
#include "tkc/socket_helper.h"
#include "tkc/object_default.h"
#include "fscript_ext/fscript_ext.h"
#include "conf_io/app_conf_init_json.h"
#include "tkc/data_reader_factory.h"
#include "tkc/data_writer_factory.h"
#include "tkc/data_writer_file.h"
#include "tkc/data_writer_wbuffer.h"
#include "tkc/data_reader_file.h"
#include "tkc/data_reader_mem.h"
#include "debugger/debugger_global.h"
#include "debugger/debugger_server_tcp.h"

static ret_t run_fscript(const char* code, uint32_t times, bool_t debug) {
  value_t v;
  char buff[64];
  uint64_t start = time_now_us();
  tk_object_t* obj = object_default_create();
  tk_mem_dump();
  log_debug("======================\n");
  if (times > 1 && !debug) {
    /*stress test*/
    uint32_t i = 0;
    fscript_t* fscript = fscript_create(obj, code);
    for (i = 0; i < times; i++) {
      log_debug("%u times....\n", i);
      tk_mem_dump();
      fscript_exec(fscript, &v);
      tk_mem_dump();
      value_reset(&v);
    }
    fscript_destroy(fscript);
  } else if (debug) {
    str_t str;
    fscript_t* fscript = NULL;

    debugger_global_init();
    debugger_server_tcp_init(DEBUGGER_TCP_PORT);
    debugger_server_tcp_start();
    debugger_server_set_single_mode(TRUE);
    sleep_ms(1000);
    str_init(&str, 1000);
    str_set(&str, code);
    str_append_format(&str, 100, "//code_id(\"%s\")\n", DEBUGGER_DEFAULT_CODE_ID);
    code = str.str;

    fscript = fscript_create(obj, code);
    fscript_exec(fscript, &v);
    fscript_destroy(fscript);

    str_reset(&str);
    debugger_server_tcp_deinit();
    debugger_global_deinit();

    log_debug("result:%s\n", value_str_ex(&v, buff, sizeof(buff) - 1));
    value_reset(&v);
  } else {
    fscript_eval(obj, code, &v);
    value_reset(&v);
  }
  TK_OBJECT_UNREF(obj);
  log_debug("cost: %d us\n", (int)(time_now_us() - start));

  return RET_OK;
}

static ret_t run_fscript_file(const char* filename, uint32_t times, bool_t debug) {
  uint32_t size = 0;
  char* code = (char*)file_read(filename, &size);
  return_value_if_fail(code != NULL, RET_BAD_PARAMS);
  run_fscript(code, times, debug);
  TKMEM_FREE(code);

  return RET_OK;
}

int main(int argc, char* argv[]) {
  platform_prepare();
  tk_socket_init();
  tk_mem_dump();
  fscript_global_init();
  fscript_ext_init();
  tk_mem_dump();
  async_call_init();
  data_writer_factory_set(data_writer_factory_create());
  data_reader_factory_set(data_reader_factory_create());
  data_writer_factory_register(data_writer_factory(), "file", data_writer_file_create);
  data_reader_factory_register(data_reader_factory(), "file", data_reader_file_create);
  data_reader_factory_register(data_reader_factory(), "mem", data_reader_mem_create);
  data_writer_factory_register(data_writer_factory(), "wbuffer", data_writer_wbuffer_create);
  tk_mem_dump();

  app_conf_init_json("runFScript");
  tk_mem_dump();
  if (argc < 2) {
    printf("Usage: %s script [debug|times]\n", argv[0]);
    printf("Usage: %s @filename [debug|times\n", argv[0]);
    return 0;
  } else {
    const char* code = argv[1];
    uint32_t times = argc > 2 ? tk_atoi(argv[2]) : 1;
    bool_t debug = argc > 2 ? tk_str_eq(argv[2], "debug") : FALSE;

    if (*code == '@') {
      run_fscript_file(code + 1, times, debug);
    } else {
      run_fscript(code, times, debug);
    }
  }
  tk_mem_dump();
  app_conf_set_int("foobar", 1);
  app_conf_save();
  data_writer_factory_destroy(data_writer_factory());
  data_reader_factory_destroy(data_reader_factory());
  data_writer_factory_set(NULL);
  data_reader_factory_set(NULL);
  fscript_global_deinit();
  tk_socket_deinit();
  async_call_deinit();

  return 0;
}