#4998

Committed 21 Mar 2026 01:22PM UTC coverage: 72.335% (+0.6%) from 71.739%

Build # #4998

Build Type

push

travis-ci

Committed by

web-flow

Commit Message

enh:register add secondEp (#34867)

Coverage Stats

61 of 69 new or added lines in 1 file covered. (88.41%)

8670 existing lines in 142 files now uncovered.

253516 of 350475 relevant lines covered (72.33%)

133451446.75 hits per line

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

85.0

/source/dnode/vnode/src/vnd/vnodeHash.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/>.
 */

#include "vnodeHash.h"

#define VNODE_HASH_DEFAULT_NUM_BUCKETS 1024

typedef struct SVHashEntry SVHashEntry;

struct SVHashEntry {
  SVHashEntry* next;
  void*        obj;
};

static void vHashRehash(SVHashTable* ht, uint32_t newNumBuckets) {
  SVHashEntry** newBuckets = (SVHashEntry**)taosMemoryCalloc(newNumBuckets, sizeof(SVHashEntry*));
  if (newBuckets == NULL) {
    return;
  }

  for (int32_t i = 0; i < ht->numBuckets; i++) {
    SVHashEntry* entry = ht->buckets[i];
    while (entry != NULL) {
      SVHashEntry* next = entry->next;
      uint32_t     bucketIndex = ht->hash(entry->obj) % newNumBuckets;
      entry->next = newBuckets[bucketIndex];
      newBuckets[bucketIndex] = entry;
      entry = next;
    }
  }

  taosMemoryFree(ht->buckets);
  ht->buckets = newBuckets;
  ht->numBuckets = newNumBuckets;

  return;
}

int32_t vHashInit(SVHashTable** ht, uint32_t (*hash)(const void*), int32_t (*compare)(const void*, const void*)) {
  if (ht == NULL || hash == NULL || compare == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }

  (*ht) = (SVHashTable*)taosMemoryMalloc(sizeof(SVHashTable));
  if (*ht == NULL) {
    return terrno;
  }

  (*ht)->hash = hash;
  (*ht)->compare = compare;
  (*ht)->numEntries = 0;
  (*ht)->numBuckets = VNODE_HASH_DEFAULT_NUM_BUCKETS;
  (*ht)->buckets = (SVHashEntry**)taosMemoryCalloc((*ht)->numBuckets, sizeof(SVHashEntry*));
  if ((*ht)->buckets == NULL) {
    taosMemoryFree(*ht);
    return terrno;
  }

  return 0;
}

void vHashDestroy(SVHashTable** ht) {
  if (ht == NULL) {
    return;
  }

  if (*ht) {
    taosMemoryFree((*ht)->buckets);
    taosMemoryFree(*ht);
    (*ht) = NULL;
  }
}

int32_t vHashPut(SVHashTable* ht, void* obj) {
  if (ht == NULL || obj == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }

  uint32_t bucketIndex = ht->hash(obj) % ht->numBuckets;
  for (SVHashEntry* entry = ht->buckets[bucketIndex]; entry != NULL; entry = entry->next) {
    if (ht->compare(entry->obj, obj) == 0) {
      return TSDB_CODE_DUP_KEY;
    }
  }

  if (ht->numEntries >= ht->numBuckets) {
    vHashRehash(ht, ht->numBuckets * 2);
    bucketIndex = ht->hash(obj) % ht->numBuckets;
  }

  SVHashEntry* entry = (SVHashEntry*)taosMemoryMalloc(sizeof(SVHashEntry));
  if (entry == NULL) {
    return terrno;
  }
  entry->obj = obj;
  entry->next = ht->buckets[bucketIndex];
  ht->buckets[bucketIndex] = entry;
  ht->numEntries++;

  return 0;
}

int32_t vHashGet(SVHashTable* ht, const void* obj, void** retObj) {
  if (ht == NULL || obj == NULL || retObj == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }

  uint32_t bucketIndex = ht->hash(obj) % ht->numBuckets;
  for (SVHashEntry* entry = ht->buckets[bucketIndex]; entry != NULL; entry = entry->next) {
    if (ht->compare(entry->obj, obj) == 0) {
      *retObj = entry->obj;
      return 0;
    }
  }

  *retObj = NULL;
  return TSDB_CODE_NOT_FOUND;
}

int32_t vHashDrop(SVHashTable* ht, const void* obj) {
  if (ht == NULL || obj == NULL) {
    return TSDB_CODE_INVALID_PARA;
  }

  uint32_t bucketIndex = ht->hash(obj) % ht->numBuckets;
  for (SVHashEntry** entry = &ht->buckets[bucketIndex]; *entry != NULL; entry = &(*entry)->next) {
    if (ht->compare((*entry)->obj, obj) == 0) {
      SVHashEntry* tmp = *entry;
      *entry = (*entry)->next;
      taosMemoryFree(tmp);
      ht->numEntries--;
      if (ht->numBuckets > VNODE_HASH_DEFAULT_NUM_BUCKETS && ht->numEntries < ht->numBuckets / 4) {
        vHashRehash(ht, ht->numBuckets / 2);
      }
      return 0;
    }
  }

  return TSDB_CODE_NOT_FOUND;
}

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	#include "vnodeHash.h"
17
18	#define VNODE_HASH_DEFAULT_NUM_BUCKETS 1024
19
20	typedef struct SVHashEntry SVHashEntry;
21
22	struct SVHashEntry {
23	SVHashEntry* next;
24	void* obj;
25	};
26
27	static void vHashRehash(SVHashTable* ht, uint32_t newNumBuckets) {	15,216✔
28	SVHashEntry newBuckets = (SVHashEntry)taosMemoryCalloc(newNumBuckets, sizeof(SVHashEntry*));	15,216✔
29	if (newBuckets == NULL) {	15,216✔
30	return;	×
31	}
32
33	for (int32_t i = 0; i < ht->numBuckets; i++) {	23,386,992✔
34	SVHashEntry* entry = ht->buckets[i];	23,371,776✔
35	while (entry != NULL) {	35,050,056✔
36	SVHashEntry* next = entry->next;	11,678,280✔
37	uint32_t bucketIndex = ht->hash(entry->obj) % newNumBuckets;	11,678,280✔
38	entry->next = newBuckets[bucketIndex];	11,678,280✔
39	newBuckets[bucketIndex] = entry;	11,678,280✔
40	entry = next;	11,678,280✔
41	}
42	}
43
44	taosMemoryFree(ht->buckets);	15,216✔
45	ht->buckets = newBuckets;	15,216✔
46	ht->numBuckets = newNumBuckets;	15,216✔
47
48	return;	15,216✔
49	}
50
51	int32_t vHashInit(SVHashTable** ht, uint32_t (hash)(const void), int32_t (compare)(const void, const void*)) {	9,043,811✔
52	if (ht == NULL \|\| hash == NULL \|\| compare == NULL) {	9,043,811✔
53	return TSDB_CODE_INVALID_PARA;	×
54	}
55
56	(ht) = (SVHashTable)taosMemoryMalloc(sizeof(SVHashTable));	9,045,727✔
57	if (*ht == NULL) {	9,046,543✔
58	return terrno;	×
59	}
60
61	(*ht)->hash = hash;	9,045,420✔
62	(*ht)->compare = compare;	9,044,934✔
63	(*ht)->numEntries = 0;	9,044,482✔
64	(*ht)->numBuckets = VNODE_HASH_DEFAULT_NUM_BUCKETS;	9,044,614✔
65	(ht)->buckets = (SVHashEntry)taosMemoryCalloc((ht)->numBuckets, sizeof(SVHashEntry*));	9,042,954✔
66	if ((*ht)->buckets == NULL) {	9,047,128✔
67	taosMemoryFree(*ht);	×
68	return terrno;	×
69	}
70
71	return 0;	9,046,324✔
72	}
73
74	void vHashDestroy(SVHashTable** ht) {	9,046,409✔
75	if (ht == NULL) {	9,046,409✔
76	return;	×
77	}
78
79	if (*ht) {	9,046,409✔
80	taosMemoryFree((*ht)->buckets);	9,046,786✔
81	taosMemoryFree(*ht);	9,046,362✔
82	(*ht) = NULL;	9,047,128✔
83	}
84	}
85
86	int32_t vHashPut(SVHashTable* ht, void* obj) {	24,197,419✔
87	if (ht == NULL \|\| obj == NULL) {	24,197,419✔
88	return TSDB_CODE_INVALID_PARA;	×
89	}
90
91	uint32_t bucketIndex = ht->hash(obj) % ht->numBuckets;	24,198,486✔
92	for (SVHashEntry* entry = ht->buckets[bucketIndex]; entry != NULL; entry = entry->next) {	31,443,957✔
93	if (ht->compare(entry->obj, obj) == 0) {	7,244,239✔
94	return TSDB_CODE_DUP_KEY;	×
95	}
96	}
97
98	if (ht->numEntries >= ht->numBuckets) {	24,195,378✔
99	vHashRehash(ht, ht->numBuckets * 2);	7,608✔
100	bucketIndex = ht->hash(obj) % ht->numBuckets;	7,608✔
101	}
102
103	SVHashEntry* entry = (SVHashEntry*)taosMemoryMalloc(sizeof(SVHashEntry));	24,196,007✔
104	if (entry == NULL) {	24,204,084✔
105	return terrno;	×
106	}
107	entry->obj = obj;	24,204,084✔
108	entry->next = ht->buckets[bucketIndex];	24,198,241✔
109	ht->buckets[bucketIndex] = entry;	24,200,100✔
110	ht->numEntries++;	24,201,775✔
111
112	return 0;	24,203,270✔
113	}
114
115	int32_t vHashGet(SVHashTable* ht, const void* obj, void** retObj) {	105,329,742✔
116	if (ht == NULL \|\| obj == NULL \|\| retObj == NULL) {	105,329,742✔
UNCOV 117	return TSDB_CODE_INVALID_PARA;	×
118	}
119
120	uint32_t bucketIndex = ht->hash(obj) % ht->numBuckets;	105,338,055✔
121	for (SVHashEntry* entry = ht->buckets[bucketIndex]; entry != NULL; entry = entry->next) {	120,460,501✔
122	if (ht->compare(entry->obj, obj) == 0) {	96,638,980✔
123	*retObj = entry->obj;	81,494,114✔
124	return 0;	81,497,552✔
125	}
126	}
127
128	*retObj = NULL;	23,817,624✔
129	return TSDB_CODE_NOT_FOUND;	23,818,442✔
130	}
131
132	int32_t vHashDrop(SVHashTable* ht, const void* obj) {	24,208,797✔
133	if (ht == NULL \|\| obj == NULL) {	24,208,797✔
134	return TSDB_CODE_INVALID_PARA;	×
135	}
136
137	uint32_t bucketIndex = ht->hash(obj) % ht->numBuckets;	24,208,797✔
138	for (SVHashEntry** entry = &ht->buckets[bucketIndex]; entry != NULL; entry = &(entry)->next) {	27,273,744✔
139	if (ht->compare((*entry)->obj, obj) == 0) {	27,273,222✔
140	SVHashEntry* tmp = *entry;	24,208,031✔
141	entry = (entry)->next;	24,208,031✔
142	taosMemoryFree(tmp);	24,208,031✔
143	ht->numEntries--;	24,208,797✔
144	if (ht->numBuckets > VNODE_HASH_DEFAULT_NUM_BUCKETS && ht->numEntries < ht->numBuckets / 4) {	24,208,797✔
145	vHashRehash(ht, ht->numBuckets / 2);	7,608✔
146	}
147	return 0;	24,208,608✔
148	}
149	}
150
151	return TSDB_CODE_NOT_FOUND;	×
152	}

taosdata / TDengine / #4998

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