ch_os_app_tag.go

/*
 * Copyright (c) 2024 Yunshan Networks
 *
 * 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.
 */

package tagrecorder

import (
	"github.com/deepflowio/deepflow/server/controller/common"
	"github.com/deepflowio/deepflow/server/controller/db/mysql"
	"github.com/deepflowio/deepflow/server/controller/recorder/pubsub/message"
)

type ChOSAppTag struct {
	SubscriberComponent[*message.ProcessFieldsUpdate, message.ProcessFieldsUpdate, mysql.Process, mysql.ChOSAppTag, OSAPPTagKey]
}

func NewChOSAppTag() *ChOSAppTag {
	mng := &ChOSAppTag{
		newSubscriberComponent[*message.ProcessFieldsUpdate, message.ProcessFieldsUpdate, mysql.Process, mysql.ChOSAppTag, OSAPPTagKey](
			common.RESOURCE_TYPE_PROCESS_EN, RESOURCE_TYPE_CH_OS_APP_TAG,
		),
	}
	mng.subscriberDG = mng
	return mng
}

// onResourceUpdated implements SubscriberDataGenerator
func (c *ChOSAppTag) onResourceUpdated(sourceID int, fieldsUpdate *message.ProcessFieldsUpdate, db *mysql.DB) {
	keysToAdd := make([]OSAPPTagKey, 0)
	targetsToAdd := make([]mysql.ChOSAppTag, 0)
	keysToDelete := make([]OSAPPTagKey, 0)
	targetsToDelete := make([]mysql.ChOSAppTag, 0)
	var chItem mysql.ChOSAppTag
	var updateKey OSAPPTagKey
	updateInfo := make(map[string]interface{})

	if fieldsUpdate.OSAPPTags.IsDifferent() {
		_, new := common.StrToJsonAndMap(fieldsUpdate.OSAPPTags.GetNew())
		_, old := common.StrToJsonAndMap(fieldsUpdate.OSAPPTags.GetOld())

		for k, v := range new {
			oldV, ok := old[k]
			if !ok {
				keysToAdd = append(keysToAdd, OSAPPTagKey{PID: sourceID, Key: k})
				targetsToAdd = append(targetsToAdd, mysql.ChOSAppTag{
					PID:   sourceID,
					Key:   k,
					Value: v,
				})
			} else {
				if oldV != v {
					updateKey = OSAPPTagKey{PID: sourceID, Key: k}
					updateInfo[k] = v
					db.Where("pid = ? and `key` = ?", sourceID, k).First(&chItem) // TODO common
					if chItem.PID == 0 {
						keysToAdd = append(keysToAdd, OSAPPTagKey{PID: sourceID, Key: k})
						targetsToAdd = append(targetsToAdd, mysql.ChOSAppTag{
							PID:   sourceID,
							Key:   k,
							Value: v,
						})
					} else if len(updateInfo) > 0 {
						c.SubscriberComponent.dbOperator.update(chItem, updateInfo, updateKey, db)
					}
				}
			}
		}
		for k := range old {
			if _, ok := new[k]; !ok {
				keysToDelete = append(keysToDelete, OSAPPTagKey{PID: sourceID, Key: k})
				targetsToDelete = append(targetsToDelete, mysql.ChOSAppTag{
					PID: sourceID,
					Key: k,
				})
			}
		}
	}
	if len(keysToAdd) > 0 {
		c.SubscriberComponent.dbOperator.add(keysToAdd, targetsToAdd, db)
	}
	if len(keysToDelete) > 0 {
		c.SubscriberComponent.dbOperator.delete(keysToDelete, targetsToDelete, db)
	}
}

// onResourceUpdated implements SubscriberDataGenerator
func (c *ChOSAppTag) sourceToTarget(md *message.Metadata, source *mysql.Process) (keys []OSAPPTagKey, targets []mysql.ChOSAppTag) {
	_, osAppTagsMap := common.StrToJsonAndMap(source.OSAPPTags)

	for k, v := range osAppTagsMap {
		keys = append(keys, OSAPPTagKey{PID: source.ID, Key: k})
		targets = append(targets, mysql.ChOSAppTag{
			PID:      source.ID,
			Key:      k,
			Value:    v,
			TeamID:   md.TeamID,
			DomainID: md.DomainID,
		})
	}
	return
}

// softDeletedTargetsUpdated implements SubscriberDataGenerator
func (c *ChOSAppTag) softDeletedTargetsUpdated(targets []mysql.ChOSAppTag, db *mysql.DB) {

}