#5017

Committed 09 Apr 2026 02:37PM UTC coverage: 72.248% (-0.05%) from 72.299%

Build # #5017

Build Type

push

travis-ci

Committed by

web-flow

Commit Message

merge: from main to 3.0 branch #35095 

merge: from main to 3.0 branch

Coverage Stats

499 of 655 new or added lines in 34 files covered. (76.18%)

821 existing lines in 156 files now uncovered.

257359 of 356215 relevant lines covered (72.25%)

132044878.66 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

79.25

/source/util/src/tarray.c

/*
 * 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/>.
 */

#define _DEFAULT_SOURCE
#include "tarray.h"
#include "tcoding.h"

// todo refactor API
#define BOUNDARY_SIZE         1024 * 1024 * 1024  // 1G
#define BOUNDARY_SMALL_FACTOR 1.2
#define BOUNDARY_BIG_FACTOR   2

SArray* taosArrayInit(size_t size, size_t elemSize) {
  if (elemSize == 0) {
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }

  if (size < TARRAY_MIN_SIZE) {
    size = TARRAY_MIN_SIZE;
  }

  SArray* pArray = taosMemoryMalloc(sizeof(SArray));
  if (pArray == NULL) {
    return NULL;
  }

  pArray->size = 0;
  pArray->pData = taosMemoryCalloc(size, elemSize);
  if (pArray->pData == NULL) {
    taosMemoryFree(pArray);
    return NULL;
  }

  pArray->capacity = size;
  pArray->elemSize = elemSize;
  return pArray;
}

SArray* taosArrayInit_s(size_t elemSize, size_t initialSize) {
  SArray* pArray = taosMemoryMalloc(sizeof(SArray));
  if (pArray == NULL) {
    return NULL;
  }

  pArray->size = initialSize;
  pArray->pData = taosMemoryCalloc(initialSize, elemSize);
  if (pArray->pData == NULL) {
    taosMemoryFree(pArray);
    return NULL;
  }

  pArray->capacity = initialSize;
  pArray->elemSize = elemSize;
  return pArray;
}

static int32_t taosArrayResize(SArray* pArray) {
  size_t size = pArray->capacity;
  size = (size << 1u);

  void* tmp = taosMemoryRealloc(pArray->pData, size * pArray->elemSize);
  if (tmp == NULL) {  // reallocate failed, the original buffer remains
    return terrno;
  }

  pArray->pData = tmp;
  pArray->capacity = size;

  return 0;
}

int32_t taosArrayEnsureCap(SArray* pArray, size_t newCap) {
  if (newCap > pArray->capacity) {
    float factor = BOUNDARY_BIG_FACTOR;
    if (newCap * pArray->elemSize > BOUNDARY_SIZE) {
      factor = BOUNDARY_SMALL_FACTOR;
    }

    size_t tsize = (pArray->capacity * factor);
    while (newCap > tsize) {
      size_t newSize = (tsize * factor);
      tsize = (newSize == tsize) ? (tsize + 2) : newSize;
    }

    char* p = taosMemoryRealloc(pArray->pData, tsize * pArray->elemSize);
    if (p == NULL) {
      return terrno;
    }

    pArray->pData = p;
    pArray->capacity = tsize;
  }
  return 0;
}

void* taosArrayAddBatch(SArray* pArray, const void* pData, int32_t nEles) {
  if (pData == NULL) {
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }

  int32_t code = taosArrayEnsureCap(pArray, pArray->size + nEles);
  if (code) {
    terrno = code;
    return NULL;
  }

  void* dst = TARRAY_GET_ELEM(pArray, pArray->size);
  memcpy(dst, pData, pArray->elemSize * nEles);

  pArray->size += nEles;
  return dst;
}

void taosArrayRemoveDuplicate(SArray* pArray, __compar_fn_t comparFn, void (*fp)(void*)) {
  size_t size = pArray->size;
  if (size <= 1) {
    return;
  }

  int32_t pos = 0;
  for (int32_t i = 1; i < size; ++i) {
    char* p1 = taosArrayGet(pArray, pos);
    char* p2 = taosArrayGet(pArray, i);

    if (comparFn(p1, p2) == 0) {
      // do nothing
    } else {
      if (pos + 1 != i) {
        void* p = taosArrayGet(pArray, pos + 1);
        if (fp != NULL) {
          fp(p);
        }

        taosArraySet(pArray, pos + 1, p2);
        memset(TARRAY_GET_ELEM(pArray, i), 0, pArray->elemSize);
        pos += 1;
      } else {
        pos += 1;
      }
    }
  }

  if (fp != NULL) {
    for (int32_t i = pos + 1; i < pArray->size; ++i) {
      void* p = taosArrayGet(pArray, i);
      fp(p);
    }
  }

  pArray->size = pos + 1;
}

void* taosArrayAddAll(SArray* pArray, const SArray* pInput) {
  if (pInput) {
    return taosArrayAddBatch(pArray, pInput->pData, (int32_t)taosArrayGetSize(pInput));
  } else {
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }
}

void* taosArrayReserve(SArray* pArray, int32_t num) {
  int32_t code = 0;
  if (pArray == NULL || num < 0) {
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }
  code = taosArrayEnsureCap(pArray, pArray->size + num);
  if (code) {
    terrno = code;
    return NULL;
  }

  void* dst = TARRAY_GET_ELEM(pArray, pArray->size);
  pArray->size += num;

  memset(dst, 0, num * pArray->elemSize);

  return dst;
}

void* taosArrayPop(SArray* pArray) {
  if (pArray->size == 0) {
    return NULL;
  }

  pArray->size -= 1;
  return TARRAY_GET_ELEM(pArray, pArray->size);
}

