package db

import (
	"chainweaver.org.cn/chainweaver/did/did-mgr-common-service/internal/config"
	"context"
	"time"
)

func FindEnterpriseInfoById(id int64) (*EnterpriseInfo, error) {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	enterpriseInfo := &EnterpriseInfo{}
	tx := config.DB.WithContext(ctx)
	if err := tx.Where("id = ?", id).First(enterpriseInfo).Error; err != nil {
		return nil, err
	}
	return enterpriseInfo, nil
}

func UpdateEnterpriseInfo(enterInfo *EnterpriseInfo) error {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	return config.DB.WithContext(ctx).Model(enterInfo).Where("id = ?", enterInfo.EnterpriseID).Updates(enterInfo).Error
}