tdbUtil.h

/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef _TDB_UTIL_H_
#define _TDB_UTIL_H_

#ifdef __cplusplus
extern "C" {
#endif

#if __STDC_VERSION__ >= 201112L
#define TDB_STATIC_ASSERT(op, info) static_assert(op, info)
#else
#define TDB_STATIC_ASSERT(op, info)
#endif

#define TDB_ROUND8(x) (((x) + 7) & ~7)

int tdbGnrtFileID(const char *fname, uint8_t *fileid, bool unique);

// #define TDB_F_OK 0x1
// #define TDB_R_OK 0x2
// #define TDB_W_OK 0x4
// int tdbCheckFileAccess(const char *pathname, int mode);

int tdbGetFileSize(const char *fname, int pgSize, SPgno *pSize);

int tdbPRead(int fd, void *pData, int count, i64 offset);
int tdbWrite(int fd, void *pData, int count);

#define TDB_REALLOC(PTR, SIZE)                                                               \
  ({                                                                                         \
    void *nPtr;                                                                              \
    if ((PTR) == NULL || ((int *)(PTR))[-1] < (SIZE)) {                                      \
      nPtr = tdbOsRealloc((PTR) ? (char *)(PTR) - sizeof(int) : NULL, (SIZE) + sizeof(int)); \
      if (nPtr) {                                                                            \
        ((int *)nPtr)[0] = (SIZE);                                                           \
        nPtr = (char *)nPtr + sizeof(int);                                                   \
      }                                                                                      \
    } else {                                                                                 \
      nPtr = (PTR);                                                                          \
    }                                                                                        \
    nPtr;                                                                                    \
  })

#define TDB_FREE(PTR)                    \
  do {                                   \
    if (PTR) {                           \
      tdbOsFree((char *)(PTR) - sizeof(int)); \
    }                                    \
  } while (0)

static inline void *tdbDefaultMalloc(void *arg, size_t size) {
  void *ptr;

  ptr = tdbOsMalloc(size);

  return ptr;
}

static inline void tdbDefaultFree(void *arg, void *ptr) { tdbOsFree(ptr); }

static inline int tdbPutVarInt(u8 *p, int v) {
  int n = 0;

  for (;;) {
    if (v <= 0x7f) {
      p[n++] = v;
      break;
    }

    p[n++] = (v & 0x7f) | 0x80;
    v >>= 7;
  }

  ASSERT(n < 6);

  return n;
}

static inline int tdbGetVarInt(const u8 *p, int *v) {
  int n = 0;
  int tv = 0;

  for (;;) {
    if (p[n] <= 0x7f) {
      tv = (tv << 7) | p[n];
      n++;
      break;
    }

    tv = (tv << 7) | (p[n] & 0x7f);
    n++;
  }

  ASSERT(n < 6);

  *v = tv;
  return n;
}

#ifdef __cplusplus
}
#endif

#endif /*_TDB_UTIL_H_*/