void* taosArrayGet(const SArray* pArray, size_t index) {
  if (NULL == pArray) {
    terrno = TSDB_CODE_INVALID_PARA;
    uError("failed to return value from array of null ptr");
    return NULL;
  }

  if (index >= pArray->size) {
    uWarn("index is out of range, current:%" PRIzu " max:%" PRIzu, index, pArray->size);
    terrno = TSDB_CODE_OUT_OF_RANGE;
    return NULL;
  }

  return TARRAY_GET_ELEM(pArray, index);
}

void* taosArrayGetP(const SArray* pArray, size_t index) {
  void** p = taosArrayGet(pArray, index);
  if (p == NULL) {
    return NULL;
  }
  return *p;
}

void* taosArrayGetLast(const SArray* pArray) {
  if (pArray->size == 0) {
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }

  return TARRAY_GET_ELEM(pArray, pArray->size - 1);
}

size_t taosArrayGetSize(const SArray* pArray) {
  if (pArray == NULL) {
    return 0;
  }
  return TARRAY_SIZE(pArray);
}

void* taosArrayInsert(SArray* pArray, size_t index, const void* pData) {
  if (pArray == NULL || pData == NULL) {
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }

  if (index >= pArray->size) {
    return taosArrayPush(pArray, pData);
  }

  if (pArray->size >= pArray->capacity) {
    int32_t ret = taosArrayResize(pArray);
    if (ret < 0) {
      terrno = ret;
      return NULL;
    }
  }

  void* dst = TARRAY_GET_ELEM(pArray, index);

  int32_t remain = (int32_t)(pArray->size - index);
  memmove((char*)dst + pArray->elemSize, (char*)dst, pArray->elemSize * remain);
  memcpy(dst, pData, pArray->elemSize);

  pArray->size += 1;

  return dst;
}

void taosArraySet(SArray* pArray, size_t index, void* pData) {
  if (index < pArray->size) {
    memcpy(TARRAY_GET_ELEM(pArray, index), pData, pArray->elemSize);
  }
}

void taosArrayPopFrontBatch(SArray* pArray, size_t cnt) {
  if (cnt > pArray->size) {
    cnt = pArray->size;
  }

  pArray->size = pArray->size - cnt;
  if (pArray->size == 0 || cnt == 0) {
    return;
  }
  memmove(pArray->pData, (char*)pArray->pData + cnt * pArray->elemSize, pArray->size * pArray->elemSize);
}

void taosArrayPopTailBatch(SArray* pArray, size_t cnt) {
  if (cnt >= pArray->size) {
    cnt = pArray->size;
  }

  pArray->size = pArray->size - cnt;
}

void taosArrayRemove(SArray* pArray, size_t index) {
  if (!(index < pArray->size)) {
    return;
  }

  if (index == pArray->size - 1) {
    void* t = taosArrayPop(pArray);
    return;
  }

  size_t remain = pArray->size - index - 1;
  memmove((char*)pArray->pData + index * pArray->elemSize, (char*)pArray->pData + (index + 1) * pArray->elemSize,
          remain * pArray->elemSize);
  pArray->size -= 1;
}

void taosArrayRemoveP(SArray* pArray, size_t index, void (*fp)(void*)){
  if (!(index < pArray->size)) {
    return;
  }

  if (index == pArray->size - 1) {
    void** t = taosArrayPop(pArray);
    if (fp && t!= NULL) {
      fp(*t);
    }
    return;
  }

  void* p = taosArrayGetP(pArray, index);
  if (fp) {
    fp(p);
  }

  size_t remain = pArray->size - index - 1;
  memmove((char*)pArray->pData + index * pArray->elemSize, (char*)pArray->pData + (index + 1) * pArray->elemSize,
          remain * pArray->elemSize);
  pArray->size -= 1;
}

void taosArrayRemoveBatch(SArray* pArray, size_t index, size_t num, FDelete fp) {
  if(num == 0) {
    return;
  }
  if (index + num <= pArray->size) {
    if (fp) {
      for (int32_t i = 0; i < num; i++) {
        fp(taosArrayGet(pArray, index + i));
      }
    }

    memmove((char*)pArray->pData + index * pArray->elemSize, (char*)pArray->pData + (index + num) * pArray->elemSize,
            (pArray->size - index - num) * pArray->elemSize);
    pArray->size -= num;
  }
}

SArray* taosArrayFromList(const void* src, size_t size, size_t elemSize) {
  if (elemSize <= 0) {
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }

  SArray* pDst = taosArrayInit(size, elemSize);
  if (pDst) {
    memcpy(pDst->pData, src, elemSize * size);
    pDst->size = size;
  }

  return pDst;
}

SArray* taosArrayDup(const SArray* pSrc, __array_item_dup_fn_t fn) {
  if (NULL == pSrc) {
    terrno = TSDB_CODE_INVALID_PARA;
    return NULL;
  }

  if (pSrc->size == 0) {  // empty array list
    return taosArrayInit(8, pSrc->elemSize);
  }

  SArray* dst = taosArrayInit(pSrc->size, pSrc->elemSize);

  if (dst) {
    if (fn == NULL) {
      memcpy(dst->pData, pSrc->pData, pSrc->elemSize * pSrc->size);
    } else {
      for (int32_t i = 0; i < pSrc->size; ++i) {
        void* p = fn(taosArrayGetP(pSrc, i));
        memcpy(((char*)dst->pData) + i * dst->elemSize, &p, dst->elemSize);
      }
    }

    dst->size = pSrc->size;
  }

  return dst;
}

