pkg/core/runtime/runtime.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.
 */

package runtime

import (
	"sync"
	"time"

	"github.com/apache/dubbo-kubernetes/pkg/core/client/cert"
	"github.com/apache/dubbo-kubernetes/pkg/core/client/webhook"

	"github.com/apache/dubbo-kubernetes/pkg/core/kubeclient/client"

	dubbo_cp "github.com/apache/dubbo-kubernetes/pkg/config/app/dubbo-cp"
	"github.com/apache/dubbo-kubernetes/pkg/core/cert/provider"
	"github.com/apache/dubbo-kubernetes/pkg/core/runtime/component"
	"github.com/apache/dubbo-kubernetes/pkg/cp-server/server"
)

type Runtime interface {
	RuntimeInfo
	RuntimeContext
	component.Manager
}

type RuntimeInfo interface {
	GetInstanceId() string
	SetClusterId(clusterId string)
	GetClusterId() string
	GetStartTime() time.Time
}

type runtimeInfo struct {
	mtx sync.RWMutex

	instanceId string
	clusterId  string
	startTime  time.Time
}

func (i *runtimeInfo) GetInstanceId() string {
	return i.instanceId
}

func (i *runtimeInfo) SetClusterId(clusterId string) {
	i.mtx.Lock()
	defer i.mtx.Unlock()
	i.clusterId = clusterId
}

func (i *runtimeInfo) GetClusterId() string {
	i.mtx.RLock()
	defer i.mtx.RUnlock()
	return i.clusterId
}

func (i *runtimeInfo) GetStartTime() time.Time {
	return i.startTime
}

type RuntimeContext interface {
	Config() *dubbo_cp.Config
	GrpcServer() *server.GrpcServer
	CertStorage() *provider.CertStorage
	CertClient() cert.Client
	KubeClient() *client.KubeClient
	WebHookClient() webhook.Client
}

type runtime struct {
	RuntimeInfo
	RuntimeContext
	component.Manager
}

var _ RuntimeInfo = &runtimeInfo{}

var _ RuntimeContext = &runtimeContext{}

type runtimeContext struct {
	cfg           *dubbo_cp.Config
	grpcServer    *server.GrpcServer
	certStorage   *provider.CertStorage
	kubeClient    *client.KubeClient
	certClient    cert.Client
	webhookClient webhook.Client
}

func (rc *runtimeContext) WebHookClient() webhook.Client {
	return rc.webhookClient
}

func (rc *runtimeContext) CertClient() cert.Client {
	return rc.certClient
}

func (rc *runtimeContext) CertStorage() *provider.CertStorage {
	return rc.certStorage
}

func (rc *runtimeContext) Config() *dubbo_cp.Config {
	return rc.cfg
}

func (rc *runtimeContext) GrpcServer() *server.GrpcServer {
	return rc.grpcServer
}

func (rc *runtimeContext) KubeClient() *client.KubeClient {
	return rc.kubeClient
}