summaryrefslogtreecommitdiff
path: root/vendor/github.com/bytedance/sonic/loader/register.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/bytedance/sonic/loader/register.go')
-rw-r--r--vendor/github.com/bytedance/sonic/loader/register.go55
1 files changed, 55 insertions, 0 deletions
diff --git a/vendor/github.com/bytedance/sonic/loader/register.go b/vendor/github.com/bytedance/sonic/loader/register.go
new file mode 100644
index 000000000..d32f9fa50
--- /dev/null
+++ b/vendor/github.com/bytedance/sonic/loader/register.go
@@ -0,0 +1,55 @@
+// +build !bytedance_tango
+
+/**
+ * Copyright 2024 ByteDance Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package loader
+
+import (
+ "sync/atomic"
+ "unsafe"
+)
+
+func registerModule(mod *moduledata) {
+ registerModuleLockFree(&lastmoduledatap, mod)
+}
+
+func registerModuleLockFree(tail **moduledata, mod *moduledata) {
+ for {
+ oldTail := loadModule(tail)
+ if casModule(tail, oldTail, mod) {
+ storeModule(&oldTail.next, mod)
+ break
+ }
+ }
+}
+
+func loadModule(p **moduledata) *moduledata {
+ return (*moduledata)(atomic.LoadPointer((*unsafe.Pointer)(unsafe.Pointer(p))))
+}
+
+func storeModule(p **moduledata, value *moduledata) {
+ atomic.StorePointer((*unsafe.Pointer)(unsafe.Pointer(p)), unsafe.Pointer(value))
+}
+
+func casModule(p **moduledata, oldValue *moduledata, newValue *moduledata) bool {
+ return atomic.CompareAndSwapPointer(
+ (*unsafe.Pointer)(unsafe.Pointer(p)),
+ unsafe.Pointer(oldValue),
+ unsafe.Pointer(newValue),
+ )
+}
+