void taosArrayClear(SArray* pArray) {
  if (pArray == NULL) return;
  pArray->size = 0;
}

void taosArrayClearEx(SArray* pArray, void (*fp)(void*)) {
  if (pArray == NULL) return;
  if (fp == NULL) {
    pArray->size = 0;
    return;
  }

  for (int32_t i = 0; i < pArray->size; ++i) {
    fp(TARRAY_GET_ELEM(pArray, i));
  }

  pArray->size = 0;
}
void taosArrayClearP(SArray* pArray, void (*fp)(void*)) {
  // if (pArray == NULL) return;
  // if (fp == NULL) {
  //   pArray->size = 0;
  //   return;
  // }

  // for (int32_t i = 0; i < pArray->size; ++i) {
  //   fp(TARRAY_GET_ELEM(pArray, i));
  // }
  if (pArray) {
    if (NULL == fp) {
      for (int32_t i = 0; i < pArray->size; i++) {
        taosMemoryFree(*(void**)TARRAY_GET_ELEM(pArray, i));
      }
    } else {
      for (int32_t i = 0; i < pArray->size; i++) {
        fp(*(void**)TARRAY_GET_ELEM(pArray, i));
      }
    }
  }
  taosArrayClear(pArray);
}

void taosArrayDestroy(SArray* pArray) {
  if (pArray) {
    taosMemoryFree(pArray->pData);
    taosMemoryFree(pArray);
  }
}

void taosArrayDestroyP(SArray* pArray, FDelete fp) {
  if (pArray) {
    if (NULL == fp) {
      for (int32_t i = 0; i < pArray->size; i++) {
        taosMemoryFree(*(void**)TARRAY_GET_ELEM(pArray, i));
      }
    } else {
      for (int32_t i = 0; i < pArray->size; i++) {
        fp(*(void**)TARRAY_GET_ELEM(pArray, i));
      }
    }
    taosArrayDestroy(pArray);
  }
}

void taosArrayDestroyEx(SArray* pArray, FDelete fp) {
  if (pArray == NULL) {
    return;
  }

  if (fp == NULL) {
    taosArrayDestroy(pArray);
    return;
  }

  for (int32_t i = 0; i < pArray->size; ++i) {
    fp(TARRAY_GET_ELEM(pArray, i));
  }

  taosArrayDestroy(pArray);
}

void taosArraySort(SArray* pArray, __compar_fn_t compar) {
  taosSort(pArray->pData, pArray->size, pArray->elemSize, compar);
}

int32_t taosArrayMSort(SArray* pArray, __compar_fn_t compar) {
  return taosMergeSort(pArray->pData, pArray->size, pArray->elemSize, compar);
}

void* taosArraySearch(const SArray* pArray, const void* key, __compar_fn_t comparFn, int32_t flags) {
  return taosbsearch(key, pArray->pData, pArray->size, pArray->elemSize, comparFn, flags);
}

int32_t taosArraySearchIdx(const SArray* pArray, const void* key, __compar_fn_t comparFn, int32_t flags) {
  void* item = taosArraySearch(pArray, key, comparFn, flags);
  return item == NULL ? -1 : (int32_t)((char*)item - (char*)pArray->pData) / pArray->elemSize;
}

#ifdef BUILD_NO_CALL
static int32_t taosArrayPartition(SArray* pArray, int32_t i, int32_t j, __ext_compar_fn_t fn, const void* userData) {
  void* key = taosArrayGetP(pArray, i);
  while (i < j) {
    while (i < j && fn(taosArrayGetP(pArray, j), key, userData) >= 0) {
      j--;
    }
    if (i < j) {
      void* a = taosArrayGetP(pArray, j);
      taosArraySet(pArray, i, &a);
    }
    while (i < j && fn(taosArrayGetP(pArray, i), key, userData) <= 0) {
      i++;
    }
    if (i < j) {
      void* a = taosArrayGetP(pArray, i);
      taosArraySet(pArray, j, &a);
    }
  }
  taosArraySet(pArray, i, &key);
  return i;
}

static void taosArrayQuicksortImpl(SArray* pArray, int32_t low, int32_t high, __ext_compar_fn_t fn, const void* param) {
  if (low < high) {
    int32_t idx = taosArrayPartition(pArray, low, high, fn, param);
    taosArrayQuicksortImpl(pArray, low, idx - 1, fn, param);
    taosArrayQuicksortImpl(pArray, idx + 1, high, fn, param);
  }
}

static void taosArrayQuickSort(SArray* pArray, __ext_compar_fn_t fn, const void* param) {
  if (pArray->size <= 1) {
    return;
  }
  taosArrayQuicksortImpl(pArray, 0, (int32_t)(taosArrayGetSize(pArray) - 1), fn, param);
}

static void taosArrayInsertSort(SArray* pArray, __ext_compar_fn_t fn, const void* param) {
  if (pArray->size <= 1) {
    return;
  }

  for (int32_t i = 1; i <= pArray->size - 1; ++i) {
    for (int32_t j = i; j > 0; --j) {
      if (fn(taosArrayGetP(pArray, j), taosArrayGetP(pArray, j - 1), param) == -1) {
        void* a = taosArrayGetP(pArray, j);
        void* b = taosArrayGetP(pArray, j - 1);
        taosArraySet(pArray, j - 1, &a);
        taosArraySet(pArray, j, &b);
      } else {
        break;
      }
    }
  }
}
#endif

int32_t taosEncodeArray(void** buf, const SArray* pArray, FEncode encode) {
  int32_t tlen = 0;
  int32_t sz = pArray->size;
  tlen += taosEncodeFixedI32(buf, sz);
  for (int32_t i = 0; i < sz; i++) {
    void* data = taosArrayGetP(pArray, i);
    tlen += encode(buf, data);
  }
  return tlen;
}

