wms / hhwms · Files

package com.huaheng.pc.config.customer.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;

import com.huaheng.common.utils.Wrappers;

import com.baomidou.mybatisplus.extension.service.IService;

import com.huaheng.common.utils.security.ShiroUtils;

import com.huaheng.pc.config.customer.domain.Customer;
import com.huaheng.pc.config.customer.mapper.CustomerMapper;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

import java.util.List;
import java.util.Map;

@Service("CustomerService")

public class CustomerServiceImpl extends ServiceImpl<CustomerMapper, Customer> implements IService<Customer> {

    public List<Map<String, Object>> getCode(){
        LambdaQueryWrapper<Customer> lambda = Wrappers.lambdaQuery();
        lambda.select(Customer::getCode, Customer::getId, Customer::getName)
                .eq(Customer::getWarehouseCode, ShiroUtils.getWarehouseCode());
        return this.listMaps(lambda);
    }

    public Customer getCustomerByCode(String code){
        return this.getOne(new LambdaQueryWrapper<Customer>()
                .eq(Customer::getCode,code)
                .eq(Customer::getWarehouseCode,ShiroUtils.getWarehouseCode()));
    }

    public Customer getCustomerByCode(String code,String warehouseCode){
        return this.getOne(new LambdaQueryWrapper<Customer>()
                .eq(Customer::getCode,code)
                .eq(Customer::getWarehouseCode,warehouseCode));
    }

}