swift_concurrency_hooks.cpp

/*
 * swift_concurrency_hooks.cpp
 *
 * This source file is part of the FoundationDB open source project
 *
 * Copyright 2013-2024 Apple Inc. and the FoundationDB project authors
 *
 * 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
 *
 *     http://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.
 */

#include "flow/swift_concurrency_hooks.h"
#include "flow/swift.h"
#include "flow/swift/ABI/Task.h"
#include "flow/TLSConfig.actor.h"

// ==== ----------------------------------------------------------------------------------------------------------------

struct SwiftJobTask final : public N2::Task, public FastAllocated<SwiftJobTask> {
	swift::Job* job;
	explicit SwiftJobTask(swift::Job* job) noexcept : job(job) {}

	void operator()() override {
		swift_job_run(job, ExecutorRef::generic());
		delete this;
	}
};

// ==== ----------------------------------------------------------------------------------------------------------------

double flow_gNetwork_now() {
	return g_network->now();
}

Future<class Void> flow_gNetwork_delay(double seconds, TaskPriority taskID) {
	return g_network->delay(seconds, taskID);
}

// ==== ----------------------------------------------------------------------------------------------------------------
// ==== Net2 hooks

SWIFT_CC(swift)
void net2_enqueueGlobal_hook_impl(swift::Job* _Nonnull job, void (*_Nonnull)(swift::Job*) __attribute__((swiftcall))) {
	// TODO: can't access Net2 since it's incomplete here, would be nicer to not expose API on INetwork I suppose
	auto net = g_network;
	ASSERT(net);

	//	auto swiftPriority = job->getPriority();
	//	int64_t priority = swift_priority_to_net2(swiftPriority); // default to lowest "Min"
	//
	//	TaskPriority taskID = TaskPriority::DefaultOnMainThread; // FIXME: how to determine
	//
	//	SwiftJobTask* jobTask = new SwiftJobTask(job);
	//	N2::OrderedTask* orderedTask = new N2::OrderedTask(priority, taskID, jobTask);

	net->_swiftEnqueue(job);
}

void swift_job_run_generic(swift::Job* _Nonnull job) {
	// NOTE: Guarded because swift_job_run is external import.
#ifdef WITH_SWIFT
	swift_job_run(job, ExecutorRef::generic());
#endif
}