void* taosDecodeArray(const void* buf, SArray** pArray, FDecode decode, int32_t dataSz, int8_t sver) {
  int32_t sz;
  buf = taosDecodeFixedI32(buf, &sz);
  *pArray = taosArrayInit(sz, sizeof(void*));
  if (*pArray == NULL) {
    return NULL;
  }
  for (int32_t i = 0; i < sz; i++) {
    void* data = taosMemoryCalloc(1, dataSz);
    if (data == NULL) {
      return NULL;
    }

    buf = decode(buf, data, sver);

    if (taosArrayPush(*pArray, &data) == NULL) {
      taosMemoryFree(data);
      return NULL;
    }
  }
  return (void*)buf;
}

// todo remove it
// order array<type *>
int32_t taosArraySortPWithExt(SArray* pArray, __ext_compar_fn_t fn, const void* param) {
  return taosqsort(pArray->pData, pArray->size, pArray->elemSize, param, fn);
}

void taosArraySwap(SArray* a, SArray* b) {
  if (a == NULL || b == NULL) return;
  size_t t = a->size;
  a->size = b->size;
  b->size = t;

  uint32_t cap = a->capacity;
  a->capacity = b->capacity;
  b->capacity = cap;

  uint32_t elem = a->elemSize;
  a->elemSize = b->elemSize;
  b->elemSize = elem;

  void* data = a->pData;
  a->pData = b->pData;
  b->pData = data;
}

