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);
int tdbGetFileSize(tdb_fd_t fd, int szPage, SPgno *size);

#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;
  int t;

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

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

  ASSERT(n < 6);

  *v = tv;
  return n;
}

static inline u32 tdbCstringHash(const char *s) { return MurmurHash3_32(s, strlen(s)); }

#ifdef __cplusplus
}
#endif

#endif /*_TDB_UTIL_H_*/