1	/*
2	* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3	*
4	* This program is free software: you can use, redistribute, and/or modify
5	* it under the terms of the GNU Affero General Public License, version 3
6	* or later ("AGPL"), as published by the Free Software Foundation.
7	*
8	* This program is distributed in the hope that it will be useful, but WITHOUT
9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10	* FITNESS FOR A PARTICULAR PURPOSE.
11	*
12	* You should have received a copy of the GNU Affero General Public License
13	* along with this program. If not, see <http://www.gnu.org/licenses/>.
14	*/
15
16	#define _DEFAULT_SOURCE
17	#include "tarray.h"
18	#include "tcoding.h"
19
20	// todo refactor API
21	#define BOUNDARY_SIZE 1024 * 1024 * 1024 // 1G
22	#define BOUNDARY_SMALL_FACTOR 1.2
23	#define BOUNDARY_BIG_FACTOR 2
24
25	SArray* taosArrayInit(size_t size, size_t elemSize) {	2,147,483,647✔
26	if (elemSize == 0) {	2,147,483,647✔
27	terrno = TSDB_CODE_INVALID_PARA;	200✔
28	return NULL;	200✔
29	}
30
31	if (size < TARRAY_MIN_SIZE) {	2,147,483,647✔
32	size = TARRAY_MIN_SIZE;	2,147,483,647✔
33	}
34
35	SArray* pArray = taosMemoryMalloc(sizeof(SArray));	2,147,483,647✔
36	if (pArray == NULL) {	2,147,483,647✔
37	return NULL;	×
38	}
39
40	pArray->size = 0;	2,147,483,647✔
41	pArray->pData = taosMemoryCalloc(size, elemSize);	2,147,483,647✔
42	if (pArray->pData == NULL) {	2,147,483,647✔
43	taosMemoryFree(pArray);	×
44	return NULL;	×
45	}
46
47	pArray->capacity = size;	2,147,483,647✔
48	pArray->elemSize = elemSize;	2,147,483,647✔
49	return pArray;	2,147,483,647✔
50	}
51
52	SArray* taosArrayInit_s(size_t elemSize, size_t initialSize) {	322,265,502✔
53	SArray* pArray = taosMemoryMalloc(sizeof(SArray));	322,265,502✔
54	if (pArray == NULL) {	322,281,859✔
55	return NULL;	×
56	}
57
58	pArray->size = initialSize;	322,281,859✔
59	pArray->pData = taosMemoryCalloc(initialSize, elemSize);	322,284,956✔
60	if (pArray->pData == NULL) {	322,301,551✔
61	taosMemoryFree(pArray);	×
62	return NULL;	×
63	}
64
65	pArray->capacity = initialSize;	322,291,581✔
66	pArray->elemSize = elemSize;	322,306,694✔
67	return pArray;	322,299,580✔
68	}
69
70	static int32_t taosArrayResize(SArray* pArray) {	×
71	size_t size = pArray->capacity;	×
72	size = (size << 1u);	×
73
74	void* tmp = taosMemoryRealloc(pArray->pData, size * pArray->elemSize);	×
75	if (tmp == NULL) { // reallocate failed, the original buffer remains	×
76	return terrno;	×
77	}
78
79	pArray->pData = tmp;	×
80	pArray->capacity = size;	×
81
82	return 0;	×
83	}
84
85	int32_t taosArrayEnsureCap(SArray* pArray, size_t newCap) {	2,147,483,647✔
86	if (newCap > pArray->capacity) {	2,147,483,647✔
87	float factor = BOUNDARY_BIG_FACTOR;	1,855,853,913✔
88	if (newCap * pArray->elemSize > BOUNDARY_SIZE) {	1,855,853,913✔
89	factor = BOUNDARY_SMALL_FACTOR;	×
90	}
91
92	size_t tsize = (pArray->capacity * factor);	1,855,848,959✔
93	while (newCap > tsize) {	1,880,370,033✔
94	size_t newSize = (tsize * factor);	24,662,269✔
95	tsize = (newSize == tsize) ? (tsize + 2) : newSize;	24,662,269✔
96	}
97
98	char* p = taosMemoryRealloc(pArray->pData, tsize * pArray->elemSize);	1,855,707,764✔
99	if (p == NULL) {	1,855,691,335✔
100	return terrno;	×
101	}
102
103	pArray->pData = p;	1,855,691,335✔
104	pArray->capacity = tsize;	1,855,704,286✔
105	}
106	return 0;	2,147,483,647✔
107	}
108
109	void* taosArrayAddBatch(SArray* pArray, const void* pData, int32_t nEles) {	2,147,483,647✔
110	if (pData == NULL) {	2,147,483,647✔
111	terrno = TSDB_CODE_INVALID_PARA;	200✔
112	return NULL;	200✔
113	}
114
115	int32_t code = taosArrayEnsureCap(pArray, pArray->size + nEles);	2,147,483,647✔
116	if (code) {	2,147,483,647✔
117	terrno = code;	237,670✔
118	return NULL;	×
119	}
120
121	void* dst = TARRAY_GET_ELEM(pArray, pArray->size);	2,147,483,647✔
122	memcpy(dst, pData, pArray->elemSize * nEles);	2,147,483,647✔
123
124	pArray->size += nEles;	2,147,483,647✔
125	return dst;	2,147,483,647✔
126	}
127
128	void taosArrayRemoveDuplicate(SArray* pArray, __compar_fn_t comparFn, void (fp)(void)) {	87,153,372✔
129	size_t size = pArray->size;	87,153,372✔
130	if (size <= 1) {	87,156,558✔
131	return;	3,300,909✔
132	}
133
134	int32_t pos = 0;	83,855,649✔
135	for (int32_t i = 1; i < size; ++i) {	369,775,875✔
136	char* p1 = taosArrayGet(pArray, pos);	285,912,521✔
137	char* p2 = taosArrayGet(pArray, i);	285,874,756✔
138
139	if (comparFn(p1, p2) == 0) {	285,853,436✔
140	// do nothing
141	} else {
142	if (pos + 1 != i) {	266,664,285✔
143	void* p = taosArrayGet(pArray, pos + 1);	2,923,715✔
144	if (fp != NULL) {	2,923,715✔
145	fp(p);	600✔
146	}
147
148	taosArraySet(pArray, pos + 1, p2);	2,923,715✔
149	memset(TARRAY_GET_ELEM(pArray, i), 0, pArray->elemSize);	2,923,715✔
150	pos += 1;	2,923,715✔
151	} else {
152	pos += 1;	263,740,570✔
153	}
154	}
155	}
156
157	if (fp != NULL) {	83,863,354✔
158	for (int32_t i = pos + 1; i < pArray->size; ++i) {	5,573✔
159	void* p = taosArrayGet(pArray, i);	2,336✔
160	fp(p);	2,336✔
161	}
162	}
163
164	pArray->size = pos + 1;	83,863,354✔
165	}
166
167	void* taosArrayAddAll(SArray* pArray, const SArray* pInput) {	147,496,715✔
168	if (pInput) {	147,496,715✔
169	return taosArrayAddBatch(pArray, pInput->pData, (int32_t)taosArrayGetSize(pInput));	146,633,841✔
170	} else {
171	terrno = TSDB_CODE_INVALID_PARA;	863,181✔
172	return NULL;	868,838✔
173	}
174	}
175
176	void* taosArrayReserve(SArray* pArray, int32_t num) {	2,147,483,647✔
177	int32_t code = 0;	2,147,483,647✔
178	if (pArray == NULL \|\| num < 0) {	2,147,483,647✔
179	terrno = TSDB_CODE_INVALID_PARA;	105✔
180	return NULL;	×
181	}
182	code = taosArrayEnsureCap(pArray, pArray->size + num);	2,147,483,647✔
183	if (code) {	2,147,483,647✔
UNCOV 184	terrno = code;	×
185	return NULL;	×
186	}
187
188	void* dst = TARRAY_GET_ELEM(pArray, pArray->size);	2,147,483,647✔
189	pArray->size += num;	2,147,483,647✔
190
191	memset(dst, 0, num * pArray->elemSize);	2,147,483,647✔
192
193	return dst;	2,147,483,647✔
194	}
195
196	void* taosArrayPop(SArray* pArray) {	2,147,483,647✔
197	if (pArray->size == 0) {	2,147,483,647✔
198	return NULL;	26,611,318✔
199	}
200
201	pArray->size -= 1;	2,147,483,647✔
202	return TARRAY_GET_ELEM(pArray, pArray->size);	2,147,483,647✔
203	}
204
205	void* taosArrayGet(const SArray* pArray, size_t index) {	2,147,483,647✔
206	if (NULL == pArray) {	2,147,483,647✔
207	terrno = TSDB_CODE_INVALID_PARA;	20,053✔
208	uError("failed to return value from array of null ptr");	20,053✔
209	return NULL;	20,053✔
210	}
211
212	if (index >= pArray->size) {	2,147,483,647✔
213	uWarn("index is out of range, current:%" PRIzu " max:%" PRIzu, index, pArray->size);	2,808,420✔
214	terrno = TSDB_CODE_OUT_OF_RANGE;	2,808,501✔
215	return NULL;	2,808,501✔
216	}
217
218	return TARRAY_GET_ELEM(pArray, index);	2,147,483,647✔
219	}
220
221	void* taosArrayGetP(const SArray* pArray, size_t index) {	2,147,483,647✔
222	void** p = taosArrayGet(pArray, index);	2,147,483,647✔
223	if (p == NULL) {	2,147,483,647✔
224	return NULL;	2,640,134✔
225	}
226	return *p;	2,147,483,647✔
227	}
228
229	void* taosArrayGetLast(const SArray* pArray) {	2,147,483,647✔
230	if (pArray->size == 0) {	2,147,483,647✔
231	terrno = TSDB_CODE_INVALID_PARA;	10,177,070✔
232	return NULL;	10,176,841✔
233	}
234
235	return TARRAY_GET_ELEM(pArray, pArray->size - 1);	2,147,483,647✔
236	}
237
238	size_t taosArrayGetSize(const SArray* pArray) {	2,147,483,647✔
239	if (pArray == NULL) {	2,147,483,647✔
240	return 0;	2,147,483,647✔
241	}
242	return TARRAY_SIZE(pArray);	2,147,483,647✔
243	}
244
245	void* taosArrayInsert(SArray* pArray, size_t index, const void* pData) {	2,147,483,647✔
246	if (pArray == NULL \|\| pData == NULL) {	2,147,483,647✔
247	terrno = TSDB_CODE_INVALID_PARA;	14✔
248	return NULL;	200✔
249	}
250
251	if (index >= pArray->size) {	2,147,483,647✔
252	return taosArrayPush(pArray, pData);	2,147,483,647✔
253	}
254
255	if (pArray->size >= pArray->capacity) {	5,654,475✔
256	int32_t ret = taosArrayResize(pArray);	×
257	if (ret < 0) {	×
258	terrno = ret;	63,419✔
259	return NULL;	×
260	}
261	}
262
263	void* dst = TARRAY_GET_ELEM(pArray, index);	5,591,101✔
264
265	int32_t remain = (int32_t)(pArray->size - index);	5,591,087✔
266	memmove((char)dst + pArray->elemSize, (char)dst, pArray->elemSize * remain);	5,591,087✔
267	memcpy(dst, pData, pArray->elemSize);	5,591,129✔
268
269	pArray->size += 1;	5,591,003✔
270
271	return dst;	5,591,031✔
272	}
273
274	void taosArraySet(SArray* pArray, size_t index, void* pData) {	2,147,483,647✔
275	if (index < pArray->size) {	2,147,483,647✔
276	memcpy(TARRAY_GET_ELEM(pArray, index), pData, pArray->elemSize);	2,147,483,647✔
277	}
278	}	2,147,483,647✔
279
280	void taosArrayPopFrontBatch(SArray* pArray, size_t cnt) {	7,628,905✔
281	if (cnt > pArray->size) {	7,628,905✔
282	cnt = pArray->size;	200✔
283	}
284
285	pArray->size = pArray->size - cnt;	7,628,905✔
286	if (pArray->size == 0 \|\| cnt == 0) {	7,628,905✔
287	return;	5,135,096✔
288	}
289	memmove(pArray->pData, (char)pArray->pData + cnt pArray->elemSize, pArray->size * pArray->elemSize);	2,493,578✔
290	}
291
292	void taosArrayPopTailBatch(SArray* pArray, size_t cnt) {	48,106,664✔
293	if (cnt >= pArray->size) {	48,106,664✔
294	cnt = pArray->size;	20,746,552✔
295	}
296
297	pArray->size = pArray->size - cnt;	48,108,867✔
298	}	48,110,216✔
299
300	void taosArrayRemove(SArray* pArray, size_t index) {	554,435,317✔
301	if (!(index < pArray->size)) {	554,435,317✔
302	return;	×
303	}
304
305	if (index == pArray->size - 1) {	554,436,385✔
306	void* t = taosArrayPop(pArray);	182,037,214✔
307	return;	182,037,768✔
308	}
309
310	size_t remain = pArray->size - index - 1;	372,399,322✔
311	memmove((char)pArray->pData + index pArray->elemSize, (char)pArray->pData + (index + 1) pArray->elemSize,	372,399,143✔
312	remain * pArray->elemSize);	372,396,701✔
313	pArray->size -= 1;	372,401,223✔
314	}
315
316	void taosArrayRemoveP(SArray* pArray, size_t index, void (fp)(void)){	×
317	if (!(index < pArray->size)) {	×
318	return;	×
319	}
320
321	if (index == pArray->size - 1) {	×
322	void** t = taosArrayPop(pArray);	×
323	if (fp && t!= NULL) {	×
324	fp(*t);	×
325	}
326	return;	×
327	}
328
329	void* p = taosArrayGetP(pArray, index);	×
330	if (fp) {	×
331	fp(p);	×
332	}
333
334	size_t remain = pArray->size - index - 1;	×
335	memmove((char)pArray->pData + index pArray->elemSize, (char)pArray->pData + (index + 1) pArray->elemSize,	×
336	remain * pArray->elemSize);	×
337	pArray->size -= 1;	×
338	}
339
340	void taosArrayRemoveBatch(SArray* pArray, size_t index, size_t num, FDelete fp) {	52,547,848✔
341	if(num == 0) {	52,547,848✔
342	return;	×
343	}
344	if (index + num <= pArray->size) {	52,547,848✔
345	if (fp) {	52,548,612✔
346	for (int32_t i = 0; i < num; i++) {	138,554,780✔
347	fp(taosArrayGet(pArray, index + i));	86,006,487✔
348	}
349	}
350
351	memmove((char)pArray->pData + index pArray->elemSize, (char)pArray->pData + (index + num) pArray->elemSize,	52,546,099✔
352	(pArray->size - index - num) * pArray->elemSize);	52,550,122✔
353	pArray->size -= num;	52,546,644✔
354	}
355	}
356
357	SArray* taosArrayFromList(const void* src, size_t size, size_t elemSize) {	200✔
358	if (elemSize <= 0) {	200✔
359	terrno = TSDB_CODE_INVALID_PARA;	×
360	return NULL;	×
361	}
362
363	SArray* pDst = taosArrayInit(size, elemSize);	200✔
364	if (pDst) {	200✔
365	memcpy(pDst->pData, src, elemSize * size);	200✔
366	pDst->size = size;	200✔
367	}
368
369	return pDst;	200✔
370	}
371
372	SArray* taosArrayDup(const SArray* pSrc, __array_item_dup_fn_t fn) {	1,357,304,748✔
373	if (NULL == pSrc) {	1,357,304,748✔
374	terrno = TSDB_CODE_INVALID_PARA;	4,317,033✔
375	return NULL;	4,317,033✔
376	}
377
378	if (pSrc->size == 0) { // empty array list	1,352,987,715✔
379	return taosArrayInit(8, pSrc->elemSize);	116,667,356✔
380	}
381
382	SArray* dst = taosArrayInit(pSrc->size, pSrc->elemSize);	1,236,297,151✔
383
384	if (dst) {	1,236,192,184✔
385	if (fn == NULL) {	1,236,212,495✔
386	memcpy(dst->pData, pSrc->pData, pSrc->elemSize * pSrc->size);	1,235,988,397✔
387	} else {
388	for (int32_t i = 0; i < pSrc->size; ++i) {	587,183✔
389	void* p = fn(taosArrayGetP(pSrc, i));	341,519✔
390	memcpy(((char)dst->pData) + i dst->elemSize, &p, dst->elemSize);	363,085✔
391	}
392	}
393
394	dst->size = pSrc->size;	1,236,451,019✔
395	}
396
397	return dst;	1,236,425,406✔
398	}
399
400	void taosArrayClear(SArray* pArray) {	2,147,483,647✔
401	if (pArray == NULL) return;	2,147,483,647✔
402	pArray->size = 0;	2,147,483,647✔
403	}
404
405	void taosArrayClearEx(SArray* pArray, void (fp)(void)) {	163,620,647✔
406	if (pArray == NULL) return;	163,620,647✔
407	if (fp == NULL) {	144,799,685✔
408	pArray->size = 0;	200✔
409	return;	200✔
410	}
411
412	for (int32_t i = 0; i < pArray->size; ++i) {	2,147,483,647✔
413	fp(TARRAY_GET_ELEM(pArray, i));	2,147,483,647✔
414	}
415
416	pArray->size = 0;	144,746,329✔
417	}
418	void taosArrayClearP(SArray* pArray, void (fp)(void)) {	3,847,162✔
419	// if (pArray == NULL) return;
420	// if (fp == NULL) {
421	// pArray->size = 0;
422	// return;
423	// }
424
425	// for (int32_t i = 0; i < pArray->size; ++i) {
426	// fp(TARRAY_GET_ELEM(pArray, i));
427	// }
428	if (pArray) {	3,847,162✔
429	if (NULL == fp) {	3,847,162✔
430	for (int32_t i = 0; i < pArray->size; i++) {	460,196✔
431	taosMemoryFree((void*)TARRAY_GET_ELEM(pArray, i));	2,192✔
432	}
433	} else {
434	for (int32_t i = 0; i < pArray->size; i++) {	9,646,422✔
435	fp((void*)TARRAY_GET_ELEM(pArray, i));	6,257,036✔
436	}
437	}
438	}
439	taosArrayClear(pArray);	3,847,618✔
440	}	3,847,846✔
441
442	void taosArrayDestroy(SArray* pArray) {	2,147,483,647✔
443	if (pArray) {	2,147,483,647✔
444	taosMemoryFree(pArray->pData);	2,147,483,647✔
445	taosMemoryFree(pArray);	2,147,483,647✔
446	}
447	}	2,147,483,647✔
448
449	void taosArrayDestroyP(SArray* pArray, FDelete fp) {	2,147,483,647✔
450	if (pArray) {	2,147,483,647✔
451	if (NULL == fp) {	1,188,575,426✔
452	for (int32_t i = 0; i < pArray->size; i++) {	1,121,914,112✔
453	taosMemoryFree((void*)TARRAY_GET_ELEM(pArray, i));	521,881,792✔
454	}
455	} else {
456	for (int32_t i = 0; i < pArray->size; i++) {	1,312,120,558✔
457	fp((void*)TARRAY_GET_ELEM(pArray, i));	723,608,991✔
458	}
459	}
460	taosArrayDestroy(pArray);	1,188,479,554✔
461	}
462	}	2,147,483,647✔
463
464	void taosArrayDestroyEx(SArray* pArray, FDelete fp) {	2,147,483,647✔
465	if (pArray == NULL) {	2,147,483,647✔
466	return;	2,147,483,647✔
467	}
468
469	if (fp == NULL) {	2,147,483,647✔
470	taosArrayDestroy(pArray);	189,704,202✔
471	return;	189,712,481✔
472	}
473
474	for (int32_t i = 0; i < pArray->size; ++i) {	2,147,483,647✔
475	fp(TARRAY_GET_ELEM(pArray, i));	2,147,483,647✔
476	}
477
478	taosArrayDestroy(pArray);	2,147,483,647✔
479	}
480
481	void taosArraySort(SArray* pArray, __compar_fn_t compar) {	1,195,825,199✔
482	taosSort(pArray->pData, pArray->size, pArray->elemSize, compar);	1,195,825,199✔
483	}	1,195,883,886✔
484
485	int32_t taosArrayMSort(SArray* pArray, __compar_fn_t compar) {	1,542,413✔
486	return taosMergeSort(pArray->pData, pArray->size, pArray->elemSize, compar);	1,542,413✔
487	}
488
489	void* taosArraySearch(const SArray* pArray, const void* key, __compar_fn_t comparFn, int32_t flags) {	1,475,153,060✔
490	return taosbsearch(key, pArray->pData, pArray->size, pArray->elemSize, comparFn, flags);	1,475,153,060✔
491	}
492
493	int32_t taosArraySearchIdx(const SArray* pArray, const void* key, __compar_fn_t comparFn, int32_t flags) {	398,297,640✔
494	void* item = taosArraySearch(pArray, key, comparFn, flags);	398,297,640✔
495	return item == NULL ? -1 : (int32_t)((char)item - (char)pArray->pData) / pArray->elemSize;	398,226,483✔
496	}
497
498	#ifdef BUILD_NO_CALL
499	static int32_t taosArrayPartition(SArray* pArray, int32_t i, int32_t j, __ext_compar_fn_t fn, const void* userData) {
500	void* key = taosArrayGetP(pArray, i);
501	while (i < j) {
502	while (i < j && fn(taosArrayGetP(pArray, j), key, userData) >= 0) {
503	j--;
504	}
505	if (i < j) {
506	void* a = taosArrayGetP(pArray, j);
507	taosArraySet(pArray, i, &a);
508	}
509	while (i < j && fn(taosArrayGetP(pArray, i), key, userData) <= 0) {
510	i++;
511	}
512	if (i < j) {
513	void* a = taosArrayGetP(pArray, i);
514	taosArraySet(pArray, j, &a);
515	}
516	}
517	taosArraySet(pArray, i, &key);
518	return i;
519	}
520
521	static void taosArrayQuicksortImpl(SArray* pArray, int32_t low, int32_t high, __ext_compar_fn_t fn, const void* param) {
522	if (low < high) {
523	int32_t idx = taosArrayPartition(pArray, low, high, fn, param);
524	taosArrayQuicksortImpl(pArray, low, idx - 1, fn, param);
525	taosArrayQuicksortImpl(pArray, idx + 1, high, fn, param);
526	}
527	}
528
529	static void taosArrayQuickSort(SArray* pArray, __ext_compar_fn_t fn, const void* param) {
530	if (pArray->size <= 1) {
531	return;
532	}
533	taosArrayQuicksortImpl(pArray, 0, (int32_t)(taosArrayGetSize(pArray) - 1), fn, param);
534	}
535
536	static void taosArrayInsertSort(SArray* pArray, __ext_compar_fn_t fn, const void* param) {
537	if (pArray->size <= 1) {
538	return;
539	}
540
541	for (int32_t i = 1; i <= pArray->size - 1; ++i) {
542	for (int32_t j = i; j > 0; --j) {
543	if (fn(taosArrayGetP(pArray, j), taosArrayGetP(pArray, j - 1), param) == -1) {
544	void* a = taosArrayGetP(pArray, j);
545	void* b = taosArrayGetP(pArray, j - 1);
546	taosArraySet(pArray, j - 1, &a);
547	taosArraySet(pArray, j, &b);
548	} else {
549	break;
550	}
551	}
552	}
553	}
554	#endif
555
556	int32_t taosEncodeArray(void** buf, const SArray* pArray, FEncode encode) {	×
557	int32_t tlen = 0;	×
558	int32_t sz = pArray->size;	×
559	tlen += taosEncodeFixedI32(buf, sz);	×
560	for (int32_t i = 0; i < sz; i++) {	×
561	void* data = taosArrayGetP(pArray, i);	×
562	tlen += encode(buf, data);	×
563	}
564	return tlen;	×
565	}
566
567	void* taosDecodeArray(const void* buf, SArray** pArray, FDecode decode, int32_t dataSz, int8_t sver) {	×
568	int32_t sz;	×
569	buf = taosDecodeFixedI32(buf, &sz);	×
570	pArray = taosArrayInit(sz, sizeof(void));	×
571	if (*pArray == NULL) {	×
572	return NULL;	×
573	}
574	for (int32_t i = 0; i < sz; i++) {	×
575	void* data = taosMemoryCalloc(1, dataSz);	×
576	if (data == NULL) {	×
577	return NULL;	×
578	}
579
580	buf = decode(buf, data, sver);	×
581
582	if (taosArrayPush(*pArray, &data) == NULL) {	×
583	taosMemoryFree(data);	×
584	return NULL;	×
585	}
586	}
587	return (void*)buf;	×
588	}
589
590	// todo remove it
591	// order array<type *>
592	int32_t taosArraySortPWithExt(SArray* pArray, __ext_compar_fn_t fn, const void* param) {	196✔
593	return taosqsort(pArray->pData, pArray->size, pArray->elemSize, param, fn);	196✔
594	}
595
596	void taosArraySwap(SArray* a, SArray* b) {	5,264✔
597	if (a == NULL \|\| b == NULL) return;	5,264✔
598	size_t t = a->size;	5,264✔
599	a->size = b->size;	5,264✔
600	b->size = t;	5,264✔
601
602	uint32_t cap = a->capacity;	5,264✔
603	a->capacity = b->capacity;	5,264✔
604	b->capacity = cap;	5,264✔
605
606	uint32_t elem = a->elemSize;	5,264✔
607	a->elemSize = b->elemSize;	5,264✔
608	b->elemSize = elem;	5,264✔
609
610	void* data = a->pData;	5,264✔
611	a->pData = b->pData;	5,264✔
612	b->pData = data;	5,264✔
613	}

taosdata / TDengine / #5017

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous